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())
class TestClientController(unittest.TestCase): def setUp(self): self.repo = Repository() self.rentalRepo = Repository() self.Ucontroller = UndoController() self.client = Client(11, 'alex') self.client2 = Client(12, 'ana') self.controller = ClientController(self.repo, self.Ucontroller, self.rentalRepo) def test_something(self): self.assertRaises(Exception, self.Ucontroller.undo) self.assertRaises(Exception, self.Ucontroller.redo) self.Ucontroller.newOperation() self.controller.addClient(self.client) self.assertEqual(len(self.repo), 1) self.Ucontroller.newOperation() self.controller.removeClient(11) self.assertEqual(len(self.repo), 0) self.Ucontroller.newOperation() self.controller.addClient(self.client) self.Ucontroller.newOperation() self.controller.updateClient(self.client) self.Ucontroller.newOperation() self.controller.addClient(self.client2) self.assertEqual(len(self.controller.getAllClients()), 2) self.assertRaises(ControllerException, self.controller.searchClient, 'nu') self.assertRaises(ControllerException, self.controller.searchClient, '') self.assertEqual(len(self.controller.searchClient('a')), 2) self.Ucontroller.newOperation() self.controller.removeClient(12) self.Ucontroller.undo() self.assertEqual(len(self.controller.getAllClients()), 2) self.Ucontroller.redo() self.assertEqual(len(self.controller.getAllClients()), 1)