Beispiel #1
0
def test_runs2(db_instance):
    email = 'mock' + str(random.randint(1, 101)) + '@mock.com'
    password = '******'
    example = User()
    example.email = email
    example.set_password(password)

    runs_id = ['1', '2', '3', '4', '5', '6', '7']

    db_instance.session.add(example)

    for i in runs_id:
        run = Run()
        run.runner = example
        run.strava_id = i
        db_instance.session.add(run)

    db_instance.session.commit()
    user_id = example.get_id()
    runId = 150

    run = db_instance.session.query(Run).filter(
        Run.runner_id == example.get_id(), Run.id == runId).first()
    assert run == None

    _delete_user(example)
Beispiel #2
0
def test_runs1(db_instance):
    email = 'mock' + str(random.randint(1, 101)) + '@mock.com'
    password = '******'
    example = User()
    example.email = email
    example.set_password(password)

    runs_id = ['1', '2', '3', '4', '5', '6', '7']

    db_instance.session.add(example)

    for i in runs_id:
        run = Run()
        run.runner = example
        run.strava_id = i
        db_instance.session.add(run)

    db_instance.session.commit()
    user_id = example.get_id()

    previous_run = db_instance.session.query(Run).filter(
        Run.runner_id == example.get_id(),
        Run.id < 7).order_by(Run.id.desc()).first()
    assert previous_run.id == 6

    q = db_instance.session.query(Run).filter(
        Run.runner_id == example.get_id(), Run.id == 4)
    q.delete(synchronize_session=False)
    db_instance.session.commit()
    previous_run = db_instance.session.query(Run).filter(
        Run.runner_id == example.get_id(),
        Run.id < 5).order_by(Run.id.desc()).first()
    assert previous_run.id == 3

    _delete_user(example)
Beispiel #3
0
    def test1(self):
        global _app
        tested_app = create_app(debug=True)
        _app = tested_app
        with tested_app.test_client() as client:
            with client.session_transaction() as sess:
                db.drop_all()
                db.create_all()

                # create user
                user_a = User()
                user_a.email = '*****@*****.**'
                user_a.set_password('test')
                db.session.add(user_a)
                db.session.commit()

                user_b = User()
                user_b.email = '*****@*****.**'
                user_b.set_password('test')
                db.session.add(user_b)
                db.session.commit()

                # create story
                story = Story()
                story.text = 'Text a'
                story.likes = 0
                story.dislikes = 0
                story.author_id = user_a.get_id()
                story.roll = {
                    'dice':
                    ['bike', 'tulip', 'happy', 'cat', 'ladder', 'rain']
                }
                db.session.add(story)
                db.session.commit()

                # add like
                like = Reaction()
                like.marked = 0
                like.story_id = story.id
                like.user_id = user_b.id
                like.type = 1
                db.session.add(story)
                db.session.commit()

                # create 1000 user and like the story
                users = []
                for i in range(10):
                    user = User()
                    user.email = 'user' + str(i) + '@test.com'
                    user.set_password('test')
                    db.session.add(user)
                    users.append(user)
                    db.session.add(story)

                db.session.commit()
                for u in users:
                    add_reaction(u.id, 1, 1)
                #
                # reaction = Reaction.query.count()
                # print(str(reaction)))
                res = update_reactions.apply_async(args=[1], time_limit=3)
                res.get()
                q = Story.query.filter_by(author_id=1).first()
                self.assertEqual(int(q.likes), 10)