def testClientController(): """ Test ClientController """ repo = ClientsRepository() repo.add(Client(1, "a", 11111)) client = Client(2, "aa", 12121) repo.add(client) controller = ClientController(repo, RentRepository(), UnreController()) controller.removeClient(1) assert controller.getAll() == [client] controller.removeClient(2) assert controller.getAll() == [] try: controller.searchClient(23) assert False except ExceptionMess: pass
def testUnReController(): """ Test UnReController """ client = ClientsRepository() movie = MoviesRepository() rent = RentRepository() unre = UnreController() clientC = ClientController(client, rent, unre) movieC = MovieController(movie, rent, unre) rentC = RentController(client, movie, rent, unre) #Test Client Undo clientC.addClient(Client(1, "A", 12121)) unre.Undo() try: clientC.GetClient(1) assert False except ExceptionMess: pass unre.Redo() try: clientC.GetClient(1) except ExceptionMess: assert False try: unre.Redo() assert False except ExceptionMess: pass unre.Undo() try: clientC.GetClient(1) assert False except ExceptionMess: unre.Redo() clientC.removeClient(1) unre.Undo() try: clientC.GetClient(1) except ExceptionMess: assert False unre.Redo() try: clientC.GetClient(1) assert False except ExceptionMess: pass try: unre.Undo() except ExceptionMess: assert False clientC.updateClient(Client(1, "v", 123)) unre.Undo() assert clientC.GetClient(1) == Client(1, "A", 12121) unre.Redo() assert clientC.GetClient(1) == Client(1, "v", 123) #Test Movie Undo movieC.addMovie(Movie(1, "A", "AA", "AAA")) unre.Undo() try: movieC.GetMovie(1) assert False except ExceptionMess: pass unre.Redo() try: movieC.GetMovie(1) except ExceptionMess: assert False try: unre.Redo() assert False except ExceptionMess: pass unre.Undo() try: movieC.GetMovie(1) assert False except ExceptionMess: unre.Redo() movieC.removeMovie(1) unre.Undo() try: movieC.GetMovie(1) except ExceptionMess: assert False unre.Redo() try: movieC.GetMovie(1) assert False except ExceptionMess: pass try: unre.Undo() except ExceptionMess: assert False movieC.updateMovie(Movie(1, "B", "BB", "BBB")) unre.Undo() assert movieC.GetMovie(1) == Movie(1, "A", "AA", "AAA") unre.Redo() assert movieC.GetMovie(1) == Movie(1, "B", "BB", "BBB") #Test Rent Undo rentC.rentMovie(Rent(1, 1)) unre.Undo() assert not rentC.isClient(1) unre.Redo() assert rentC.isClient(1) try: unre.Redo() assert False except ExceptionMess: pass unre.Undo() assert not rentC.isClient(1) unre.Redo() rentC.returnMovie(Rent(1,1)) unre.Undo() assert rentC.isClient(1) unre.Redo() assert not rentC.isClient(1) try: unre.Undo() except ExceptionMess: assert False