140 lines
4.6 KiB
Python
140 lines
4.6 KiB
Python
from typing import Dict
|
|
from typing import List
|
|
|
|
import pytest
|
|
|
|
from ytdl_sub.config.config_file import ConfigFile
|
|
from ytdl_sub.config.preset import Preset
|
|
from ytdl_sub.prebuilt_presets.tv_show import PrebuiltJellyfinTVShowPresets
|
|
from ytdl_sub.prebuilt_presets.tv_show import PrebuiltKodiTVShowPresets
|
|
from ytdl_sub.subscriptions.subscription import Subscription
|
|
|
|
|
|
@pytest.fixture
|
|
def config(working_directory) -> ConfigFile:
|
|
return ConfigFile(
|
|
name="config",
|
|
value={"configuration": {"working_directory": working_directory}, "presets": {}},
|
|
)
|
|
|
|
|
|
class TestPrebuiltTVShowPresets:
|
|
@pytest.mark.parametrize(
|
|
"media_player_preset",
|
|
[
|
|
*PrebuiltKodiTVShowPresets.get_non_collection_preset_names(),
|
|
*PrebuiltJellyfinTVShowPresets.get_non_collection_preset_names(),
|
|
],
|
|
)
|
|
@pytest.mark.parametrize(
|
|
"tv_show_structure_preset",
|
|
[
|
|
"season_by_year_month__episode_by_day",
|
|
"season_by_year__episode_by_month_day",
|
|
"season_by_year__episode_by_month_day_reversed",
|
|
],
|
|
)
|
|
@pytest.mark.parametrize(
|
|
"episode_hour_granularity_preset",
|
|
[
|
|
None,
|
|
"episode_add_hour_granularity",
|
|
"episode_add_hour_granularity_reversed",
|
|
],
|
|
)
|
|
def test_non_collection_presets_compile(
|
|
self,
|
|
config: ConfigFile,
|
|
media_player_preset: str,
|
|
tv_show_structure_preset: str,
|
|
episode_hour_granularity_preset: str,
|
|
):
|
|
parent_presets = [media_player_preset, tv_show_structure_preset]
|
|
if episode_hour_granularity_preset:
|
|
parent_presets.append(episode_hour_granularity_preset)
|
|
|
|
preset = Preset.from_dict(
|
|
config=config,
|
|
preset_name=f"{media_player_preset}_test",
|
|
preset_dict={
|
|
"preset": parent_presets,
|
|
"overrides": {
|
|
"url": "https://your.name.here",
|
|
"tv_show_name": "test tv show",
|
|
"tv_show_directory": "output_path",
|
|
},
|
|
},
|
|
)
|
|
assert preset
|
|
|
|
@pytest.mark.parametrize(
|
|
"media_player_preset",
|
|
[
|
|
*PrebuiltKodiTVShowPresets.get_collection_preset_names(),
|
|
*PrebuiltJellyfinTVShowPresets.get_collection_preset_names(),
|
|
],
|
|
)
|
|
@pytest.mark.parametrize(
|
|
"tv_show_structure_preset",
|
|
[
|
|
"season_by_collection__episode_by_year_month_day",
|
|
"season_by_collection__episode_by_year_month_day_reversed",
|
|
],
|
|
)
|
|
@pytest.mark.parametrize(
|
|
"episode_hour_granularity_preset",
|
|
[
|
|
None,
|
|
"episode_add_hour_granularity",
|
|
"episode_add_hour_granularity_reversed",
|
|
],
|
|
)
|
|
@pytest.mark.parametrize("season_indices", [[1], [1, 2]])
|
|
@pytest.mark.parametrize("is_season_1_youtube_channel", [True, False])
|
|
def test_collection_presets_compile(
|
|
self,
|
|
config,
|
|
subscription_name,
|
|
output_directory,
|
|
mock_download_collection_entries,
|
|
media_player_preset: str,
|
|
tv_show_structure_preset: str,
|
|
episode_hour_granularity_preset: str,
|
|
season_indices: List[int],
|
|
is_season_1_youtube_channel: bool,
|
|
):
|
|
parent_presets: List[str] = [media_player_preset, tv_show_structure_preset]
|
|
if episode_hour_granularity_preset:
|
|
parent_presets.append(episode_hour_granularity_preset)
|
|
|
|
overrides: Dict[str, str] = {}
|
|
for season_index in season_indices:
|
|
parent_presets.append(f"tv_show_collection_season_{season_index}")
|
|
if season_index == 1 and is_season_1_youtube_channel:
|
|
parent_presets[-1] += "_youtube_channel"
|
|
|
|
overrides = dict(
|
|
overrides,
|
|
**{
|
|
f"collection_season_{season_index}_name": f"Season {season_index}",
|
|
f"collection_season_{season_index}_url": f"https://season.{season_index}.com",
|
|
},
|
|
)
|
|
|
|
subscription = Subscription.from_dict(
|
|
config=config,
|
|
preset_name=subscription_name,
|
|
preset_dict={
|
|
"preset": parent_presets,
|
|
"overrides": dict(
|
|
overrides,
|
|
**{
|
|
"tv_show_name": "Test TV Show",
|
|
"tv_show_directory": output_directory,
|
|
},
|
|
),
|
|
},
|
|
)
|
|
|
|
output_transactoin = subscription.download(dry_run=False)
|
|
assert output_transactoin
|