def test_director():
movie = Movie("Moana", 2016)
director = Director("Ron Clements")
movie.director = director
assert repr(movie.director) == "<Director Ron Clements>"
movie.director = "fake imposter director"
assert repr(movie.director) == "<Director Ron Clements>"
movie.director = Director("fake imposter director")
assert repr(movie.director) == "<Director Ron Clements>"
def test_movie_director():
# correct input
movie1 = Movie("Moana", 2016)
director1 = Director("Ron Clements")
movie1.director = director1
assert movie1.director == director1
# incorrect input
movie2 = Movie("Frozen", 2014)
movie2.director = "Chris Buck"
assert movie2.director is None
def read_csv_file(self, file_name):
with open(file_name, mode='r', encoding='utf-8-sig') as csvfile:
movie_file_reader = csv.DictReader(csvfile)
for row in movie_file_reader:
id = int(row['Rank'])
title = row['Title']
release_year = int(row['Year'])
movie = Movie(title, release_year, id)
self.add_movie(movie)
actors = row["Actors"]
actors = actors.split(",")
for actor in actors:
actor = Actor(actor)
self.add_actor(actor)
actor = self.get_actor(actor.actor_full_name)
movie.add_actor(actor)
actor.add_movie(movie)
director = Director(row["Director"])
self.add_director(director)
director = self.get_director(director.director_full_name)
movie.director = director
director.add_movie(movie)
genres = row["Genre"]
genres = genres.split(",")
for genre in genres:
genre = Genre(genre)
movie.add_genre(genre)
self.add_genre(genre)
movie.description = row["Description"]
movie.runtime_minutes = int(row["Runtime (Minutes)"])
def test_init():
movie = Movie("Moana", 2016)
assert repr(movie) == "<Movie Moana, 2016>"
assert movie.title == "Moana"
movie1 = Movie("", "1")
assert repr(movie1) == "<Movie None, None>"
director = Director("Ron Clements")
movie.director = director
assert repr(movie.director) == "<Director Ron Clements>"
movie.description = "Moana, daughter of chief Tui, embarks on a journey to return the heart of goddess Te Fitti from Maui, a demigod, after the plants and the fish on her island start dying due to a blight."
assert movie.description == "Moana, daughter of chief Tui, embarks on a journey to return the heart of goddess Te Fitti from Maui, a demigod, after the plants and the fish on her island start dying due to a blight."
actors = [
Actor("Auli'i Cravalho"),
Actor("Dwayne Johnson"),
Actor("Rachel House"),
Actor("Temuera Morrison")
]
for actor in actors:
movie.add_actor(actor)
assert repr(
movie.actors
) == "[<Actor Auli'i Cravalho>, <Actor Dwayne Johnson>, <Actor Rachel House>, <Actor Temuera Morrison>]"
movie.runtime_minutes = 107
assert repr(movie.runtime_minutes) == "107"
def dict_to_movie(dict):
movie = Movie(dict.title, dict.release)
movie.description = dict.description
movie.director = dict.director
movie.actors = dict.actors
movie.genres = dict.genres
movie.runtime_minutes = dict.runtime
movie.revenue = dict.revenue
def test_repository_does_not_add_a_review_without_a_user(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie = Movie("Asdsa", 2014)
movie.description = "Sam is sam he is"
movie.director = Director('James')
movie.runtime_minutes = 140
repo.add_movie(movie)
review = Review(None, movie, "Wow", 8, datetime.today())
with pytest.raises(RepositoryException):
repo.add_review(review)
def test_repository_can_add_a_movie(session_factory):
repo = SqlAlchemyRepository(session_factory)
movie1 = Movie("Sam I am", 2000)
movie1.description = "Sam is sam he is"
movie1.director = Director('James')
movie1.runtime_minutes = 140
repo.add_movie(movie1)
movie = repo.get_movie(1001)
assert movie == Movie('Sam I am', 2000)
def test_repository_can_add_a_review(session_factory):
repo = SqlAlchemyRepository(session_factory)
user = User('aidan', 'hi1234')
movie = Movie("Asdsa", 2014)
movie.description = "Sam is sam he is"
movie.director = Director('James')
movie.runtime_minutes = 140
review = make_review("Wow good movie", user, movie, 9)
repo.add_user(user)
repo.add_movie(movie)
repo.add_review(review)
reviews = repo.get_reviews()
assert len(reviews) == 1
assert review in reviews
def read_csv_file(self):
with open(self.__file_name, mode='r', encoding='utf-8-sig') as csvfile:
movie_file_reader = csv.DictReader(csvfile)
# index = 0
for row in movie_file_reader:
title = row['Title']
release_year = int(row['Year'])
movie = Movie(title, release_year)
movie.description = row['Description']
movie.director = Director(row['Director'])
if Director(
row['Director']) not in self.__dataset_of_directors:
self.__dataset_of_directors.append(
Director(row['Director']))
for actor in row['Actors'].split(','):
movie.add_actor(Actor(actor))
if Actor(actor) not in self.dataset_of_actors:
self.dataset_of_actors.append(Actor(actor))
for genre in row['Genre'].split(','):
movie.add_genre(Genre(genre))
if Genre(genre) not in self.dataset_of_genres:
self.dataset_of_genres.append(Genre(genre))
movie.runtime_minutes = int(row['Runtime (Minutes)'])
try:
movie.revenue = float(row['Revenue (Millions)'])
except:
pass
self.__dataset_of_movies.append(movie)
