def test_delete_user_null(self): """ Tests that nothing breaks when deleting a nonexistant resident """ expected = User.query.count() # This id is NOT the resident with self.assertRaises(AttributeError): UserService.delete_user(self.test_user.id + 1) self.assertEqual(User.query.count(), expected)
def test_delete_user(self): """ Tests that a user can be deleted """ user = UserService.create_user('*****@*****.**', '', '', 'STAFF') expected = User.query.count() - 1 # check if deletion successful UserService.delete_user(user.id) self.assertEqual(expected, User.query.count())
def test_delete_user_resident(self): """ Tests that deleting a user deletes their resident info """ resident = UserService.create_user('*****@*****.**', 'John', 'Smith', 'RESIDENT') expected_user = User.query.count() - 1 expected_resident = Resident.query.count() - 1 UserService.delete_user(resident.id) self.assertEqual(User.query.count(), expected_user) self.assertEqual(Resident.query.count(), expected_resident)
def test_delete_user_last_admin(self): """ Tests that a user can be deleted """ expected = self.db.session.query(User).count() # check if deletion successful self.assertFalse(UserService.delete_user(self.test_user.id)) self.assertEqual(self.db.session.query(User).count(), expected)