def test_call_to_youtube_api_endpoint(self, build):
build.return_value.playlistItems.return_value.list.\
return_value.execute.return_value = self.short_playlist_response
importer = BulkTaskYoutubeImport(**self.form_data)
importer._fetch_all_youtube_videos('fakeId')
build.assert_called_with('youtube', 'v3', developerKey=self.form_data['youtube_api_server_key'])
def test_call_to_youtube_api_endpoint(self, build):
build.return_value.playlistItems.return_value.list.\
return_value.execute.return_value = self.short_playlist_response
importer = BulkTaskYoutubeImport(**self.form_data)
importer._fetch_all_youtube_videos('fakeId')
build.assert_called_with(
'youtube',
'v3',
developerKey=self.form_data['youtube_api_server_key'])
def test_call_to_youtube_api_short_playlist(self, build):
build.return_value.playlistItems.return_value.list.\
return_value.execute.return_value = self.short_playlist_response
importer = BulkTaskYoutubeImport(**self.form_data)
playlist = importer._fetch_all_youtube_videos('fakeId')
assert playlist == self.short_playlist_response, playlist
def test_call_to_youtube_api_short_playlist(self, build):
build.return_value.playlistItems.return_value.list.\
return_value.execute.return_value = self.short_playlist_response
importer = BulkTaskYoutubeImport(**self.form_data)
playlist = importer._fetch_all_youtube_videos('fakeId')
assert playlist == self.short_playlist_response, playlist
def test_call_to_youtube_api_long_playlist(self, build):
build.return_value.playlistItems.return_value.list.\
return_value.execute.side_effect = [self.long_playlist_response, self.long_playlist_response, self.short_playlist_response]
importer = BulkTaskYoutubeImport(**self.form_data)
expected_playlist = {'items': ''}
expected_playlist['items'] = self.long_playlist_response['items'] + self.long_playlist_response['items'] + self.short_playlist_response['items']
playlist = importer._fetch_all_youtube_videos('fakeId')
assert playlist == expected_playlist, playlist
def test_call_to_youtube_api_long_playlist(self, build):
build.return_value.playlistItems.return_value.list.\
return_value.execute.side_effect = [self.long_playlist_response, self.long_playlist_response, self.short_playlist_response]
importer = BulkTaskYoutubeImport(**self.form_data)
expected_playlist = {'items': ''}
expected_playlist['items'] = self.long_playlist_response[
'items'] + self.long_playlist_response[
'items'] + self.short_playlist_response['items']
playlist = importer._fetch_all_youtube_videos('fakeId')
assert playlist == expected_playlist, playlist
