heckyel/mediagoblin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add test for get_all_media()

Browse Source 
Check for it's returned values.
This commit is contained in:
vijeth-aradhya 2017-08-16 21:51:55 +05:30
parent 7c5c0cfcb6
commit 283043f14d
1 changed files with 20 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mediagoblin/tests/test_submission.py
View File

@ -55,7 +55,7 @@ from celery import Signature
from mediagoblin.tests.tools import ( from mediagoblin.tests.tools import (
fixture_add_user, fixture_add_collection, get_app) fixture_add_user, fixture_add_collection, get_app)
from mediagoblin import mg_globals from mediagoblin import mg_globals
from mediagoblin.db.models import MediaEntry, User, LocalUser, Activity from mediagoblin.db.models import MediaEntry, User, LocalUser, Activity, MediaFile
from mediagoblin.db.base import Session from mediagoblin.db.base import Session
from mediagoblin.tools import template from mediagoblin.tools import template
from mediagoblin.media_types.image import ImageMediaManager from mediagoblin.media_types.image import ImageMediaManager
@ -561,6 +561,25 @@ class TestSubmissionVideo(BaseTestSubmission):
for each_res in video_config['available_resolutions']: for each_res in video_config['available_resolutions']:
assert (('webm_' + str(each_res)) in media.media_files) assert (('webm_' + str(each_res)) in media.media_files)
result = media.get_all_media()
if result[0][0] == 'default':
media_file = MediaFile.query.filter_by(media_entry=media.id,
name=('webm_video')).first()
assert len(result) == 1
assert len(result[0]) == 3
assert result[0][1] == list(ACCEPTED_RESOLUTIONS['webm'])
assert result[0][2] == media_file.file_path
else:
assert len(result) == len(video_config['available_resolutions'])
for i in range(len(video_config['available_resolutions'])):
assert len(result[i]) == 3
media_file = MediaFile.query.filter_by(media_entry=media.id,
name=('webm_'+str(result[i][0]))).first()
assert result[i][0] == video_config['available_resolutions'][i]
assert result[i][1] == list(ACCEPTED_RESOLUTIONS[
video_config['available_resolutions'][i]])
assert result[i][2] == media_file.file_path
@mock.patch('mediagoblin.media_types.video.processing.processing_cleanup.signature') @mock.patch('mediagoblin.media_types.video.processing.processing_cleanup.signature')
@mock.patch('mediagoblin.media_types.video.processing.complimentary_task.signature') @mock.patch('mediagoblin.media_types.video.processing.complimentary_task.signature')
@mock.patch('mediagoblin.media_types.video.processing.main_task.signature') @mock.patch('mediagoblin.media_types.video.processing.main_task.signature')