def test_load_movie(empty_session):
movie_title = 'Passengers'
genre_list = ['Adventure', 'Drama', 'Romance']
director = 'Morten Tyldum'
description = 'A spacecraft traveling to a distant colony planet and transporting thousands of people has a malfunction in its sleep chambers. As a result, two passengers are awakened 90 years early.'
actors = [
'Jennifer Lawrence', 'Chris Pratt', 'Michael Sheen',
'Laurence Fishburne'
]
run_time = 116
release_year = 2016
insert_movie(empty_session, movie_title, director, genre_list, actors,
description, release_year, run_time)
a_movie = Movie(movie_title, release_year)
a_movie.actors = [Actor(name) for name in actors]
a_movie.description = description
a_movie.director = Director(director)
a_movie.genres = [Genre(name) for name in genre_list]
a_movie.runtime_minutes = run_time
b_movie = empty_session.query(Movie).filter(
Movie._title == 'Passengers').one()
assert b_movie == a_movie
assert b_movie.director == a_movie.director
assert b_movie.runtime_minutes == a_movie.runtime_minutes
assert a_movie.description == b_movie.description
assert len(a_movie.actors) == len(b_movie.actors)
assert len(a_movie.genres) == len(b_movie.genres)
for genre in a_movie.genres:
assert genre in b_movie.genres
for actor in a_movie.actors:
assert actor in b_movie.actors
def test_database_repository_can_add_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
new_movie = Movie('Guardians of Goatboy', 1914)
new_movie.genres = [Genre('Action'), Genre('Sci-Fi')]
new_movie.description = 'Adventures of Space soldiers sworn to protect a boy that is also a goat.'
new_movie.director = Director("Ben Span")
new_movie.actors = [Actor('Chris Pratt'), Actor("Bill Clinton")]
new_movie.runtime_minutes = 400
the_repos_movie_list = repo.get_movies()
count = 0
for movie in the_repos_movie_list:
if movie == Movie('Guardians of Goatboy', 1914):
count += 1
assert count == 0
repo.add_movie(new_movie)
the_repos_movie_list = repo.get_movies()
count = 0
for movie in the_repos_movie_list:
if movie == Movie('Guardians of Goatboy', 1914):
b_movie = movie
count += 1
assert count == 1
assert b_movie.description == new_movie.description
assert b_movie.director == new_movie.director
assert b_movie.runtime_minutes == new_movie.runtime_minutes
assert len(b_movie.genres) == len(new_movie.genres)
for genre in new_movie.genres:
assert genre in b_movie.genres
assert len(b_movie.actors) == len(new_movie.actors)
for actor in new_movie.actors:
assert actor in b_movie.actors
def read_csv_file(self) -> None:
with open(self.__filename, encoding='utf-8-sig') as file_data:
reader = csv.DictReader(file_data)
self.__dataset_of_movies = list()
for row in reader:
new_movie = Movie(row['Title'], int(row['Year']))
new_movie.description = row['Description']
new_movie.genres = [
Genre(i.strip()) for i in row['Genre'].split(',')
]
new_movie.actors = [
Actor(i.strip()) for i in row['Actors'].split(',')
]
new_movie.director = Director(row['Director'].strip())
if row['Runtime (Minutes)'].isdigit():
new_movie.runtime_minutes = int(row['Runtime (Minutes)'])
self.__dataset_of_movies.append(new_movie)
for i in new_movie.genres:
self.add_genre(i)
for i in new_movie.actors:
self.add_actor(i)
self.add_director(new_movie.director)
def test_load_review(empty_session):
movie_title = 'Passengers'
genre_list = ['Adventure', 'Drama', 'Romance']
director = 'Morten Tyldum'
description = 'A spacecraft traveling to a distant colony planet and transporting thousands of people has a malfunction in its sleep chambers. As a result, two passengers are awakened 90 years early.'
actors = [
'Jennifer Lawrence', 'Chris Pratt', 'Michael Sheen',
'Laurence Fishburne'
]
run_time = 116
release_year = 2016
insert_movie(empty_session, movie_title, director, genre_list, actors,
description, release_year, run_time)
a_movie = Movie(movie_title, release_year)
a_movie.actors = [Actor(name) for name in actors]
a_movie.description = description
a_movie.director = Director(director)
a_movie.genres = [Genre(name) for name in genre_list]
a_movie.runtime_minutes = run_time
users = [["jill", '98765432'], ['jim', 'mjydfefs']]
insert_users(empty_session, users)
review_text = 'was definitely a 5'
rating = 5
result = empty_session.execute(
f"SELECT id from users WHERE username = '******'").fetchone()
insert_review(empty_session, movie_title, release_year, result[0],
review_text, rating)
users = empty_session.query(User).all()
found_review = None
for user in users:
if user.username == 'jim':
for review in user.reviews:
if review.movie == Movie(movie_title, release_year):
found_review = review
break
assert found_review.review_text == 'was definitely a 5'
assert found_review.rating == 5
def test_add_genre():
movie = Movie("Moana", 2016)
cartoon = Genre('Cartoon')
movie.genres = [cartoon]
assert movie.genres == [Genre('Cartoon')]
def test_remove_genre():
movie = Movie("Moana", 2016)
movie.genres = [Genre('Cartoon'), Genre('Kids'), Genre('Action')]
movie.remove_genre(Genre('Action'))
assert movie.genres == [Genre('Cartoon'), Genre('Kids')]