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 add(self, new_movie):
Movie_Repository.add(self, new_movie)
self._store_file()
def add(self, new_client):
Movie_Repository.add(self, new_client)
self._save_file()
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)