def test_get_all_simple_users(self): """ Ensure get all simple users behaves correctly. """ add_simple_user(simple_user_dict) add_simple_user(simple_user_dict2) with self.client: response = self.client.get("/users/simple") data = json.loads(response.data.decode()) self.assertEqual(response.status_code, 200) self.assertEqual(len(data["simple_users"]), 2) self.assertIn( "*****@*****.**", data["simple_users"][0]["email"] ) self.assertIn( "*****@*****.**", data["simple_users"][1]["email"] )
def test_get_simple_user(self): """ Ensure get single user behaves correctly. """ simple_user = add_simple_user(simple_user_dict) with self.client: response = self.client.get(f"/users/simple/{simple_user.id}") data = json.loads(response.data.decode()) self.assertEqual(response.status_code, 200) self.assertEqual( simple_user.email, data["simple_user"]["email"] )
def test_to_dict(self): user = add_simple_user(simple_user_dict) self.assertTrue(isinstance(user.to_dict(), dict))
def test_add_user_duplicate_email(self): add_simple_user(simple_user_dict) duplicate_user = SimpleUser(**simple_user_dict) db.session.add(duplicate_user) self.assertRaises(IntegrityError, db.session.commit)
def test_add_simple_user(self): user = add_simple_user(simple_user_dict) self.assertTrue(user.id) self.assertEqual(user.email, '*****@*****.**') self.assertTrue(user.online)
def seed_db(): """Seeds the database.""" add_simple_user(simple_user_dict) add_simple_user(simple_user_dict2) add_user(user_dict) add_user(user_dict2)