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)
