"""
Test configuration and fixtures.
"""
import pytest
import os
from unittest.mock import Mock, AsyncMock, MagicMock
from pathlib import Path
# Set up test environment
os.environ["SUPABASE_URL"] = "https://test.supabase.co"
os.environ["SUPABASE_SECRET_KEY"] = "test-secret-key"
@pytest.fixture
def mock_supabase_client():
"""Create a mock Supabase client."""
client = MagicMock()
# Mock table operations
client.table = MagicMock(return_value=MagicMock())
client.rpc = MagicMock(return_value=MagicMock())
return client
@pytest.fixture
def mock_directives_search_result():
"""Mock search results for directives."""
return {
"data": [
{
"id": "1",
"name": "bootstrap_directive",
"category": "setup/initialization",
"description": "Bootstrap a new project with essential configurations",
"is_official": True,
"download_count": 150,
"quality_score": 95.0,
"created_at": "2024-01-01T00:00:00Z",
"updated_at": "2024-01-15T00:00:00Z",
},
{
"id": "2",
"name": "setup_auth",
"category": "authentication/oauth",
"description": "Setup OAuth authentication with multiple providers",
"is_official": False,
"download_count": 50,
"quality_score": 80.0,
"created_at": "2024-01-05T00:00:00Z",
"updated_at": "2024-01-10T00:00:00Z",
},
]
}
@pytest.fixture
def mock_tools_search_result():
"""Mock search results for tools."""
return {
"data": [
{
"id": "1",
"name": "google_maps_scraper",
"category": "scraping/maps", # Combined from category + subcategory
"description": "Scrape business locations from Google Maps",
"is_official": True,
"download_count": 500,
"quality_score": 98.0,
"tech_stack": ["Python", "Selenium"],
"tags": ["scraping", "google-maps"],
"success_rate": 0.95,
"estimated_cost_usd": 0.05,
"latest_version": "2.1.0",
"created_at": "2024-01-01T00:00:00Z",
"updated_at": "2024-01-15T00:00:00Z",
},
{
"id": "2",
"name": "email_validator",
"category": "validation/email", # Combined from category + subcategory
"description": "Validate email addresses and check deliverability",
"is_official": False,
"download_count": 200,
"quality_score": 85.0,
"tech_stack": ["Python", "Email"],
"tags": ["email", "validation"],
"success_rate": 0.90,
"estimated_cost_usd": 0.01,
"latest_version": "1.5.0",
"created_at": "2024-01-05T00:00:00Z",
"updated_at": "2024-01-10T00:00:00Z",
},
]
}
@pytest.fixture
def mock_knowledge_search_result():
"""Mock search results for knowledge entries."""
return {
"data": [
{
"id": "001-email-deliverability",
"title": "Email Deliverability Best Practices",
"entry_type": "pattern",
"category": "email-infrastructure",
"tags": ["email", "smtp", "deliverability"],
"snippet": "Email deliverability refers to the ability of a transactional email to reach its intended email inbox.",
},
{
"id": "002-jwt-auth",
"title": "JWT Authentication in Python",
"entry_type": "learning",
"category": "authentication",
"tags": ["jwt", "python", "auth"],
"snippet": "JWT tokens are self-contained and can be used for stateless authentication.",
},
]
}
@pytest.fixture
def directive_registry(monkeypatch):
"""Create a DirectiveRegistry with mocked Supabase client."""
from kiwi_mcp.api.directive_registry import DirectiveRegistry
registry = DirectiveRegistry()
# Mock the client to avoid actual connection attempts
registry.client = MagicMock()
return registry
@pytest.fixture
def tool_registry(monkeypatch):
"""Create a ToolRegistry with mocked Supabase client."""
from kiwi_mcp.api.tool_registry import ToolRegistry
registry = ToolRegistry()
# Mock the client to avoid actual connection attempts
registry.client = MagicMock()
return registry
@pytest.fixture
def knowledge_registry(monkeypatch):
"""Create a KnowledgeRegistry with mocked Supabase client."""
from kiwi_mcp.api.knowledge_registry import KnowledgeRegistry
registry = KnowledgeRegistry()
# Mock the client to avoid actual connection attempts
registry.client = MagicMock()
return registry
@pytest.fixture
def mock_filesystem(tmp_path):
"""Create a mock filesystem for testing."""
return {
"project_root": tmp_path,
"directives_dir": tmp_path / ".ai" / "directives",
"tools_dir": tmp_path / ".ai" / "scripts",
"knowledge_dir": tmp_path / ".ai" / "knowledge",
}
# ============================================================================
# Test File Creation Helpers
# ============================================================================
@pytest.fixture
def sample_directive_content():
"""Sample directive XML content for testing."""
return """<directive name="test_directive" version="1.0.0">
<metadata>
<description>Test directive for unit testing</description>
<permissions>
<allow type="read" scope="all" />
</permissions>
<model tier="reasoning" />
</metadata>
<process>
<step name="test">
<action>Test action</action>
</step>
</process>
</directive>"""
@pytest.fixture
def sample_directive_file(tmp_path, sample_directive_content):
"""Create a sample directive file for testing."""
directive_dir = tmp_path / ".ai" / "directives" / "test"
directive_dir.mkdir(parents=True)
file_content = f"""# Test Directive
```xml
{sample_directive_content}
```
"""
directive_file = directive_dir / "test_directive.md"
directive_file.write_text(file_content)
return directive_file
@pytest.fixture
def sample_tool_content():
"""Sample tool content for testing."""
return '''"""Test tool for unit testing"""
__version__ = "1.0.0"
import sys
import json
def main():
result = {"status": "success", "data": {"message": "test"}}
print(json.dumps(result))
if __name__ == "__main__":
main()
'''
@pytest.fixture
def sample_tool_file(tmp_path, sample_tool_content):
"""Create a sample tool file for testing."""
tool_dir = tmp_path / ".ai" / "scripts" / "test"
tool_dir.mkdir(parents=True)
tool_file = tool_dir / "test_tool.py"
tool_file.write_text(sample_tool_content)
return tool_file
@pytest.fixture
def sample_knowledge_content():
"""Sample knowledge entry content for testing."""
return "This is test knowledge content for unit testing."
@pytest.fixture
def sample_knowledge_file(tmp_path, sample_knowledge_content):
"""Create a sample knowledge entry file for testing."""
knowledge_dir = tmp_path / ".ai" / "knowledge" / "test"
knowledge_dir.mkdir(parents=True)
file_content = f"""---
version: "1.0.0"
id: 001-test
title: Test Knowledge Entry
entry_type: learning
category: test
tags: []
---
{sample_knowledge_content}
"""
knowledge_file = knowledge_dir / "001-test.md"
knowledge_file.write_text(file_content)
return knowledge_file
