danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix tests

Browse source
This commit is contained in:
Jesse Bannon 2025-07-02 23:34:22 -07:00
parent 0a2d6e274f
commit 7c940f98c6
2 changed files with 11 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/integration/conftest.py
View file

@ -265,11 +265,6 @@ def mock_download_collection_entries(
patch.object( patch.object(
MultiUrlDownloader, "_extract_entry_info_with_retry", new=lambda _, entry: entry MultiUrlDownloader, "_extract_entry_info_with_retry", new=lambda _, entry: entry
), ),
# Throttle protection is included in all prebuilt presets. Mock the sleep avoid
# actual sleeps
patch.object(
ThrottleProtectionPlugin, "perform_sleep", new=lambda _1, _2: None
)
): ):
# Stub out metadata. TODO: update this if we do metadata plugins # Stub out metadata. TODO: update this if we do metadata plugins
yield yield

15
tests/integration/plugins/test_throttle_protection.py
View file

@ -1,8 +1,10 @@
from typing import Dict from typing import Dict
from unittest.mock import patch
import pytest import pytest
from conftest import assert_logs from conftest import assert_logs
from ytdl_sub.plugins.throttle_protection import ThrottleProtectionPlugin
from ytdl_sub.plugins.throttle_protection import logger as throttle_protection_logger from ytdl_sub.plugins.throttle_protection import logger as throttle_protection_logger
from ytdl_sub.subscriptions.subscription import Subscription from ytdl_sub.subscriptions.subscription import Subscription
@ -49,13 +51,18 @@ class TestThrottleProtectionPlugin:
) )
with ( with (
patch.object(
ThrottleProtectionPlugin,
"perform_sleep",
new=ThrottleProtectionPlugin.perform_sleep,
),
mock_download_collection_entries( mock_download_collection_entries(
is_youtube_channel=False, num_urls=1, is_extracted_audio=False is_youtube_channel=False, num_urls=1, is_extracted_audio=False
), ),
assert_logs( assert_logs(
logger=throttle_protection_logger, logger=throttle_protection_logger,
expected_message="Sleeping between downloads for %0.2f seconds", expected_message="Sleeping between downloads for %0.2f seconds",
log_level="debug", log_level="info",
expected_occurrences=4, expected_occurrences=4,
), ),
): ):
@ -68,7 +75,7 @@ class TestThrottleProtectionPlugin:
assert_logs( assert_logs(
logger=throttle_protection_logger, logger=throttle_protection_logger,
expected_message="Sleeping between subscriptions for %0.2f seconds", expected_message="Sleeping between subscriptions for %0.2f seconds",
log_level="debug", log_level="info",
expected_occurrences=1, expected_occurrences=1,
), ),
): ):
@ -92,7 +99,7 @@ class TestThrottleProtectionPlugin:
mock_download_collection_entries, mock_download_collection_entries,
disable_value, disable_value,
): ):
throttle_subscription_dict["throttle_protection"]["enable"] = disable_value throttle_subscription_dict["overrides"]["enable_throttle_protection"] = disable_value
subscription = Subscription.from_dict( subscription = Subscription.from_dict(
config=config, config=config,
preset_name=subscription_name, preset_name=subscription_name,
@ -106,7 +113,7 @@ class TestThrottleProtectionPlugin:
assert_logs( assert_logs(
logger=throttle_protection_logger, logger=throttle_protection_logger,
expected_message="Sleeping between downloads for %0.2f seconds", expected_message="Sleeping between downloads for %0.2f seconds",
log_level="debug", log_level="info",
expected_occurrences=0, expected_occurrences=0,
), ),
): ):