from pathlib import Path import unittest ROOT = Path(__file__).resolve().parents[1] class DeployScriptTest(unittest.TestCase): def test_deploy_uses_saved_managerd_agent_token_without_random_fallback(self): text = (ROOT / "scripts" / "deploy.sh").read_text(encoding="utf-8") self.assertIn('AGENT_TOKEN="${AGENT_TOKEN:-}"', text) self.assertIn('AGENT_TOKEN_FILE="$AGENT_INSTALL_DIR/agent.token"', text) self.assertIn("resolve_agent_token", text) self.assertIn('AGENT_TOKEN="$(tr -d', text) self.assertIn('"token": "$AGENT_TOKEN"', text) self.assertIn('printf \'%s\\n\' "$AGENT_TOKEN" > "$AGENT_TOKEN_FILE"', text) self.assertNotIn("openssl rand", text) self.assertIn("不会本地生成 token", text) self.assertIn("首次安装必须使用", text) if __name__ == "__main__": unittest.main()