530d2f7666
Adds the mechanism to partially resolve an entire script (aka every underlying variable in ytdl-sub). This will be used in the upcoming `inspect` sub-command where it will dump a subscription's raw contents at multiple levels of resolution. Potentially useful to optimize script execution as well.
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
import json
|
|
import os
|
|
import shutil
|
|
from pathlib import Path
|
|
from typing import Dict
|
|
|
|
DISABLE_YOUTUBE_TESTS: bool = True
|
|
REGENERATE_FIXTURES: bool = False
|
|
|
|
RESOURCE_PATH: Path = Path("tests") / "resources"
|
|
_FILE_FIXTURE_PATH: Path = RESOURCE_PATH / "file_fixtures"
|
|
_EXPECTED_JSON_PATH: Path = RESOURCE_PATH / "expected_json"
|
|
|
|
|
|
def file_fixture_path(fixture_name: str) -> Path:
|
|
return _FILE_FIXTURE_PATH / fixture_name
|
|
|
|
|
|
def expected_json(input_json: Dict, json_name: str) -> Dict:
|
|
expected_json_path = _EXPECTED_JSON_PATH / json_name
|
|
os.makedirs(os.path.dirname(expected_json_path), exist_ok=True)
|
|
|
|
if REGENERATE_FIXTURES:
|
|
with open(expected_json_path, "w", encoding="utf-8") as json_file:
|
|
json.dump(input_json, json_file, sort_keys=True, indent=2)
|
|
|
|
return input_json
|
|
|
|
with open(expected_json_path, "r", encoding="utf-8") as json_file:
|
|
expected_json = json.load(json_file)
|
|
|
|
return expected_json
|
|
|
|
|
|
def copy_file_fixture(fixture_name: str, output_file_path: Path) -> None:
|
|
os.makedirs(os.path.dirname(output_file_path), exist_ok=True)
|
|
shutil.copy(file_fixture_path(fixture_name), output_file_path)
|