def test_save(self):
# Arrange
movie_name = "Black swan"
year = 2011
director = "Darren Aronofsky"
poster_url = "https://blackswan/poster"
video_url = "https://blackswan/video"
movie = Movie(title = movie_name, year = year, director = director,
poster_url = poster_url, video_url = video_url)
quote = "I am gonna die"
subtitle = Subtitle(quote = quote, start_time = datetime.now(),
end_time = datetime.now(), movie = movie)
expected_movie = {'title': movie_name, 'year': year, 'director': director,
'poster_url': poster_url, 'video_url': video_url}
subtitle_repo = SubtitleRepo()
# Act
subtitle_repo.save(subtitle)
subtitle_orm = SubtitleORM.objects.all().filter(quote = quote).first()
# Assert
self.assertIsNotNone(subtitle_orm)
movie_orm = subtitle_orm.movie
self.assertIsNotNone(movie_orm)
actual_movie = {'title': movie_orm.title,
'year': movie_orm.year, 'director': movie_orm.director,
'poster_url': movie_orm.poster_url, 'video_url': movie_orm.video_url}
self.assertDictEqual(expected_movie, actual_movie)
def test_scenario(self):
# *********************** Populate the database *******************
movie_LOTR = MovieRepo().save(
Movie(title='The Lord of the Rings',
year=2001,
director='Peter Jackson',
poster_url='http://someurl/',
video_url='http://anotherurl/'))
subtitles_LOTR = SubtitleParser.parse(
os.path.dirname(__file__) + '/res/LOTR.en.srt')
subtitle_repo = SubtitleRepo()
for sub in subtitles_LOTR:
sub.movie = movie_LOTR
subtitle_repo.save(sub)
# -----------------------------------------------------------------
# Arrange
search_quote = "Run, Frodo"
expected_json = [{
"quote":
search_quote,
"movie": {
"id": "1",
"title": 'The Lord of the Rings',
"year": "2001",
"director": "Peter Jackson",
"poster": 'http://someurl/',
"url": 'http://anotherurl/'
},
"quotes": [{
"id": "652",
"quote": "Run, Frodo!",
"time": "00:56:54.940000"
}, {
"id": "1712",
"quote": "Run, Frodo. Go!",
"time": "03:06:04.190000"
}]
}]
request_data = {"quote": search_quote}
# Act
resp = self.client.post('/api/v0/movies/quote/',
request_data,
format='json')
# Assert
self.assertEqual(resp.status_code, status.HTTP_200_OK)
self.assertEqual(expected_json, resp.data)
def handle(self, *args, **options):
movie_orm_lotr = MovieORM.objects.create(
title='Lord of the Rings',
year=2001,
director='Peter Jackson',
poster_url='https://someurl.ru',
video_url='http://videourl.com'
)
movie_lotr = MovieRepo().save(movie_orm_lotr)
subtitle_repo = SubtitleRepo()
subtitles_lotr = SubtitleParser.parse(os.path.dirname(__file__) + '/res/LOTR.ru.srt')
for sub in subtitles_lotr:
sub.movie = movie_lotr
subtitle_repo.save(sub)
self.stdout.write(f'Added {len(subtitles_lotr)} subtitles for movie {movie_lotr.title}')
def test_save__try_to_update_with_none_values(self):
# Arrange
movie_name = "Black swan"
year = 2011
director = "Darren Aronofsky"
poster_url = "https://blackswan/poster"
video_url = "https://blackswan/video"
movie_orm = MovieORM.objects.create(
title=movie_name,
year=year,
director=director,
poster_url=poster_url,
video_url=video_url
)
quote = "I am gonna die"
start_time = time(hour=0, minute=0, second=37, microsecond=673000)
end_time = time(hour=0, minute=0, second=40, microsecond=673000)
subtitle_orm = SubtitleORM.objects.create(
quote=quote,
start_time=start_time,
end_time=end_time,
movie=movie_orm
)
expected_subtitle_data = {
'quote': quote,
'start_time': start_time,
'end_time': end_time,
'movie_id': movie_orm.id
}
# Act
subtitle_repo = SubtitleRepo()
existing_subtitle = subtitle_repo.get(subtitle_orm.id)
existing_subtitle.quote = None
existing_subtitle.movie = None
actual_subtitle = subtitle_repo.save(existing_subtitle)
actual_subtitle_data = {
'quote': actual_subtitle.quote,
'start_time': actual_subtitle.start_time,
'end_time': actual_subtitle.end_time,
'movie_id': actual_subtitle.movie.id
}
# Assert
self.assertDictEqual(expected_subtitle_data, actual_subtitle_data)
def test__deserialize_subtitle_and_update_from_repo(self):
# Arrange
incomplete_subtitle_json = f'{{"id": "{self.subtitle_1.id}", \
"quote": "{self.subtitle_1.quote}"}}'
# Act
subtitle_repo = SubtitleRepo()
subtitle_serializer = SubtitleSerializer()
incomplete_subtitle = subtitle_serializer.deserealize(
incomplete_subtitle_json)
actual_subtitle = subtitle_repo.save(incomplete_subtitle)
# Assert
compare(self.subtitle_1, actual_subtitle)
def save(self, match: Match) -> Match:
"""
Saves given match object if it exists in database,
othervise creates a new
"""
if match.id is None:
saved_match = self._create(match)
else:
if MatchORM.objects.filter(pk=match.id).exists():
match.movie = MovieRepo().save(match.movie)
subtitle_repo = SubtitleRepo()
for sub in match.subtitles:
sub = subtitle_repo.save(sub)
match_orm = MatchORM.objects.get(pk=match.id)
match_orm.quote = set_if_not_none(match_orm.quote, match.quote)
match_orm.save()
else:
saved_match = self._create(match)
return saved_match
def _create(self, match: Match) -> Match:
if match.user_profile is None:
raise NoUserDefinedForMatch()
movie = MovieRepo().save(match.movie)
movie_orm = MovieRepo.Mapper.from_domain(movie)
subtitle_repo = SubtitleRepo()
subtitles = [subtitle_repo.save(sub) for sub in match.subtitles]
subtitles_orm = [
SubtitleRepo.Mapper.from_domain(sub) for sub in subtitles
]
user_profile_orm = UserProfileRepo.Mapper.from_domain(
match.user_profile)
created_match = MatchORM.objects.create(quote=match.quote,
user_profile=user_profile_orm,
movie=movie_orm)
created_match.subtitles.set(subtitles_orm)
created_match.save()
return self.Mapper.to_domain(created_match)
class Command(BaseCommand):
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
# pylint: disable-msg=C0103
self.HOST = 'http://movie-quotes.ru'
self.STATIC_PATH = '/static/'
self._movie_repo = MovieRepo()
self._subtitle_repo = SubtitleRepo()
def handle(self, *args, **options):
# pylint: disable-msg=C0103
# ---------------------- The Lord Of the Rings 2001 ------------------------
movie_the_lord_of_the_rings_2001 = Movie(
title='The Lord of The Rings',
year=2001,
director='Peter Jackson',
poster_url=self.HOST + self.STATIC_PATH +
'the_lord_of_the_rings_2001/poster.jpg',
video_url=self.HOST + self.STATIC_PATH +
'the_lord_of_the_rings_2001/movie.mp4')
dir_name = 'the_lord_of_the_rings_2001'
self._populate_movie(
movie_the_lord_of_the_rings_2001, dir_name,
[self.HOST + self.STATIC_PATH + dir_name + '/' + 'sub.rus.srt'])
# --------------------------- Forrest Gump 1994 ----------------------------
movie_forrest_gump_1994 = Movie(
title='Forrest Gump',
year=1994,
director='Robert Zemeckis',
poster_url=self.HOST + self.STATIC_PATH +
'forrest_gump_1994/poster.jpg',
video_url=self.HOST + self.STATIC_PATH +
'forrest_gump_1994/movie.mp4')
dir_name = 'forrest_gump_1994'
self._populate_movie(
movie_forrest_gump_1994, dir_name,
[self.HOST + self.STATIC_PATH + dir_name + '/' + 'sub.rus.srt'])
# --------------------------- The Godfather 1972 ----------------------------
movie_the_godfather_1972 = Movie(
title='The Godfather',
year=1972,
director='Francis Ford Coppola',
poster_url=self.HOST + self.STATIC_PATH +
'the_godfather_1972/poster.jpg',
video_url=self.HOST + self.STATIC_PATH +
'the_godfather_1972/movie.mp4')
dir_name = 'the_godfather_1972'
self._populate_movie(movie_the_godfather_1972, dir_name, [
self.HOST + self.STATIC_PATH + dir_name + '/' + 'sub_pt1.rus.srt',
self.HOST + self.STATIC_PATH + dir_name + '/' + 'sub_pt2.rus.srt'
])
def _download_files(self, tmp_dirname, urls: list) -> List[str]:
'''
Downloads files from given urls into temporary directory tmp_dirname.
Returns list of filenames, which were download succesfully.
'''
downloaded_filepaths = []
os.mkdir(tmp_dirname)
self.stdout.write(f'created temporary directory: {tmp_dirname}')
for url in urls:
self.stdout.write(f'downloading file: {url} ...')
filepath = tmp_dirname + '/' + url.split('/')[-1]
urllib.request.urlretrieve(url, filepath)
downloaded_filepaths.append(filepath)
return downloaded_filepaths
def _read_subtitles(self, movie_dirname,
subtitle_files_urls: list) -> List[Subtitle]:
downloaded_files = self._download_files(movie_dirname,
subtitle_files_urls)
subtitles = []
for filepath in downloaded_files:
self.stdout.write(f'reading file with subtitles: {filepath} ...')
subtitles_part = SubtitleParser.parse(filepath)
subtitles += subtitles_part
shutil.rmtree(movie_dirname)
self.stdout.write(f'deleted temporary directory: {movie_dirname}')
return subtitles
def _populate_movie(self, movie, movie_dir_name, subtitle_files_urls):
movie = self._movie_repo.save(movie)
subtitles = self._read_subtitles(movie_dir_name, subtitle_files_urls)
for sub in subtitles:
sub.movie = movie
self._subtitle_repo.save(sub)
self._log(movie, len(subtitles))
def _log(self, movie, subtitles_count):
self.stdout.write('Added Movie:')
self.stdout.write(f'\t{movie.title}\n' + f'\t{movie.year}\n' +
f'\t{movie.director}\n' + f'\t{movie.poster_url}\n' +
f'\t{movie.video_url}\n')
self.stdout.write(f'\tsubtitles count {subtitles_count}')
self.stdout.write('\n')
def test_scenario(self):
# /********************* Populate the database *******************\
movie_LOTR = MovieRepo().save(
Movie(title='The Lord of the Rings',
year=2001,
director='Peter Jackson',
poster_url='http://someurl/',
video_url='http://anotherurl/'))
subtitles_LOTR = SubtitleParser.parse(
os.path.dirname(__file__) + '/res/LOTR.en.srt')
subtitle_repo = SubtitleRepo()
for sub in subtitles_LOTR:
sub.movie = movie_LOTR
subtitle_repo.save(sub)
# \----------------------------------------------------------------/
# /-------------------- User registration -------------------------\
# Arrange
request_data = {
'username': '******',
'email': '*****@*****.**',
'password': '******',
'repeatedPassword': '******'
}
expected_response_status = status.HTTP_200_OK
expected_response = {
'username': '******',
'email': '*****@*****.**'
}
# Act
response = self.client.post('/api/v0/session/registration/',
request_data)
# Assert
self.assertEqual(expected_response_status, response.status_code)
self.assertTrue('token' in response.cookies)
self.assertEqual(expected_response, response.data)
# \----------------------------------------------------------------/
# /----------------- User Login (incorrect data) ------------------\
# Arange
request_data = {
'email': '*****@*****.**',
'password': '******'
}
expected_response_status = status.HTTP_401_UNAUTHORIZED
# Act
response = self.client.post('/api/v0/session/login/', request_data)
# Assert
self.assertEqual(expected_response_status, response.status_code)
# \----------------------------------------------------------------/
# /------------------ User Login (correct data) ------------------\
# Arrange
request_data = {'email': '*****@*****.**', 'password': '******'}
expected_response_status = status.HTTP_200_OK
expected_response_data = {
'username': '******',
'email': '*****@*****.**'
}
# Act
response = self.client.post('/api/v0/session/login/', request_data)
# Assert
self.assertEqual(expected_response_status, response.status_code)
self.assertEqual(expected_response_data, response.data)
# \----------------------------------------------------------------/
# /------------------------- Search -------------------------------\
# Arrange
search_quote = "Run, Frodo"
expected_response_status = status.HTTP_200_OK
expected_response_data = [{
"quote":
search_quote,
"movie": {
"id": "1",
"title": 'The Lord of the Rings',
"year": "2001",
"director": "Peter Jackson",
"poster": 'http://someurl/',
"url": 'http://anotherurl/'
},
"quotes": [{
"id": "652",
"quote": "Run, Frodo!",
"time": "00:56:54.940000"
}, {
"id": "1712",
"quote": "Run, Frodo. Go!",
"time": "03:06:04.190000"
}]
}]
request_data = {"quote": search_quote}
# Act
response = self.client.post('/api/v0/movies/quote/', request_data)
# Assert
self.assertEqual(expected_response_status, response.status_code)
self.assertEqual(expected_response_data, response.data)
# \----------------------------------------------------------------/
# /--------------------- User history update ----------------------\
# Arrange
request_data = {
"movie_id": "1",
"quote": search_quote,
"subtitle_ids": ["652", "1712"]
}
expected_response_status = status.HTTP_200_OK
# Act
response = self.client.post('/api/v0/user/testusername/',
request_data,
format='json')
# Assert
self.assertEqual(expected_response_status, response.status_code)
# \----------------------------------------------------------------/
# /----------------------- History show ---------------------------\
# Arrange
expected_response_status = status.HTTP_200_OK
expected_response_data = [{
"quote":
search_quote,
"movie": {
"id": "1",
"title": 'The Lord of the Rings',
"year": "2001",
"director": "Peter Jackson",
"poster": 'http://someurl/',
"url": 'http://anotherurl/'
},
"quotes": [{
"id": "1712",
"quote": "Run, Frodo. Go!",
"time": "03:06:04.190000"
}, {
"id": "652",
"quote": "Run, Frodo!",
"time": "00:56:54.940000"
}]
}]
# Act
response = self.client.get('/api/v0/user/testusername/')
# Assert
self.assertEqual(expected_response_status, response.status_code)
self.assertEqual(expected_response_data, response.data)