def test_set_admin(self):
""" Checks the user admin works """
self.assertFalse(self.user1.is_admin)
self.user1.set_admin()
self.assertTrue(self.user1.is_admin)
self.assertTrue(User.get_user_by_id(self.user1.get_id()).is_admin)
self.assertFalse(User.get_user_by_id(self.user2.get_id()).is_admin)
self.user1.set_admin(False)
self.assertFalse(self.user1.is_admin)
self.assertFalse(User.get_user_by_id(self.user1.get_id()).is_admin)
self.assertFalse(User.get_user_by_id(self.user2.get_id()).is_admin)
def load_user(user_id: int) -> User:
"""
Loads the user from the database
:param user_id: id of the user to load
:return: a User instance
"""
return User.get_user_by_id(user_id)
def test_get_user_by_id(self):
""" Checks that getting a user by id indeed returns the same user (also tests our __eq__ function) """
user_id = self.user1.get_id()
new_user = User.get_user_by_id(user_id)
self.assertEqual(self.user1, new_user)
