25 lines
885 B
Python
25 lines
885 B
Python
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()
|