165 lines
3.4 KiB
Python
165 lines
3.4 KiB
Python
import pytest
|
|
|
|
from ytdl_sub.entries.entry import Entry
|
|
from ytdl_sub.validators.string_formatter_validators import StringFormatterValidator
|
|
|
|
|
|
def _pad(num):
|
|
if num < 10:
|
|
return f"0{num}"
|
|
return str(num)
|
|
|
|
|
|
@pytest.fixture
|
|
def uid():
|
|
return "abc123"
|
|
|
|
|
|
@pytest.fixture
|
|
def extractor():
|
|
return "xtract"
|
|
|
|
|
|
@pytest.fixture
|
|
def title():
|
|
return "entry title"
|
|
|
|
|
|
@pytest.fixture
|
|
def description():
|
|
return "a description"
|
|
|
|
|
|
@pytest.fixture
|
|
def upload_year():
|
|
return 2021
|
|
|
|
|
|
@pytest.fixture
|
|
def upload_month():
|
|
return 1
|
|
|
|
|
|
@pytest.fixture
|
|
def upload_day():
|
|
return 12
|
|
|
|
|
|
@pytest.fixture
|
|
def upload_date(upload_year, upload_month, upload_day):
|
|
return f"{upload_year}{_pad(upload_month)}{_pad(upload_day)}"
|
|
|
|
|
|
@pytest.fixture
|
|
def ext():
|
|
return "mp5"
|
|
|
|
|
|
@pytest.fixture
|
|
def thumbnail_ext():
|
|
return "jpg"
|
|
|
|
|
|
@pytest.fixture
|
|
def download_thumbnail_name(uid, thumbnail_ext):
|
|
return f"{uid}.{thumbnail_ext}"
|
|
|
|
|
|
@pytest.fixture
|
|
def download_file_name(uid, ext):
|
|
return f"{uid}.{ext}"
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_entry_to_dict(
|
|
uid,
|
|
title,
|
|
ext,
|
|
extractor,
|
|
upload_date,
|
|
upload_year,
|
|
thumbnail_ext,
|
|
description,
|
|
upload_month,
|
|
upload_day,
|
|
):
|
|
return {
|
|
"uid": uid,
|
|
"title": title,
|
|
"title_sanitized": title,
|
|
"ext": ext,
|
|
"extractor": extractor,
|
|
"upload_date": upload_date,
|
|
"upload_date_standardized": f"{upload_year}-{_pad(upload_month)}-{_pad(upload_day)}",
|
|
"upload_year": upload_year,
|
|
"upload_year_truncated": int(str(upload_year)[:2]),
|
|
"upload_month": upload_month,
|
|
"upload_month_padded": _pad(upload_month),
|
|
"upload_day": upload_day,
|
|
"upload_day_padded": _pad(upload_day),
|
|
"thumbnail_ext": thumbnail_ext,
|
|
"description": description,
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_entry_kwargs(
|
|
uid, title, ext, upload_date, extractor, description, download_thumbnail_name
|
|
):
|
|
return {
|
|
"id": uid,
|
|
"extractor": extractor,
|
|
"title": title,
|
|
"ext": ext,
|
|
"upload_date": upload_date,
|
|
"thumbnail": download_thumbnail_name,
|
|
"description": description,
|
|
}
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_entry(mock_entry_kwargs):
|
|
return Entry(entry_dict=mock_entry_kwargs, working_directory=".")
|
|
|
|
|
|
@pytest.fixture
|
|
def validate_entry_properties(
|
|
uid,
|
|
title,
|
|
upload_date,
|
|
upload_year,
|
|
ext,
|
|
extractor,
|
|
thumbnail_ext,
|
|
download_file_name,
|
|
download_thumbnail_name,
|
|
):
|
|
def _validate_entry_properties(entry: Entry):
|
|
assert entry.uid == uid
|
|
assert entry.title == title
|
|
assert entry.title_sanitized == title
|
|
assert entry.upload_date == upload_date
|
|
assert entry.upload_year == upload_year
|
|
assert entry.ext == ext
|
|
assert entry.thumbnail_ext == thumbnail_ext
|
|
assert entry.extractor == extractor
|
|
|
|
return True
|
|
|
|
return _validate_entry_properties
|
|
|
|
|
|
@pytest.fixture
|
|
def validate_entry_dict_contains_valid_formatters():
|
|
def _validate_entry_dict_contains_valid_formatters(entry: Entry):
|
|
for key, value in entry.to_dict().items():
|
|
expected_string = f"test {value} formatting works"
|
|
formatter = StringFormatterValidator(
|
|
name="test", value=f"test {{{key}}} formatting works"
|
|
)
|
|
|
|
assert formatter.apply_formatter(entry.to_dict()) == expected_string
|
|
|
|
return True
|
|
|
|
return _validate_entry_dict_contains_valid_formatters
|