add test

Author: 0xrushi

tests/unit/test_wizard_strixhalo.py

@@ -185,3 +185,126 @@ def test_select_hardware_profile_skips_when_not_needed(monkeypatch):
     )
 
     assert result is None
+
+
+def test_run_service_setup_openmemory_prefills_local_embeddings_from_backend(
+    monkeypatch,
+):
+    captured = {}
+
+    def fake_run(cmd, cwd, check, timeout):
+        captured["cmd"] = cmd
+        return argparse.Namespace(returncode=0)
+
+    def mock_read_env_value(path, key):
+        if path == "backends/advanced/.env":
+            values = {
+                "OPENAI_API_KEY": "local-embeddings-key",
+                "OPENAI_BASE_URL": "http://host.docker.internal:11434/v1",
+                "OPENAI_EMBEDDING_MODEL": "nomic-embed-text",
+                "OPENAI_EMBEDDING_DIMENSIONS": "768",
+            }
+            return values.get(key)
+        return None
+
+    monkeypatch.setattr(wizard, "check_service_exists", lambda *_: (True, "OK"))
+    monkeypatch.setattr(wizard, "read_env_value", mock_read_env_value)
+    monkeypatch.setattr(wizard.subprocess, "run", fake_run)
+
+    ok = wizard.run_service_setup(
+        service_name="openmemory-mcp",
+        selected_services=["advanced", "openmemory-mcp"],
+    )
+
+    assert ok is True
+    cmd = captured["cmd"]
+    assert "--embeddings-provider" in cmd
+    provider_idx = cmd.index("--embeddings-provider") + 1
+    assert cmd[provider_idx] == "local"
+    assert "--embeddings-base-url" in cmd
+    assert (
+        cmd[cmd.index("--embeddings-base-url") + 1]
+        == "http://host.docker.internal:11434/v1"
+    )
+    assert "--embeddings-model" in cmd
+    assert cmd[cmd.index("--embeddings-model") + 1] == "nomic-embed-text"
+    assert "--embeddings-api-key" in cmd
+    assert cmd[cmd.index("--embeddings-api-key") + 1] == "local-embeddings-key"
+    assert "--embeddings-dimensions" in cmd
+    assert cmd[cmd.index("--embeddings-dimensions") + 1] == "768"
+    assert "--openai-api-key" not in cmd
+
+
+def test_run_service_setup_openmemory_reuses_existing_local_config(monkeypatch):
+    captured = {}
+
+    def fake_run(cmd, cwd, check, timeout):
+        captured["cmd"] = cmd
+        return argparse.Namespace(returncode=0)
+
+    def mock_read_env_value(path, key):
+        if path == "extras/openmemory-mcp/.env":
+            values = {
+                "OPENMEMORY_EMBEDDINGS_PROVIDER": "local",
+                "OPENMEMORY_EMBEDDINGS_BASE_URL": "http://local-embeddings:11434/v1",
+                "OPENMEMORY_EMBEDDINGS_MODEL": "mxbai-embed-large",
+                "OPENMEMORY_EMBEDDINGS_API_KEY": "existing-local-key",
+                "OPENMEMORY_EMBEDDINGS_DIMENSIONS": "1024",
+            }
+            return values.get(key)
+        if path == "backends/advanced/.env" and key == "OPENAI_API_KEY":
+            return "unused-openai-key"
+        return None
+
+    monkeypatch.setattr(wizard, "check_service_exists", lambda *_: (True, "OK"))
+    monkeypatch.setattr(wizard, "read_env_value", mock_read_env_value)
+    monkeypatch.setattr(wizard.subprocess, "run", fake_run)
+
+    ok = wizard.run_service_setup(
+        service_name="openmemory-mcp",
+        selected_services=["advanced", "openmemory-mcp"],
+    )
+
+    assert ok is True
+    cmd = captured["cmd"]
+    assert "--embeddings-provider" in cmd
+    assert cmd[cmd.index("--embeddings-provider") + 1] == "local"
+    assert (
+        cmd[cmd.index("--embeddings-base-url") + 1]
+        == "http://local-embeddings:11434/v1"
+    )
+    assert cmd[cmd.index("--embeddings-model") + 1] == "mxbai-embed-large"
+    assert cmd[cmd.index("--embeddings-api-key") + 1] == "existing-local-key"
+    assert cmd[cmd.index("--embeddings-dimensions") + 1] == "1024"
+
+
+def test_run_service_setup_openmemory_falls_back_to_openai_key(monkeypatch):
+    captured = {}
+
+    def fake_run(cmd, cwd, check, timeout):
+        captured["cmd"] = cmd
+        return argparse.Namespace(returncode=0)
+
+    def mock_read_env_value(path, key):
+        if path == "backends/advanced/.env":
+            values = {
+                "OPENAI_API_KEY": "sk-openai",
+                "OPENAI_BASE_URL": "https://api.openai.com/v1",
+            }
+            return values.get(key)
+        return None
+
+    monkeypatch.setattr(wizard, "check_service_exists", lambda *_: (True, "OK"))
+    monkeypatch.setattr(wizard, "read_env_value", mock_read_env_value)
+    monkeypatch.setattr(wizard.subprocess, "run", fake_run)
+
+    ok = wizard.run_service_setup(
+        service_name="openmemory-mcp",
+        selected_services=["advanced", "openmemory-mcp"],
+    )
+
+    assert ok is True
+    cmd = captured["cmd"]
+    assert "--openai-api-key" in cmd
+    assert cmd[cmd.index("--openai-api-key") + 1] == "sk-openai"
+    assert "--embeddings-provider" not in cmd