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)
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)
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)