d5e647554e
* [DEV] Fix resolve_once script bug with custom functions * function name * fix adding custom functions
87 lines
3 KiB
Python
87 lines
3 KiB
Python
from ytdl_sub.script.script import Script
|
|
from ytdl_sub.script.script_output import ScriptOutput
|
|
from ytdl_sub.script.types.map import Map
|
|
from ytdl_sub.script.types.resolvable import String
|
|
|
|
|
|
class TestScript:
|
|
def test_pre_resolved(self):
|
|
assert Script(
|
|
{
|
|
"%custom_func": "return {[$0, $1]}",
|
|
"aa": "a",
|
|
"bb": "b",
|
|
"cc": "{%custom_func(aa, bb)}",
|
|
}
|
|
).resolve(resolved={"bb": String("bb_override")}) == ScriptOutput(
|
|
{
|
|
"aa": String("a"),
|
|
"bb": String("bb_override"),
|
|
"cc": String('return ["a", "bb_override"]'),
|
|
}
|
|
)
|
|
|
|
def test_partial_resolve(self):
|
|
assert Script(
|
|
{
|
|
"%custom_func": "return {[$0, $1]}",
|
|
"aa": "a",
|
|
"bb": "b",
|
|
"cc": "{%custom_func(aa, bb)}",
|
|
}
|
|
).resolve(unresolvable={"bb"}) == ScriptOutput({"aa": String("a")})
|
|
|
|
def test_partial_update_script(self):
|
|
# to be resolved later
|
|
entry_map = Map({String("title"): String("the title")})
|
|
|
|
script = Script(
|
|
{
|
|
"entry": "{%throw('entry has not been populated yet')}",
|
|
"title": "{%map_get(entry, 'title')}",
|
|
"override": "hi",
|
|
"resolved_override": "{override} mom",
|
|
}
|
|
)
|
|
|
|
script.resolve(unresolvable={"entry"}, update=True)
|
|
assert script.get("override") == String("hi")
|
|
assert script.get("resolved_override") == String("hi mom")
|
|
|
|
script.add(
|
|
{
|
|
"new_variable_titlecase": "{%titlecase(new_variable_upper)}",
|
|
"new_variable": "{resolved_override} {title}",
|
|
"new_variable_upper": "{%upper(new_variable)}",
|
|
}
|
|
).resolve(resolved={"entry": entry_map}, update=True)
|
|
|
|
assert script.get("title") == String("the title")
|
|
assert script.get("new_variable") == String("hi mom the title")
|
|
assert script.get("new_variable_upper") == String("HI MOM THE TITLE")
|
|
assert script.get("new_variable_titlecase") == String("Hi Mom The Title")
|
|
assert script.get("entry") == entry_map
|
|
|
|
def test_resolve_once_with_custom_functions(self):
|
|
script = Script(
|
|
{
|
|
"%is_bilateral_url": "{ %not(%contains( $0, 'youtube.com/playlist' )) }",
|
|
"%bilateral_url": """{
|
|
%if(
|
|
%and(
|
|
enable_bilateral_scraping,
|
|
%is_bilateral_url($0)
|
|
),
|
|
$0,
|
|
""
|
|
)
|
|
}""",
|
|
"enable_bilateral_scraping": "True",
|
|
}
|
|
)
|
|
|
|
script.add({"%bilateral_url_wrap": "{ %bilateral_url($0) }"})
|
|
|
|
assert (
|
|
script.resolve_once({"url": "{ %bilateral_url_wrap('nope') }"})["url"].native == "nope"
|
|
)
|