summaryrefslogtreecommitdiffhomepage
path: root/tests/subliminal_patch/test_subdl.py
blob: 47fe7577d9c17c5a1dfc13966407bb743095107f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
import os

import pytest
from subliminal_patch.providers.subdl import SubdlProvider
from subliminal_patch.providers.subdl import SubdlSubtitle


@pytest.fixture(scope="session")
def provider():
    with SubdlProvider(os.environ["SUBDL_TOKEN"]) as provider:
        yield provider


def test_list_subtitles_movie(provider, movies, languages):
    for sub in provider.list_subtitles(movies["dune"], {languages["en"]}):
        assert sub.language == languages["en"]


def test_download_subtitle(provider, languages):
    data = {
        "language": languages["en"],
        "forced": False,
        "hearing_impaired": False,
        "page_link": "https://subdl.com/s/info/ebC6BrLCOC",
        "download_link": "/subtitle/2808552-2770424.zip",
        "file_id": "SUBDL::dune-2021-2770424.zip",
        "release_names": ["Dune Part 1 WebDl"],
        "uploader": "makoto77",
        "season": 0,
        "episode": None,
    }

    sub = SubdlSubtitle(**data)
    provider.download_subtitle(sub)

    assert sub.is_valid()