| 1 | # GNU MediaGoblin -- federated, autonomous media hosting |
| 2 | # Copyright (C) 2013 MediaGoblin contributors. See AUTHORS. |
| 3 | # |
| 4 | # This program is free software: you can redistribute it and/or modify |
| 5 | # it under the terms of the GNU Affero General Public License as published by |
| 6 | # the Free Software Foundation, either version 3 of the License, or |
| 7 | # (at your option) any later version. |
| 8 | # |
| 9 | # This program is distributed in the hope that it will be useful, |
| 10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| 12 | # GNU Affero General Public License for more details. |
| 13 | # |
| 14 | # You should have received a copy of the GNU Affero General Public License |
| 15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 16 | |
| 17 | import tempfile |
| 18 | import os |
| 19 | from contextlib import contextmanager |
| 20 | import imghdr |
| 21 | |
| 22 | #os.environ['GST_DEBUG'] = '4,python:4' |
| 23 | import pytest |
| 24 | pytest.importorskip("gi.repository.Gst") |
| 25 | |
| 26 | import gi |
| 27 | gi.require_version('Gst', '1.0') |
| 28 | from gi.repository import Gst |
| 29 | Gst.init(None) |
| 30 | |
| 31 | from mediagoblin.media_types.video.transcoders import (capture_thumb, |
| 32 | VideoTranscoder) |
| 33 | from mediagoblin.media_types.tools import discover |
| 34 | |
| 35 | @contextmanager |
| 36 | def create_data(suffix=None, make_audio=False): |
| 37 | video = tempfile.NamedTemporaryFile() |
| 38 | src = Gst.ElementFactory.make('videotestsrc', None) |
| 39 | src.set_property('num-buffers', 10) |
| 40 | videorate = Gst.ElementFactory.make('videorate', None) |
| 41 | enc = Gst.ElementFactory.make('theoraenc', None) |
| 42 | mux = Gst.ElementFactory.make('oggmux', None) |
| 43 | dst = Gst.ElementFactory.make('filesink', None) |
| 44 | dst.set_property('location', video.name) |
| 45 | pipeline = Gst.Pipeline() |
| 46 | pipeline.add(src) |
| 47 | pipeline.add(videorate) |
| 48 | pipeline.add(enc) |
| 49 | pipeline.add(mux) |
| 50 | pipeline.add(dst) |
| 51 | src.link(videorate) |
| 52 | videorate.link(enc) |
| 53 | enc.link(mux) |
| 54 | mux.link(dst) |
| 55 | if make_audio: |
| 56 | audio_src = Gst.ElementFactory.make('audiotestsrc', None) |
| 57 | audio_src.set_property('num-buffers', 10) |
| 58 | audiorate = Gst.ElementFactory.make('audiorate', None) |
| 59 | audio_enc = Gst.ElementFactory.make('vorbisenc', None) |
| 60 | pipeline.add(audio_src) |
| 61 | pipeline.add(audio_enc) |
| 62 | pipeline.add(audiorate) |
| 63 | audio_src.link(audiorate) |
| 64 | audiorate.link(audio_enc) |
| 65 | audio_enc.link(mux) |
| 66 | pipeline.set_state(Gst.State.PLAYING) |
| 67 | state = pipeline.get_state(3 * Gst.SECOND) |
| 68 | assert state[0] == Gst.StateChangeReturn.SUCCESS |
| 69 | bus = pipeline.get_bus() |
| 70 | message = bus.timed_pop_filtered( |
| 71 | 3 * Gst.SECOND, |
| 72 | Gst.MessageType.ERROR | Gst.MessageType.EOS) |
| 73 | pipeline.set_state(Gst.State.NULL) |
| 74 | if suffix: |
| 75 | result = tempfile.NamedTemporaryFile(suffix=suffix) |
| 76 | else: |
| 77 | result = tempfile.NamedTemporaryFile() |
| 78 | yield (video.name, result.name) |
| 79 | |
| 80 | |
| 81 | #TODO: this should be skipped if video plugin is not enabled |
| 82 | def test_thumbnails(): |
| 83 | ''' |
| 84 | Test thumbnails generation. |
| 85 | 1. Create a video (+audio) from gst's videotestsrc |
| 86 | 2. Capture thumbnail |
| 87 | 3. Everything should get removed because of temp files usage |
| 88 | ''' |
| 89 | #data create_data() as (video_name, thumbnail_name): |
| 90 | test_formats = [('.png', 'png'), ('.jpg', 'jpeg'), ('.gif', 'gif')] |
| 91 | for suffix, format in test_formats: |
| 92 | with create_data(suffix) as (video_name, thumbnail_name): |
| 93 | capture_thumb(video_name, thumbnail_name, width=40) |
| 94 | # check result file format |
| 95 | assert imghdr.what(thumbnail_name) == format |
| 96 | # TODO: check height and width |
| 97 | # FIXME: it doesn't work with small width, say, 10px. This should be |
| 98 | # fixed somehow |
| 99 | suffix, format = test_formats[0] |
| 100 | with create_data(suffix, True) as (video_name, thumbnail_name): |
| 101 | capture_thumb(video_name, thumbnail_name, width=40) |
| 102 | assert imghdr.what(thumbnail_name) == format |
| 103 | with create_data(suffix, True) as (video_name, thumbnail_name): |
| 104 | capture_thumb(video_name, thumbnail_name, width=10) # smaller width |
| 105 | assert imghdr.what(thumbnail_name) == format |
| 106 | with create_data(suffix, True) as (video_name, thumbnail_name): |
| 107 | capture_thumb(video_name, thumbnail_name, width=100) # bigger width |
| 108 | assert imghdr.what(thumbnail_name) == format |
| 109 | |
| 110 | |
| 111 | def test_transcoder(): |
| 112 | # test without audio |
| 113 | with create_data() as (video_name, result_name): |
| 114 | transcoder = VideoTranscoder() |
| 115 | transcoder.transcode( |
| 116 | video_name, result_name, |
| 117 | vp8_quality=8, |
| 118 | vp8_threads=0, # autodetect |
| 119 | vorbis_quality=0.3, |
| 120 | dimensions=(640, 640)) |
| 121 | assert len(discover(result_name).get_video_streams()) == 1 |
| 122 | # test with audio |
| 123 | with create_data(make_audio=True) as (video_name, result_name): |
| 124 | transcoder = VideoTranscoder() |
| 125 | transcoder.transcode( |
| 126 | video_name, result_name, |
| 127 | vp8_quality=8, |
| 128 | vp8_threads=0, # autodetect |
| 129 | vorbis_quality=0.3, |
| 130 | dimensions=(640, 640)) |
| 131 | assert len(discover(result_name).get_video_streams()) == 1 |
| 132 | assert len(discover(result_name).get_audio_streams()) == 1 |