def populate_user(): new_user = User() new_user.email = "*****@*****.**" new_user.phone = '3333333333' new_user.firstname = "firstname_test" new_user.lastname = "lastname_test" new_user.password = "******" new_user.dateofbirth = datetime.date(2020, 10, 5) new_user.role = "customer" return new_user
def add_user(email, phone, firstname, lastname, password, date, role): new_user = User() new_user.email = email new_user.phone = phone new_user.firstname = firstname new_user.lastname = lastname new_user.password = password new_user.role = role new_user.dateofbirth = date db.session.add(new_user) db.session.commit() return db.session.query(User).filter(User.email == email).first()
def test_delete_wrong_author_story_index(self): global _app if _app is None: tested_app = create_app(debug=True) _app = tested_app else: tested_app = _app restart_db_tables(db, tested_app) with tested_app.test_client() as client: # login reply = login(client, '*****@*****.**', 'admin') self.assertEqual(reply.status_code, 200) # add reaction to a story reply = client.get('/stories/reaction/1/1') self.assertEqual(reply.status_code, 200) users = User.query.all() self.assertEqual(len(users), 1) story = db.session.query(Story).filter_by(id=1).first() self.assertNotEqual(story, None) # logout reply = logout(client) self.assertEqual(reply.status_code, 200) # signup user1 = User() user1.firstname = "Mario" user1.lastname = "Rossi" user1.email = "*****@*****.**" user1.dateofbirth = "1994" user1.password = user1.set_password("12345") reply = signup(client, user1) self.assertEqual(reply.status_code, 200) users = User.query.all() self.assertEqual(len(users), 2) story = db.session.query(Story).filter_by(id=1).first() self.assertNotEqual(story, None) stories = Story.query.filter_by(id=1).all() self.assertEqual(len(stories), 1) reactions = Reaction.query.filter_by(story_id=1).all() self.assertEqual(len(reactions), 1) story = db.session.query(Story).filter_by(id=1).first() self.assertNotEqual(story, None) reactions = Reaction.query.filter_by(story_id=1).all() self.assertEqual(len(reactions), 1) # add reaction to a story reply = client.get('/stories/reaction/1/1') self.assertEqual(reply.status_code, 200) reply = client.post('stories/1/remove/index', follow_redirects=True) self.assertEqual(reply.status_code, 200) story = db.session.query(Story).filter_by(id=1).first() self.assertNotEqual(story, None) # logout reply = logout(client) self.assertEqual(reply.status_code, 200)