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)