#!/usr/bin/env python3
"""
Integration test for Documentation Indexer with the MCP server.
Tests that the index_documentation and search_docs functions work correctly.
"""
import sys
import os
from pathlib import Path
# Add parent directory to path
sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
# Set up environment
os.environ["QDRANT_HOST"] = "localhost"
os.environ["QDRANT_PORT"] = "6333"
os.environ["EMBEDDING_MODEL"] = "all-MiniLM-L6-v2"
def test_documentation_integration():
"""Test the documentation indexing through MCP server functions."""
# Import after env setup
from qdrant_mcp_context_aware import (
index_documentation,
search_docs,
get_qdrant_client,
ensure_collection
)
print("Documentation Integration Test")
print("=" * 60)
# Ensure documentation collection exists
ensure_collection("documentation_collection")
print("✓ Documentation collection initialized")
# Test file
test_file = str(Path(__file__).parent.parent / "README.md")
# Test 1: Index documentation
print(f"\n1. Testing index_documentation with: {test_file}")
try:
result = index_documentation(file_path=test_file)
print(f" ✓ Indexed successfully: {result['message']}")
print(f" Chunks created: {result['indexed']}")
print(f" File type: {result['doc_type']}")
print(f" Title: {result['title']}")
except Exception as e:
print(f" ❌ Error: {e}")
import traceback
traceback.print_exc()
return False
# Test 2: Search documentation
print("\n2. Testing search_docs")
try:
# Search for something we know is in README
search_result = search_docs(query="MCP server", n_results=3)
print(f" ✓ Search completed")
print(f" Results found: {len(search_result['results'])}")
if search_result['results']:
print("\n First result:")
first = search_result['results'][0]
print(f" - Score: {first['score']:.3f}")
print(f" - File: {first['metadata']['file_name']}")
print(f" - Heading: {first['metadata'].get('heading', 'N/A')}")
print(f" - Content preview: {first['content'][:100]}...")
except Exception as e:
print(f" ❌ Error: {e}")
import traceback
traceback.print_exc()
return False
# Test 3: Index multiple docs in directory
print("\n3. Testing documentation indexing through index_directory")
try:
# This would be called through index_directory in real usage
docs_dir = Path(__file__).parent.parent / "docs"
md_files = list(docs_dir.glob("*.md"))[:3] # Just test first 3
print(f" Found {len(md_files)} markdown files to index")
for md_file in md_files:
result = index_documentation(file_path=str(md_file))
print(f" ✓ Indexed {md_file.name}: {result['indexed']} chunks")
except Exception as e:
print(f" ❌ Error: {e}")
import traceback
traceback.print_exc()
return False
# Test 4: Search across all docs
print("\n4. Testing cross-document search")
try:
search_result = search_docs(query="setup installation", n_results=5)
print(f" ✓ Found {len(search_result['results'])} results")
# Show file distribution
files = set(r['metadata']['file_name'] for r in search_result['results'])
print(f" Results from {len(files)} different files: {files}")
except Exception as e:
print(f" ❌ Error: {e}")
import traceback
traceback.print_exc()
return False
print("\n" + "=" * 60)
print("✓ All integration tests passed!")
return True
def test_collection_isolation():
"""Test that documentation collection is isolated from code/config."""
from qdrant_mcp_context_aware import get_qdrant_client
print("\n" + "=" * 60)
print("COLLECTION ISOLATION TEST:")
print("=" * 60)
qdrant_client = get_qdrant_client()
try:
# Check collections
collections = qdrant_client.get_collections()
collection_names = [c.name for c in collections.collections]
print(f"\nAvailable collections: {collection_names}")
# Verify documentation collection exists
if "documentation_collection" in collection_names:
print("✓ Documentation collection exists")
# Get collection info
info = qdrant_client.get_collection("documentation_collection")
print(f" Points count: {info.points_count}")
print(f" Vectors size: {info.config.params.vectors.size}")
else:
print("⚠ Documentation collection not found (will be created on first index)")
# Verify isolation from other collections
for coll in ["code_collection", "config_collection"]:
if coll in collection_names:
info = qdrant_client.get_collection(coll)
print(f"\n{coll}:")
print(f" Points count: {info.points_count}")
print(f" Status: Isolated from documentation")
print("\n✓ Collections are properly isolated")
except Exception as e:
print(f"❌ Error checking collections: {e}")
import traceback
traceback.print_exc()
return False
return True
if __name__ == "__main__":
print("Starting Documentation Integration Tests")
print("=" * 60)
print("Note: Requires Qdrant to be running on localhost:6333")
print()
try:
# Run main integration test
success = test_documentation_integration()
if success:
# Run collection isolation test
test_collection_isolation()
print("\n" + "=" * 60)
print("ALL TESTS COMPLETED! 🎉")
print("Documentation Indexer v0.2.3 is ready for release.")
except Exception as e:
print(f"\n❌ Integration test failed: {e}")
import traceback
traceback.print_exc()
