def testAddClient(self):
client1 = Client(1, "Name1")
client2 = Client(1, "Name2")
repo = Repository()
functions = ClientController(repo, Statistics(repo))
msg1 = functions.addClient(client1.getId(), client1.getName())
self.assertTrue(len(msg1) == 0)
self.assertTrue(functions.getClients()[0].getId() == 1)
msg2 = functions.addClient(client2.getId(), client2.getName())
self.assertTrue(msg2 == "Cannot add an existing element")
self.assertTrue(functions.getClients()[0].getId() == 1)
def testRemoveClient(self):
client1 = Client(1, "Name1")
client2 = Client(2, "Name2")
repo = Repository()
functions = ClientController(repo, Statistics(repo))
functions.addClient(client1.getId(), client1.getName())
functions.addClient(client2.getId(), client2.getName())
msg1 = functions.removeClient(1)
self.assertTrue(len(msg1) == 0)
self.assertTrue(functions.getClients()[0].getId() == client2.getId())
self.assertTrue(functions.getClients()[0].getName() == client2.getName())
msg2 = functions.removeClient(1)
self.assertTrue(msg2 == "The provided ID does not exist")
self.assertTrue(functions.getClients()[0].getId() == client2.getId())
self.assertTrue(functions.getClients()[0].getName() == client2.getName())
