tests, TODO, need docs

This commit is contained in:
Jesse Bannon 2025-10-10 22:19:34 -07:00
parent ef2ca5e6cc
commit a4e1f2cd22
3 changed files with 24 additions and 3 deletions

View file

@ -1162,6 +1162,7 @@ class VariableDefinitions(
""" """
return { return {
self.uid, self.uid,
self.extractor,
self.extractor_key, self.extractor_key,
self.epoch, self.epoch,
self.webpage_url, self.webpage_url,

View file

@ -74,8 +74,6 @@ class TestUrl:
upload_date_idx=0, upload_date_idx=0,
).initialize_script(subscription.overrides) ).initialize_script(subscription.overrides)
entry.add({ entry.add({v.ytdl_sub_input_url_index.variable_name: 0})
v.ytdl_sub_input_url_index.variable_name: 0
})
assert downloader.webpage_url(entry) == "youtube.com/test_url_append_test" assert downloader.webpage_url(entry) == "youtube.com/test_url_append_test"

View file

@ -58,3 +58,25 @@ class TestEntry(object):
assert entry.get(v.upload_day_of_year_reversed, int) == day_year_rev assert entry.get(v.upload_day_of_year_reversed, int) == day_year_rev
assert entry.get(v.upload_day_of_year_padded, str) == day_year_pad assert entry.get(v.upload_day_of_year_padded, str) == day_year_pad
assert entry.get(v.upload_day_of_year_reversed_padded, str) == day_year_rev_pad assert entry.get(v.upload_day_of_year_reversed_padded, str) == day_year_rev_pad
def test_entry_required_variables(self):
entry_dict = {
v.uid.metadata_key: "0",
v.extractor.metadata_key: "test",
v.extractor_key.metadata_key: "test",
v.epoch.metadata_key: 123,
v.webpage_url.metadata_key: "youtube.com/test_url",
v.ext.metadata_key: "mp4",
}
assert set(entry_dict.keys()) == {_.metadata_key for _ in v.required_entry_variables()}
# Ensure adding variables works
entry = Entry(
entry_dict=entry_dict,
working_directory=".",
).initialize_script()
entry.add({v.channel: "can add"})
assert entry.get(v.channel, str) == "can add"