Subdl Provider: add tests and use standard utils

1 files changed, 36 insertions, 0 deletions

diff --git a/tests/subliminal_patch/test_subdl.py b/tests/subliminal_patch/test_subdl.py
new file mode 100644
index 000000000..47fe7577d
--- /dev/null
+++ b/tests/subliminal_patch/test_subdl.py
@@ -0,0 +1,36 @@
+import os
+
+import pytest
+from subliminal_patch.providers.subdl import SubdlProvider
+from subliminal_patch.providers.subdl import SubdlSubtitle
+
+
[email protected](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()