142 lines
4.8 KiB
Python
142 lines
4.8 KiB
Python
import os
|
|
import tempfile
|
|
from pathlib import Path
|
|
from typing import Callable
|
|
from typing import Dict
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from ytdl_sub.config.config_file import ConfigFile
|
|
from ytdl_sub.downloaders.downloader import Downloader
|
|
from ytdl_sub.entries.variables.kwargs import EPOCH
|
|
from ytdl_sub.entries.variables.kwargs import EXT
|
|
from ytdl_sub.entries.variables.kwargs import EXTRACTOR
|
|
from ytdl_sub.entries.variables.kwargs import PLAYLIST_COUNT
|
|
from ytdl_sub.entries.variables.kwargs import PLAYLIST_ENTRY
|
|
from ytdl_sub.entries.variables.kwargs import PLAYLIST_INDEX
|
|
from ytdl_sub.entries.variables.kwargs import TITLE
|
|
from ytdl_sub.entries.variables.kwargs import UID
|
|
from ytdl_sub.entries.variables.kwargs import UPLOAD_DATE
|
|
from ytdl_sub.entries.variables.kwargs import WEBPAGE_URL
|
|
|
|
|
|
@pytest.fixture
|
|
def working_directory() -> str:
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
yield temp_dir
|
|
|
|
|
|
@pytest.fixture
|
|
def subscription_name(working_directory) -> str:
|
|
name = "subscription_test"
|
|
os.makedirs(Path(working_directory) / name, exist_ok=True)
|
|
return name
|
|
|
|
|
|
@pytest.fixture
|
|
def output_directory() -> str:
|
|
with tempfile.TemporaryDirectory() as temp_dir:
|
|
yield temp_dir
|
|
|
|
|
|
@pytest.fixture
|
|
def config(working_directory) -> ConfigFile:
|
|
return ConfigFile(
|
|
name="config",
|
|
value={"configuration": {"working_directory": working_directory}, "presets": {}},
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_file_factory(working_directory: str, subscription_name: str):
|
|
def _mock_file_factory(file_name: str):
|
|
with open(Path(working_directory) / subscription_name / file_name, "w"):
|
|
pass
|
|
|
|
return _mock_file_factory
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_entry_dict_factory(mock_file_factory) -> Callable:
|
|
def _mock_entry_dict_factory(
|
|
uid: int, upload_date: str, playlist_index: int = 1, playlist_count: int = 1
|
|
) -> Dict:
|
|
entry_dict = {
|
|
UID: uid,
|
|
EPOCH: 1596878400,
|
|
PLAYLIST_INDEX: playlist_index,
|
|
PLAYLIST_COUNT: playlist_count,
|
|
EXTRACTOR: "mock-entry-dict",
|
|
TITLE: f"Mock Entry {uid}",
|
|
EXT: "mp4",
|
|
UPLOAD_DATE: upload_date,
|
|
WEBPAGE_URL: f"https://{uid}.com",
|
|
PLAYLIST_ENTRY: {
|
|
"thumbnails": [
|
|
{
|
|
"id": "avatar_uncropped",
|
|
"url": "https://avatar_uncropped.com",
|
|
},
|
|
{
|
|
"id": "banner_uncropped",
|
|
"url": "https://banner_uncropped.com",
|
|
},
|
|
]
|
|
},
|
|
}
|
|
|
|
# Create mock video file
|
|
mock_file_factory(file_name=f"{uid}.mp4")
|
|
mock_file_factory(file_name=f"{uid}.jpg")
|
|
return entry_dict
|
|
|
|
return _mock_entry_dict_factory
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_download_collection_thumbnail(mock_file_factory):
|
|
def _mock_download_thumbnail(output_path: str) -> bool:
|
|
mock_file_factory(file_name=output_path.split("/")[-1])
|
|
return True
|
|
|
|
with patch.object(
|
|
Downloader,
|
|
"_download_thumbnail",
|
|
new=lambda _, thumbnail_url, output_thumbnail_path: _mock_download_thumbnail(
|
|
output_thumbnail_path
|
|
),
|
|
):
|
|
yield # TODO: create file here
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_download_collection_entries(
|
|
mock_download_collection_thumbnail, mock_entry_dict_factory: Callable, working_directory: str
|
|
):
|
|
def _(**kwargs):
|
|
return mock_entry_dict_factory(**kwargs)
|
|
|
|
collection_1_entry_dicts = [
|
|
_(uid="21-1", upload_date="20210808", playlist_index=1, playlist_count=4), # 1
|
|
_(uid="20-1", upload_date="20200808", playlist_index=2, playlist_count=4), # 2 98
|
|
_(uid="20-2", upload_date="20200808", playlist_index=3, playlist_count=4), # 1 99
|
|
_(uid="20-3", upload_date="20200807", playlist_index=4, playlist_count=4),
|
|
]
|
|
collection_2_entry_dicts = [
|
|
# 20-3 should resolve to collection 1 (which is season 2)
|
|
_(uid="20-3", upload_date="20200807", playlist_index=1, playlist_count=5),
|
|
_(uid="20-4", upload_date="20200806", playlist_index=2, playlist_count=5),
|
|
_(uid="20-5", upload_date="20200706", playlist_index=3, playlist_count=5),
|
|
_(uid="20-6", upload_date="20200706", playlist_index=4, playlist_count=5),
|
|
_(uid="20-7", upload_date="20200606", playlist_index=5, playlist_count=5),
|
|
]
|
|
|
|
with patch.object(
|
|
Downloader, "extract_info_via_info_json"
|
|
) as mock_download_metadata, patch.object(
|
|
Downloader, "_extract_entry_info_with_retry", new=lambda _, entry: entry
|
|
):
|
|
# Stub out metadata. TODO: update this if we do metadata plugins
|
|
mock_download_metadata.side_effect = [collection_1_entry_dicts, collection_2_entry_dicts]
|
|
yield
|