def testSearch(): movies = Movie_Repository() clients = Clients_Repository() movies.initialise() #print(movies.toString()) src = Search(movies, clients, "Titanic") print(src.search())
class Test_UI(unittest.TestCase): pass def setUp(self): self.repo = Movie_Repository() def test_remove_movieId(self): movie = Movie(12, "Titanic", "Long", "Romance") self.repo.add(movie) self.repo.remove(12) self.assertEqual(len(self.repo.getAll()), 0)
def _load_file(self): try: f = open(self.file_name, 'r') line = f.readline() while line != '': tok = line.split(",") movie = Movie(int(tok[0]), tok[1], tok[2], tok[3]) Movie_Repository.add(self, movie) line = f.readline() except IOError as e: raise Exception("cannot load file - " + str(e)) finally: f.close()
def start(): settings = read_settings() initialise = False if settings['repo_type'] == 'text': clients_repo = TextfileClientsRepo(settings['client_repo']) movies_repo = TextfileMoviesRepo(settings['movie_repo']) rentals_repo = TextfileRentalsRepo(settings['rental_repo']) elif settings['repo_type'] == 'binary': clients_repo = BinaryClientsRepo(settings['clients_binary']) movies_repo = BinaryMoviesRepo(settings['movies_binary']) rentals_repo = BinaryRentalsRepo(settings['rentals_binary']) else: #iter1 = Iterable() #iter2 = Iterable() clients_repo = Clients_Repository() movies_repo = Movie_Repository() rentals_repo = Rental_Repository() initialise = True clients_validator = client_validator.ValidateClient() movies_validator = movie_validator.ValidateMovie() undo_controller = UndoController() rental_controller = RentalController(rentals_repo, undo_controller, initialise) client_controller = ClientController(undo_controller, clients_repo, rentals_repo, initialise) movie_controller = MovieController(undo_controller, movies_repo, rentals_repo, initialise) ui = Ui(undo_controller, rental_controller, client_controller, movie_controller) ui.commandMenu()
def __init__(self, file_name="movies_binary.txt"): Movie_Repository.__init__(self) self._file_name = file_name self._load_file()
def remove(self, pos): Movie_Repository.remove(pos) self._store_file()
def add(self, new_movie): Movie_Repository.add(self, new_movie) self._store_file()
def __init__(self, file_name='repository/clients.txt'): Movie_Repository.__init__(self) self.file_name = file_name self._load_file()
def add(self, new_client): Movie_Repository.add(self, new_client) self._save_file()
def setUp(self): self.repo = Movie_Repository()
def setUp(self): unittest.TestCase.setUp(self) repo = Movie_Repository() movie = Movie(12, "Titanic", "Long", "Romance") repo.add(movie) repo.remove(12)
def test_add_movie(self): repo2 = Movie_Repository() new_movie = Movie(123, "Abc", "desc", "genre") repo2.add(new_movie) #print(repo2) self.assertEqual(len(repo2), 1)