class UserTest(unittest.TestCase):
def setUp(self):
self.user = User()
self.user_data = {"user@email": "passwordofuser"}
def testusercreation(self):
self.user.addUser(self.user_data)
self.assertEqual(1, len(self.user.getUsers()))
def testdublicateuseraddition(self):
self.user.addUser(self.user_data)
self.assertEqual(1, len(self.user.getUsers()))
self.assertEqual("Dublicate user", self.user.addUser(self.user_data))
def testmultipleusercreation(self):
self.user.addUser(self.user_data)
self.assertEqual(1, len(self.user.getUsers()))
self.user.addUser({"*****@*****.**": "newpassword"})
self.assertEqual(2, len(self.user.getUsers()))
def testgetsingleuser(self):
self.user.addUser(self.user_data)
self.assertEqual(1, len(self.user.getUsers()))
self.assertIn("user@email", self.user.getUser("user@email"))
def testdeleteuser(self):
self.user.addUser(self.user_data)
self.assertEqual(1, len(self.user.getUsers()))
self.user.addUser({"*****@*****.**": "newpassword"})
self.assertEqual(2, len(self.user.getUsers()))
self.user.deleteUser("*****@*****.**")
self.assertEqual(1, len(self.user.getUsers()))
def testupdateuser(self):
self.user.addUser(self.user_data)
self.assertEqual(1, len(self.user.getUsers()))
new_data = {"*****@*****.**": "123456"}
self.user.updateUser("user@email", new_data)
self.assertIn("*****@*****.**",
self.user.getUser("*****@*****.**"))
