This commit is contained in:
root
2026-05-17 04:31:59 +00:00
parent 75a0ca4e31
commit 07aa02bca6
51 changed files with 1322 additions and 435 deletions

View File

@@ -9,6 +9,7 @@ from src.youtube_pro_runtime import (
managed_url_config_relpath_for_pm2,
managed_youtube_ini_relpath_for_pm2,
normalize_youtube_pro_channels,
validate_youtube_runtime_files,
)
@@ -76,6 +77,28 @@ class YoutubeProRuntimeTests(unittest.TestCase):
self.assertIn("legacy-key", youtube_ini)
self.assertIn("https://live.douyin.com/88888888", url_ini)
def test_validate_runtime_files_reports_missing_single_lane_inputs(self) -> None:
with tempfile.TemporaryDirectory() as tmp:
root = Path(tmp)
config_dir = root / "config"
config_dir.mkdir(parents=True, exist_ok=True)
errors = validate_youtube_runtime_files(root, config_dir, "youtube", [])
self.assertEqual(errors, ["youtube: stream key missing", "youtube: source URL missing"])
def test_validate_runtime_files_accepts_legacy_single_lane_pair(self) -> None:
with tempfile.TemporaryDirectory() as tmp:
root = Path(tmp)
config_dir = root / "config"
config_dir.mkdir(parents=True, exist_ok=True)
(config_dir / "youtube.ini").write_text("[youtube]\nkey = legacy-key\nrtmps = 是\n", encoding="utf-8")
(config_dir / "URL_config.ini").write_text("https://live.douyin.com/123456\n", encoding="utf-8")
errors = validate_youtube_runtime_files(root, config_dir, "youtube", [])
self.assertEqual(errors, [])
if __name__ == "__main__":
unittest.main()