File size: 4,201 Bytes
020c337 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 | import json
import os
import subprocess
import tempfile
import unittest
from pathlib import Path
class EntrypointConfigTests(unittest.TestCase):
@staticmethod
def _repo_root() -> Path:
return Path(__file__).resolve().parents[1]
@classmethod
def _entrypoint_path(cls) -> Path:
return cls._repo_root() / "scripts" / "openclaw-entrypoint.sh"
@staticmethod
def _write_fake_openclaw(bin_dir: Path) -> None:
fake_openclaw = bin_dir / "openclaw"
fake_openclaw.write_text("#!/usr/bin/env bash\nexit 0\n", encoding="utf-8")
fake_openclaw.chmod(0o755)
def _run_gateway(self, extra_env=None):
if extra_env is None:
extra_env = {}
with tempfile.TemporaryDirectory() as tmp:
tmp_path = Path(tmp)
state_dir = tmp_path / "state"
home_dir = tmp_path / "home"
workspace_dir = state_dir / "workspace"
logs_dir = tmp_path / "logs"
bin_dir = tmp_path / "bin"
config_path = state_dir / "openclaw.json"
state_dir.mkdir(parents=True)
home_dir.mkdir(parents=True)
workspace_dir.mkdir(parents=True)
logs_dir.mkdir(parents=True)
bin_dir.mkdir(parents=True)
self._write_fake_openclaw(bin_dir)
env = os.environ.copy()
for key in ("OPENCLAW_LLM_BASE_URL", "OPENCLAW_LLM_MODEL", "OPENCLAW_LLM_API_KEY"):
env.pop(key, None)
env.update(
{
"PATH": f"{bin_dir}:{env.get('PATH', '')}",
"OPENCLAW_HOME": str(home_dir),
"OPENCLAW_STATE_DIR": str(state_dir),
"OPENCLAW_CONFIG_PATH": str(config_path),
"OPENCLAW_WORKSPACE_DIR": str(workspace_dir),
"OPENCLAW_STDOUT_LOG_PATH": str(logs_dir / "gateway.stdout.log"),
"OPENCLAW_STDERR_LOG_PATH": str(logs_dir / "gateway.stderr.log"),
"OPENCLAW_BACKUP_ENABLED": "false",
"OPENCLAW_SSHX_AUTO_START": "false",
}
)
env.update(extra_env)
result = subprocess.run(
[str(self._entrypoint_path()), "gateway"],
cwd=self._repo_root(),
env=env,
capture_output=True,
text=True,
check=True,
)
with config_path.open("r", encoding="utf-8") as fh:
config = json.load(fh)
return config, result
def test_gateway_configures_custom_model_when_llm_env_present(self):
config, _ = self._run_gateway(
{
"OPENCLAW_LLM_BASE_URL": "https://example.com/v1",
"OPENCLAW_LLM_MODEL": "demo-model",
"OPENCLAW_LLM_API_KEY": "demo-key",
}
)
self.assertEqual(config["agents"]["defaults"]["model"]["primary"], "thirdparty/demo-model")
self.assertEqual(
config["agents"]["defaults"]["models"]["thirdparty/demo-model"]["alias"],
"Default",
)
self.assertEqual(config["models"]["mode"], "merge")
self.assertEqual(config["models"]["providers"]["thirdparty"]["baseUrl"], "${OPENCLAW_LLM_BASE_URL}")
self.assertEqual(config["models"]["providers"]["thirdparty"]["apiKey"], "${OPENCLAW_LLM_API_KEY}")
self.assertEqual(config["models"]["providers"]["thirdparty"]["api"], "openai-completions")
self.assertEqual(
config["models"]["providers"]["thirdparty"]["models"],
[{"id": "demo-model", "name": "demo-model"}],
)
def test_gateway_skips_custom_model_when_llm_env_missing(self):
config, result = self._run_gateway()
defaults = config.get("agents", {}).get("defaults", {})
providers = config.get("models", {}).get("providers", {})
self.assertNotIn("model", defaults)
self.assertNotIn("models", defaults)
self.assertNotIn("thirdparty", providers)
self.assertIn("OPENCLAW_SSHX_AUTO_START=true", result.stderr)
if __name__ == "__main__":
unittest.main()
|