72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
import pytest
|
|
from conftest import assert_logs
|
|
|
|
from ytdl_sub.plugins.throttle_protection import logger as throttle_protection_logger
|
|
from ytdl_sub.subscriptions.subscription import Subscription
|
|
|
|
|
|
@pytest.fixture
|
|
def preset_dict_max_downloads_0(output_directory):
|
|
return {
|
|
"preset": "Jellyfin Music Videos",
|
|
"download": "https://youtube.com/watch?v=HKTNxEqsN3Q",
|
|
"format": "worst[ext=mp4]",
|
|
"overrides": {
|
|
"music_video_artist": "JMC",
|
|
"music_video_directory": output_directory,
|
|
},
|
|
"throttle_protection": {"max_downloads_per_subscription": {"max": 0}},
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def preset_dict_subscription_download_proba_0(output_directory):
|
|
return {
|
|
"preset": "Jellyfin Music Videos",
|
|
"download": "https://youtube.com/watch?v=HKTNxEqsN3Q",
|
|
"format": "worst[ext=mp4]",
|
|
"overrides": {
|
|
"music_video_artist": "JMC",
|
|
"music_video_directory": output_directory,
|
|
},
|
|
"throttle_protection": {"subscription_download_probability": 0.0},
|
|
}
|
|
|
|
|
|
class TestThrottleProtection:
|
|
def test_max_downloads(
|
|
self,
|
|
default_config,
|
|
preset_dict_max_downloads_0,
|
|
output_directory,
|
|
):
|
|
single_video_subscription = Subscription.from_dict(
|
|
config=default_config,
|
|
preset_name="music_video_single_video_test",
|
|
preset_dict=preset_dict_max_downloads_0,
|
|
)
|
|
|
|
with assert_logs(
|
|
logger=throttle_protection_logger,
|
|
expected_message="Reached subscription max downloads of %d",
|
|
log_level="info",
|
|
expected_occurrences=1,
|
|
):
|
|
transaction_log = single_video_subscription.download(dry_run=True)
|
|
|
|
assert transaction_log.is_empty
|
|
|
|
def test_subscription_probability(
|
|
self,
|
|
default_config,
|
|
preset_dict_subscription_download_proba_0,
|
|
output_directory,
|
|
):
|
|
single_video_subscription = Subscription.from_dict(
|
|
config=default_config,
|
|
preset_name="music_video_single_video_test",
|
|
preset_dict=preset_dict_subscription_download_proba_0,
|
|
)
|
|
|
|
transaction_log = single_video_subscription.download(dry_run=True)
|
|
assert transaction_log.is_empty
|