134 lines
4.6 KiB
Python
134 lines
4.6 KiB
Python
from pathlib import Path
|
|
import shutil
|
|
import subprocess
|
|
|
|
import yaml
|
|
|
|
|
|
class ComposeLoader(yaml.SafeLoader):
|
|
pass
|
|
|
|
|
|
def _compose_override(loader, node):
|
|
if isinstance(node, yaml.MappingNode):
|
|
return loader.construct_mapping(node)
|
|
return loader.construct_sequence(node)
|
|
|
|
|
|
ComposeLoader.add_constructor("!override", _compose_override)
|
|
|
|
|
|
def test_psd_overlay_uses_generated_workspace_for_default_and_named_commands():
|
|
root = Path(__file__).resolve().parents[2]
|
|
compose = yaml.load(
|
|
(root / "deploy/compose.psd-local.yaml.example").read_text(),
|
|
Loader=ComposeLoader,
|
|
)
|
|
core = compose["services"]["core"]
|
|
|
|
assert core["environment"]["THT_CONFIG"] == "/app/harness/config/tht.yaml"
|
|
assert core["environment"]["THT_SECRETS_FILE"] == "/run/secrets/thothii.secrets"
|
|
for name in (
|
|
"THT_DB_NAME", "THT_DWH_REST_URL", "THT_VEC_REST_URL",
|
|
"THT_VEC_WRITE_REST_URL", "THT_OLLAMA_URL", "THT_PROFILE",
|
|
"PI_PROVIDER", "PI_MODEL", "PI_THINKING",
|
|
):
|
|
assert name in core["environment"]
|
|
def target(volume):
|
|
if isinstance(volume, dict):
|
|
return volume["target"]
|
|
parts = volume.rsplit(":", 2)
|
|
return parts[-2] if parts[-1] in {"ro", "rw"} else parts[-1]
|
|
|
|
targets = {target(volume) for volume in core["volumes"]}
|
|
assert "/app/harness/config/tht.yaml" in targets
|
|
assert "/app/harness/workspaces/psd.yaml" not in targets
|
|
assert "/data/workspaces/psd/config/tht.yaml" not in targets
|
|
assert "/data" in targets
|
|
assert "/home/thoth/.pi" in targets
|
|
assert "/home/thoth/.pi/agent/models.json" in targets
|
|
assert "/home/thoth/.pi/agent/settings.json" in targets
|
|
assert "/data/evidence" in targets
|
|
assert "/run/secrets/thothii.secrets" in targets
|
|
assert set(compose["volumes"]) == {"thoth_data", "thoth_pi_config"}
|
|
assert core["networks"]["default"]["aliases"] == ["core", "thothii-core"]
|
|
frontend = compose["services"]["frontend"]
|
|
assert frontend["ports"] == ["127.0.0.1:8099:8080"]
|
|
assert frontend["build"]["args"] == {
|
|
"VITE_BASE": "/",
|
|
"VITE_BACKEND_URL": "/api",
|
|
}
|
|
assert frontend["networks"] == {
|
|
"default": {"aliases": ["frontend", "thothii-frontend"]}
|
|
}
|
|
assert compose["networks"]["default"] == {
|
|
"external": True,
|
|
"name": "thothii_default",
|
|
}
|
|
|
|
|
|
def test_core_image_prepares_the_writable_pi_profile_before_mounting_config_files():
|
|
root = Path(__file__).resolve().parents[2]
|
|
dockerfile = (root / "docker/core.Dockerfile").read_text()
|
|
|
|
assert "mkdir -p /home/thoth/.pi/agent" in dockerfile
|
|
assert "chown -R thoth:thoth /home/thoth/.pi" in dockerfile
|
|
assert (
|
|
"cp --remove-destination /app/harness/workspaces/local.yaml "
|
|
"/app/harness/config/tht.yaml" in dockerfile
|
|
)
|
|
assert "ln -sf /app/harness/workspaces/local.yaml /app/harness/config/tht.yaml" not in dockerfile
|
|
assert (
|
|
"ln -sfn /app/harness/config/tht.yaml /app/harness/workspaces/psd.yaml"
|
|
in dockerfile
|
|
)
|
|
|
|
|
|
def test_psd_bootstrap_materializes_the_base_compose_env_file(tmp_path):
|
|
source_root = Path(__file__).resolve().parents[2]
|
|
root = tmp_path / "ThothII"
|
|
(root / "scripts").mkdir(parents=True)
|
|
(root / "deploy/workspaces").mkdir(parents=True)
|
|
shutil.copy(
|
|
source_root / "scripts/bootstrap-local-psd-docker-config.sh",
|
|
root / "scripts/bootstrap-local-psd-docker-config.sh",
|
|
)
|
|
shutil.copy(
|
|
source_root / "deploy/compose.psd-local.yaml.example",
|
|
root / "deploy/compose.psd-local.yaml.example",
|
|
)
|
|
shutil.copy(
|
|
source_root / "deploy/workspaces/psd.yaml.example",
|
|
root / "deploy/workspaces/psd.yaml.example",
|
|
)
|
|
source_env = tmp_path / "source.env"
|
|
source_env.write_text("\n".join([
|
|
"THT_DB_NAME=postgres",
|
|
"THT_DWH_REST_URL=https://dwh.invalid/",
|
|
"THT_VEC_REST_URL=https://vec.invalid/read/",
|
|
"THT_VEC_WRITE_REST_URL=https://vec.invalid/write/",
|
|
"THT_DWH_API_KEY=dwh",
|
|
"THT_VEC_API_KEY=reader",
|
|
"THT_VEC_WRITE_API_KEY=writer",
|
|
"",
|
|
]))
|
|
workspace = tmp_path / "workspace"
|
|
workspace.mkdir()
|
|
auth = tmp_path / "auth.json"
|
|
auth.write_text('{"zai":{"key":"model"}}')
|
|
|
|
subprocess.run(
|
|
[
|
|
"sh", str(root / "scripts/bootstrap-local-psd-docker-config.sh"),
|
|
str(source_env), str(workspace), str(auth),
|
|
],
|
|
check=True,
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
assert (root / "deploy/thothii.env").is_file()
|
|
assert "THT_SECRETS_FILE=./deploy/secrets/thothii.secrets" in (
|
|
root / ".env"
|
|
).read_text()
|