ytdl-sub/tests/unit/entries/test_entry.py
2022-08-24 09:59:34 -07:00

55 lines
2.1 KiB
Python

import pytest
class TestEntry(object):
def test_entry_to_dict(self, mock_entry, mock_entry_to_dict):
assert mock_entry.to_dict() == mock_entry_to_dict
def test_entry_dict_contains_valid_formatters(
self, mock_entry, validate_entry_dict_contains_valid_formatters
):
assert validate_entry_dict_contains_valid_formatters(mock_entry)
def test_entry_missing_kwarg(self, mock_entry):
key = "dne"
expected_error_msg = f"Expected '{key}' in Entry but does not exist."
assert mock_entry.kwargs_contains(key) is False
with pytest.raises(KeyError, match=expected_error_msg):
mock_entry.kwargs(key)
@pytest.mark.parametrize(
"upload_date, year_rev, month_rev, day_rev, month_rev_pad, day_rev_pad",
[
("20000228", 100, 11, 2, "11", "02"),
("20200808", 80, 5, 24, "05", "24"),
],
)
def test_entry_reverse_variables(
self, mock_entry, upload_date, year_rev, month_rev, day_rev, month_rev_pad, day_rev_pad
):
mock_entry._kwargs["upload_date"] = upload_date
assert mock_entry.upload_year_truncated_reversed == year_rev
assert mock_entry.upload_month_reversed == month_rev
assert mock_entry.upload_day_reversed == day_rev
assert mock_entry.upload_month_reversed_padded == month_rev_pad
assert mock_entry.upload_day_reversed_padded == day_rev_pad
@pytest.mark.parametrize(
"upload_date, day_year, day_year_rev, day_year_pad, day_year_rev_pad",
[
("20000228", 59, 308, "059", "308"),
("20210808", 220, 146, "220", "146"),
],
)
def test_entry_upload_day_of_year_variables(
self, mock_entry, upload_date, day_year, day_year_rev, day_year_pad, day_year_rev_pad
):
mock_entry._kwargs["upload_date"] = upload_date
assert mock_entry.upload_day_of_year == day_year
assert mock_entry.upload_day_of_year_reversed == day_year_rev
assert mock_entry.upload_day_of_year_padded == day_year_pad
assert mock_entry.upload_day_of_year_reversed_padded == day_year_rev_pad