mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
Paul Hernandez
55d675e278
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
90 lines
3.0 KiB
Python
90 lines
3.0 KiB
Python
"""Tests for MCP UI template helpers (templates.py).
|
|
|
|
Covers get_ui_variant(), load_html(), and load_variant_html().
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from basic_memory.mcp.ui.templates import (
|
|
get_ui_variant,
|
|
load_html,
|
|
load_variant_html,
|
|
DEFAULT_VARIANT,
|
|
)
|
|
|
|
|
|
class TestGetUIVariant:
|
|
"""Tests for get_ui_variant()."""
|
|
|
|
def test_default_variant(self, monkeypatch):
|
|
"""Returns 'vanilla' when env var is not set."""
|
|
monkeypatch.delenv("BASIC_MEMORY_MCP_UI_VARIANT", raising=False)
|
|
assert get_ui_variant() == "vanilla"
|
|
|
|
def test_vanilla_variant(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "vanilla")
|
|
assert get_ui_variant() == "vanilla"
|
|
|
|
def test_tool_ui_variant(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "tool-ui")
|
|
assert get_ui_variant() == "tool-ui"
|
|
|
|
def test_mcp_ui_variant(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "mcp-ui")
|
|
assert get_ui_variant() == "mcp-ui"
|
|
|
|
def test_unsupported_variant_falls_back(self, monkeypatch):
|
|
"""Unsupported values fall back to DEFAULT_VARIANT."""
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "nonexistent")
|
|
assert get_ui_variant() == DEFAULT_VARIANT
|
|
|
|
def test_whitespace_trimmed(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", " tool-ui ")
|
|
assert get_ui_variant() == "tool-ui"
|
|
|
|
def test_case_insensitive(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "VANILLA")
|
|
assert get_ui_variant() == "vanilla"
|
|
|
|
|
|
class TestLoadHtml:
|
|
"""Tests for load_html()."""
|
|
|
|
def test_load_search_results_vanilla(self):
|
|
html = load_html("search-results-vanilla.html")
|
|
assert isinstance(html, str)
|
|
assert len(html) > 0
|
|
# HTML files should contain standard HTML markers
|
|
assert "<" in html
|
|
|
|
def test_load_note_preview_vanilla(self):
|
|
html = load_html("note-preview-vanilla.html")
|
|
assert isinstance(html, str)
|
|
assert len(html) > 0
|
|
|
|
def test_load_nonexistent_raises(self):
|
|
with pytest.raises(FileNotFoundError):
|
|
load_html("does-not-exist.html")
|
|
|
|
|
|
class TestLoadVariantHtml:
|
|
"""Tests for load_variant_html()."""
|
|
|
|
def test_loads_vanilla_variant(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "vanilla")
|
|
html = load_variant_html("search-results")
|
|
assert isinstance(html, str)
|
|
assert len(html) > 0
|
|
|
|
def test_loads_tool_ui_variant(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "tool-ui")
|
|
html = load_variant_html("search-results")
|
|
assert isinstance(html, str)
|
|
assert len(html) > 0
|
|
|
|
def test_loads_mcp_ui_variant(self, monkeypatch):
|
|
monkeypatch.setenv("BASIC_MEMORY_MCP_UI_VARIANT", "mcp-ui")
|
|
html = load_variant_html("note-preview")
|
|
assert isinstance(html, str)
|
|
assert len(html) > 0
|