wechat-robot-skills/tests/test_private_token_headers.py
2026-08-27 10:52:49 +08:00

28 lines
1008 B
Python

import unittest
from pathlib import Path
REPO_ROOT = Path(__file__).resolve().parents[1]
class PrivateTokenHeaderTests(unittest.TestCase):
def test_every_local_client_api_script_supports_optional_private_token(self) -> None:
client_api_scripts: list[Path] = []
for pattern in ("*.py", "*.ts"):
for script in (REPO_ROOT / "skills").glob(f"*/scripts/{pattern}"):
content = script.read_text(encoding="utf-8")
if "/api/v1/robot" not in content:
continue
client_api_scripts.append(script)
self.assertIn("ROBOT_CLIENT_PRIVATE_TOKEN", content, script)
self.assertIn("X-Private-Token", content, script)
self.assertNotIn(
"环境变量 ROBOT_CLIENT_PRIVATE_TOKEN 未配置", content, script
)
self.assertTrue(client_api_scripts, "no local client API scripts were found")
if __name__ == "__main__":
unittest.main()