Allow assrt to fetch subtitles through only ISO 639-2

This commit should fix issue #460

1 files changed, 117 insertions, 0 deletions

diff --git a/tests/conftest.py b/tests/conftest.py
new file mode 100644
index 000000000..3666e501c
--- /dev/null
+++ b/tests/conftest.py
@@ -0,0 +1,117 @@
+# -*- coding: utf-8 -*-
+import libs
+from io import BytesIO
+import os
+from zipfile import ZipFile
+
+import pytest
+import requests
+try:
+    from unittest.mock import Mock
+except ImportError:
+    from mock import Mock
+
+from subliminal import Episode, Movie
+from subliminal.cache import region
+
[email protected](autouse=True, scope='session')
+def configure_region():
+    region.configure('dogpile.cache.null')
+    region.configure = Mock()
+
+
[email protected]
+def movies():
+    return {'man_of_steel':
+            Movie(os.path.join('Man of Steel (2013)', 'man.of.steel.2013.720p.bluray.x264-felony.mkv'), 'Man of Steel',
+                  format='BluRay', release_group='felony', resolution='720p', video_codec='h264', audio_codec='DTS',
+                  imdb_id='tt0770828', size=7033732714, year=2013,
+                  hashes={'napiprojekt': '6303e7ee6a835e9fcede9fb2fb00cb36',
+                          'opensubtitles': '5b8f8f4e41ccb21e',
+                          'shooter': '314f454ab464775498ae6f1f5ad813a9;fdaa8b702d8936feba2122e93ba5c44f;'
+                                     '0a6935e3436aa7db5597ef67a2c494e3;4d269733f36ddd49f71e92732a462fe5',
+                          'thesubdb': 'ad32876133355929d814457537e12dc2'}),
+            'enders_game':
+            Movie('enders.game.2013.720p.bluray.x264-sparks.mkv', 'Ender\'s Game',
+                  format='BluRay', release_group='sparks', resolution='720p', video_codec='h264', year=2013),
+            'interstellar':
+            Movie('Interstellar.2014.2014.1080p.BluRay.x264.YIFY.rar', 'Interstellar',
+                  format='BluRay', release_group='YIFY', resolution='1080p', video_codec='h264', year=2014)}
+
+
[email protected]
+def episodes():
+    return {'bbt_s07e05':
+            Episode(os.path.join('The Big Bang Theory', 'Season 07',
+                                 'The.Big.Bang.Theory.S07E05.720p.HDTV.X264-DIMENSION.mkv'),
+                    'The Big Bang Theory', 7, 5, title='The Workplace Proximity', year=2007, tvdb_id=4668379,
+                    series_tvdb_id=80379, series_imdb_id='tt0898266', format='HDTV', release_group='DIMENSION',
+                    resolution='720p', video_codec='h264', audio_codec='AC3', imdb_id='tt3229392', size=501910737,
+                    hashes={'napiprojekt': '6303e7ee6a835e9fcede9fb2fb00cb36',
+                            'opensubtitles': '6878b3ef7c1bd19e',
+                            'shooter': 'c13e0e5243c56d280064d344676fff94;cd4184d1c0c623735f6db90841ce15fc;'
+                                       '3faefd72f92b63f2504269b4f484a377;8c68d1ef873afb8ba0cc9f97cbac41c1',
+                            'thesubdb': '9dbbfb7ba81c9a6237237dae8589fccc'}),
+            'got_s03e10':
+            Episode(os.path.join('Game of Thrones', 'Season 03',
+                                 'Game.of.Thrones.S03E10.Mhysa.720p.WEB-DL.DD5.1.H.264-NTb.mkv'),
+                    'Game of Thrones', 3, 10, title='Mhysa', tvdb_id=4517466, series_tvdb_id=121361,
+                    series_imdb_id='tt0944947', format='WEB-DL', release_group='NTb', resolution='720p',
+                    video_codec='h264', audio_codec='AC3', imdb_id='tt2178796', size=2142810931,
+                    hashes={'napiprojekt': '6303e7ee6a835e9fcede9fb2fb00cb36',
+                            'opensubtitles': 'b850baa096976c22',
+                            'shooter': 'b02d992c04ad74b31c252bd5a097a036;ef1b32f873b2acf8f166fc266bdf011a;'
+                                       '82ce34a3bcee0c66ed3b26d900d31cca;78113770551f3efd1e2d4ec45898c59c',
+                            'thesubdb': 'b1f899c77f4c960b84b8dbf840d4e42d'}),
+            'dallas_s01e03':
+            Episode('Dallas.S01E03.mkv', 'Dallas', 1, 3, title='Spy in the House', year=1978, tvdb_id=228224,
+                    series_tvdb_id=77092, series_imdb_id='tt0077000'),
+            'dallas_2012_s01e03':
+            Episode('Dallas.2012.S01E03.mkv', 'Dallas', 1, 3, title='The Price You Pay', year=2012,
+                    original_series=False, tvdb_id=4199511, series_tvdb_id=242521, series_imdb_id='tt1723760',
+                    imdb_id='tt2205526'),
+            'marvels_agents_of_shield_s02e06':
+            Episode('Marvels.Agents.of.S.H.I.E.L.D.S02E06.720p.HDTV.x264-KILLERS.mkv',
+                    'Marvel\'s Agents of S.H.I.E.L.D.', 2, 6, year=2013, format='HDTV', release_group='KILLERS',
+                    resolution='720p', video_codec='h264'),
+            'csi_cyber_s02e03':
+            Episode('CSI.Cyber.S02E03.hdtv-lol.mp4', 'CSI: Cyber', 2, 3, format='HDTV', release_group='lol'),
+            'the_x_files_s10e02':
+            Episode('The.X-Files.S10E02.HDTV.x264-KILLERS.mp4', 'The X-Files', 10, 2, format='HDTV',
+                    release_group='KILLERS', video_codec='h264'),
+            'colony_s01e09':
+            Episode('Colony.S01E09.720p.HDTV.x264-KILLERS.mkv', 'Colony', 1, 9, title='Zero Day', year=2016,
+                    tvdb_id=5463229, series_tvdb_id=284210, series_imdb_id='tt4209256', format='HDTV',
+                    release_group='KILLERS', resolution='720p', video_codec='h264', imdb_id='tt4926022'),
+            'the_jinx_e05':
+            Episode('The.Jinx-The.Life.and.Deaths.of.Robert.Durst.E05.BDRip.x264-ROVERS.mkv',
+                    'The Jinx: The Life and Deaths of Robert Durst', 1, 5, year=2015, original_series=True,
+                    format='BluRay', release_group='ROVERS', video_codec='h264'),
+            'the_100_s03e09':
+            Episode('The.100.S03E09.720p.HDTV.x264-AVS.mkv', 'The 100', 3, 9, title='Stealing Fire', year=2014,
+                    tvdb_id=5544536, series_tvdb_id=268592, series_imdb_id='tt2661044', format='HDTV',
+                    release_group='AVS', resolution='720p', video_codec='h264', imdb_id='tt4799896'),
+            'csi_s15e18':
+            Episode('CSI.S15E18.720p.HDTV.X264.DIMENSION.mkv', 'CSI: Crime Scene Investigation', 15, 18,
+                    title='The End Game', year=2000, tvdb_id=5104359, series_tvdb_id=72546, series_imdb_id='tt0247082',
+                    format='HDTV', release_group='DIMENSION', resolution='720p', video_codec='h264',
+                    imdb_id='tt4145952')}
+
+
[email protected](scope='session')
+def mkv():
+    data_path = os.path.join('tests', 'data', 'mkv')
+
+    # download matroska test suite
+    if not os.path.exists(data_path) or len(os.listdir(data_path)) != 8:
+        r = requests.get('http://downloads.sourceforge.net/project/matroska/test_files/matroska_test_w1_1.zip')
+        with ZipFile(BytesIO(r.content), 'r') as f:
+            f.extractall(data_path, [m for m in f.namelist() if os.path.splitext(m)[1] == '.mkv'])
+
+    # populate a dict with mkv files
+    files = {}
+    for path in os.listdir(data_path):
+        name, _ = os.path.splitext(path)
+        files[name] = os.path.join(data_path, path)
+
+    return files