s
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user