tests

2025-06-18 22:12:40 -07:00 · 2025-06-18 22:12:40 -07:00 · 907158cd95
commit 907158cd95
parent 701358b880
7 changed files with 283 additions and 10 deletions
--- a/src/ytdl_sub/plugins/embed_thumbnail.py
+++ b/src/ytdl_sub/plugins/embed_thumbnail.py
@ -11,12 +11,13 @@ from ytdl_sub.utils.file_handler import FileHandler
 from ytdl_sub.utils.file_handler import FileMetadata
 from ytdl_sub.utils.logger import Logger
 from ytdl_sub.validators.audo_codec_validator import AUDIO_CODEC_EXTS
+from ytdl_sub.validators.string_formatter_validators import OverridesBooleanFormatterValidator
 from ytdl_sub.validators.validators import BoolValidator

 logger = Logger.get("embed-thumbnail")


-class EmbedThumbnailOptions(BoolValidator, OptionsValidator):
+class EmbedThumbnailOptions(OverridesBooleanFormatterValidator, OptionsValidator):
    """
    Whether to embed thumbnails to the audio/video file or not.

@ -33,7 +34,7 @@ class EmbedThumbnailPlugin(Plugin[EmbedThumbnailOptions]):

    @property
    def _embed_thumbnail(self) -> bool:
-        return self.plugin_options.value
+        return self.overrides.evaluate_boolean(self.plugin_options)

    @classmethod
    def _embed_video_thumbnail(cls, entry: Entry) -> None:
--- a/src/ytdl_sub/plugins/square_thumbnail.py
+++ b/src/ytdl_sub/plugins/square_thumbnail.py
@ -41,14 +41,14 @@ class SquareThumbnailPlugin(Plugin[SquareThumbnailOptions]):
            ffmpeg_args: List[str] = [
                "-i",
                thumbnail_path,
-                "-c:v"
-                "mjpeg"
-                "-qmin"
-                "1"
-                "-qscale:v"
-                "1"
-                "-vf"
-                "crop=min(iw\\,ih):min(iw\\,ih)"
+                "-c:v",
+                "mjpeg",
+                "-qmin",
+                "1",
+                "-qscale:v",
+                "1",
+                "-vf",
+                "crop=min(iw\\,ih):min(iw\\,ih)",
                "-bitexact",  # for reproducibility
                tmp_file_path,
            ]
--- a/tests/integration/plugins/test_thumbnail_plugins.py
+++ b/tests/integration/plugins/test_thumbnail_plugins.py
@ -0,0 +1,68 @@
+from typing import Dict
+
+import pytest
+from expected_download import assert_expected_downloads
+from expected_transaction_log import assert_transaction_log_matches
+
+from ytdl_sub.subscriptions.subscription import Subscription
+
+
+@pytest.fixture
+def square_thumbnail_subscription_dict(output_directory) -> Dict:
+    return {
+        "preset": "Jellyfin Music Videos",
+        "output_options": {"output_directory": output_directory},
+        "square_thumbnail": "{perform_square_thumbnail}",
+        "embed_thumbnail": "{perform_embed_thumbnail}",
+        "overrides": {
+            "music_video_artist": "JMC",
+            "url": "https://your.name.here",
+            "perform_square_thumbnail": True,
+            "perform_embed_thumbnail": False
+        },
+    }
+
+
+class TestThumbnailPlugins:
+    @pytest.mark.parametrize("dry_run", [True, False])
+    @pytest.mark.parametrize("embed_thumbnail", [True, False])
+    def test_thumbnail(
+        self,
+        config,
+        subscription_name,
+        square_thumbnail_subscription_dict,
+        output_directory,
+        mock_download_collection_entries,
+        dry_run,
+        embed_thumbnail,
+    ):
+        square_thumbnail_subscription_dict["overrides"]["perform_embed_thumbnail"] = embed_thumbnail
+
+        subscription = Subscription.from_dict(
+            config=config,
+            preset_name=subscription_name,
+            preset_dict=square_thumbnail_subscription_dict,
+        )
+
+        with mock_download_collection_entries(
+            is_youtube_channel=False,
+            num_urls=1,
+            is_extracted_audio=False,
+            is_dry_run=dry_run,
+        ):
+            transaction_log = subscription.download(dry_run=dry_run)
+
+        expected_filename = "square_thumbnail"
+        if embed_thumbnail:
+            expected_filename = "embedded_square_thumbnail"
+
+        assert_transaction_log_matches(
+            output_directory=output_directory,
+            transaction_log=transaction_log,
+            transaction_log_summary_file_name=f"plugins/thumbnail/{expected_filename}.txt",
+        )
+        assert_expected_downloads(
+            output_directory=output_directory,
+            dry_run=dry_run,
+            expected_download_summary_file_name=f"plugins/thumbnail/{expected_filename}.json",
+        )
--- a/tests/resources/expected_downloads_summaries/plugins/thumbnail/embedded_square_thumbnail.json
+++ b/tests/resources/expected_downloads_summaries/plugins/thumbnail/embedded_square_thumbnail.json
@ -0,0 +1,19 @@
+{
+  ".ytdl-sub-subscription_test-download-archive.json": "76e202bd03ceef93daaffffee2cfa193",
+  "JMC/Mock Entry 20-1.info.json": "INFO_JSON",
+  "JMC/Mock Entry 20-1.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 20-1.mp4": "071b3d0747705613c0ec0cba71a045fa",
+  "JMC/Mock Entry 20-1.nfo": "fefcf0b3e4f4ff80ad636584d50dadec",
+  "JMC/Mock Entry 20-2.info.json": "INFO_JSON",
+  "JMC/Mock Entry 20-2.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 20-2.mp4": "fce852b2c7567b917a7d02b1216215a0",
+  "JMC/Mock Entry 20-2.nfo": "025c0b631da5ff5470382b38fce78d2d",
+  "JMC/Mock Entry 20-3.info.json": "INFO_JSON",
+  "JMC/Mock Entry 20-3.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 20-3.mp4": "b2edb14a985b95fb4c530f7e868ed66c",
+  "JMC/Mock Entry 20-3.nfo": "618b0ff948d9de2e10cf1da8c0dd6615",
+  "JMC/Mock Entry 21-1.info.json": "INFO_JSON",
+  "JMC/Mock Entry 21-1.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 21-1.mp4": "3b50ad36e92501fef9d18dcd7532a921",
+  "JMC/Mock Entry 21-1.nfo": "e5c715749efc1603a6e2f59244d87aba"
+}
--- a/tests/resources/expected_downloads_summaries/plugins/thumbnail/square_thumbnail.json
+++ b/tests/resources/expected_downloads_summaries/plugins/thumbnail/square_thumbnail.json
@ -0,0 +1,19 @@
+{
+  ".ytdl-sub-subscription_test-download-archive.json": "76e202bd03ceef93daaffffee2cfa193",
+  "JMC/Mock Entry 20-1.info.json": "INFO_JSON",
+  "JMC/Mock Entry 20-1.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 20-1.mp4": "d8dc9918d1646c92b4a4bd246291d849",
+  "JMC/Mock Entry 20-1.nfo": "fefcf0b3e4f4ff80ad636584d50dadec",
+  "JMC/Mock Entry 20-2.info.json": "INFO_JSON",
+  "JMC/Mock Entry 20-2.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 20-2.mp4": "6f925179a5b6bf7ffcaf2f70ad585296",
+  "JMC/Mock Entry 20-2.nfo": "025c0b631da5ff5470382b38fce78d2d",
+  "JMC/Mock Entry 20-3.info.json": "INFO_JSON",
+  "JMC/Mock Entry 20-3.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 20-3.mp4": "b48f1803f14be665677e9ac9da787593",
+  "JMC/Mock Entry 20-3.nfo": "618b0ff948d9de2e10cf1da8c0dd6615",
+  "JMC/Mock Entry 21-1.info.json": "INFO_JSON",
+  "JMC/Mock Entry 21-1.jpg": "e691c0591065c89e21fd552672a206af",
+  "JMC/Mock Entry 21-1.mp4": "5fb865f27ee1cac381597d1e182dfa48",
+  "JMC/Mock Entry 21-1.nfo": "e5c715749efc1603a6e2f59244d87aba"
+}
--- a/tests/resources/transaction_log_summaries/plugins/thumbnail/embedded_square_thumbnail.txt
+++ b/tests/resources/transaction_log_summaries/plugins/thumbnail/embedded_square_thumbnail.txt
@ -0,0 +1,85 @@
+Files created:
+----------------------------------------
+{output_directory}
+  .ytdl-sub-subscription_test-download-archive.json
+{output_directory}/JMC
+  Mock Entry 20-1.info.json
+  Mock Entry 20-1.jpg
+  Mock Entry 20-1.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2020-08-08
+      title: Mock Entry 20-1
+      year: 2020
+    Embedded thumbnail
+    Square thumbnail
+  Mock Entry 20-1.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2020-08-08
+        title: Mock Entry 20-1
+  Mock Entry 20-2.info.json
+  Mock Entry 20-2.jpg
+  Mock Entry 20-2.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2020-08-08
+      title: Mock Entry 20-2
+      year: 2020
+    Embedded thumbnail
+    Square thumbnail
+  Mock Entry 20-2.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2020-08-08
+        title: Mock Entry 20-2
+  Mock Entry 20-3.info.json
+  Mock Entry 20-3.jpg
+  Mock Entry 20-3.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2020-08-07
+      title: Mock Entry 20-3
+      year: 2020
+    Embedded thumbnail
+    Square thumbnail
+  Mock Entry 20-3.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2020-08-07
+        title: Mock Entry 20-3
+  Mock Entry 21-1.info.json
+  Mock Entry 21-1.jpg
+  Mock Entry 21-1.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2021-08-08
+      title: Mock Entry 21-1
+      year: 2021
+    Embedded thumbnail
+    Square thumbnail
+  Mock Entry 21-1.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2021-08-08
+        title: Mock Entry 21-1
--- a/tests/resources/transaction_log_summaries/plugins/thumbnail/square_thumbnail.txt
+++ b/tests/resources/transaction_log_summaries/plugins/thumbnail/square_thumbnail.txt
@ -0,0 +1,81 @@
+Files created:
+----------------------------------------
+{output_directory}
+  .ytdl-sub-subscription_test-download-archive.json
+{output_directory}/JMC
+  Mock Entry 20-1.info.json
+  Mock Entry 20-1.jpg
+  Mock Entry 20-1.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2020-08-08
+      title: Mock Entry 20-1
+      year: 2020
+    Square thumbnail
+  Mock Entry 20-1.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2020-08-08
+        title: Mock Entry 20-1
+  Mock Entry 20-2.info.json
+  Mock Entry 20-2.jpg
+  Mock Entry 20-2.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2020-08-08
+      title: Mock Entry 20-2
+      year: 2020
+    Square thumbnail
+  Mock Entry 20-2.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2020-08-08
+        title: Mock Entry 20-2
+  Mock Entry 20-3.info.json
+  Mock Entry 20-3.jpg
+  Mock Entry 20-3.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2020-08-07
+      title: Mock Entry 20-3
+      year: 2020
+    Square thumbnail
+  Mock Entry 20-3.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2020-08-07
+        title: Mock Entry 20-3
+  Mock Entry 21-1.info.json
+  Mock Entry 21-1.jpg
+  Mock Entry 21-1.mp4
+    Video Tags:
+      album: Music Videos
+      artist: JMC
+      genre: ytdl-sub
+      premiered: 2021-08-08
+      title: Mock Entry 21-1
+      year: 2021
+    Square thumbnail
+  Mock Entry 21-1.nfo
+    NFO tags:
+      musicvideo:
+        album: Music Videos
+        artist: JMC
+        genre: ytdl-sub
+        premiered: 2021-08-08
+        title: Mock Entry 21-1