- {['knowledge_docs', 'project_profile', 'project_index', 'project_memories'].map((type) => {
+ {['knowledge_docs', 'project_profile', 'project_index', 'project_memories', 'code_structure'].map((type) => {
const mcp = projectMCPs.find((m: ProjectMCP) => m.mcp_type === type);
return (
/dev/null; then
- echo "⚠️ Warning: Could not connect to PostgreSQL"
- echo " Make sure PostgreSQL is running and accessible"
-fi
-
echo ""
echo "Starting MCP Server..."
echo "📡 MCP Server will be available at: http://localhost:8051"
diff --git a/tests/README.md b/tests/README.md
index fea7b0c..5a531b1 100644
--- a/tests/README.md
+++ b/tests/README.md
@@ -9,21 +9,13 @@ Comprehensive tests for the Claude OS RAG system.
pip install -r requirements.txt
```
-2. **Create test database:**
+2. **Set environment variables (optional):**
```bash
- createdb codeforge_test
- psql -d codeforge_test -c "CREATE EXTENSION vector"
- ```
-
-3. **Set environment variables (optional):**
- ```bash
- export TEST_POSTGRES_HOST=localhost
- export TEST_POSTGRES_PORT=5432
- export TEST_POSTGRES_DB=codeforge_test
- export TEST_POSTGRES_USER=$USER
export TEST_OLLAMA_HOST=http://localhost:11434
```
+ Note: Tests use SQLite database which is automatically created in the test environment.
+
## Running Tests
### Run all tests:
@@ -75,7 +67,6 @@ pytest -x
## Test Structure
- **`conftest.py`** - Shared fixtures and configuration
-- **`test_pg_vector.py`** - PostgreSQL vector operations
- **`test_rag_engine.py`** - RAG engine functionality
- **`test_embeddings.py`** - Embedding generation
- **`test_document_processing.py`** - Document ingestion and chunking
@@ -126,11 +117,6 @@ Tests can be run in CI/CD pipelines:
## Troubleshooting
-### Database connection errors:
-- Ensure PostgreSQL is running
-- Verify test database exists: `psql -l | grep codeforge_test`
-- Check environment variables
-
### Ollama connection errors:
- Ensure Ollama is running: `curl http://localhost:11434/api/tags`
- Check TEST_OLLAMA_HOST environment variable
diff --git a/tests/TESTING_GUIDE.md b/tests/TESTING_GUIDE.md
index 29a04c9..f165abe 100644
--- a/tests/TESTING_GUIDE.md
+++ b/tests/TESTING_GUIDE.md
@@ -21,9 +21,8 @@ This guide provides comprehensive information about testing Claude OS, a product
### Prerequisites
-1. **PostgreSQL** with pgvector extension
-2. **Python 3.11+** with pytest
-3. **Test database** (automatically created by test runner)
+1. **Python 3.11+** with pytest
+2. **Ollama** running locally (for integration tests)
### Run All Tests
@@ -230,17 +229,6 @@ chmod +x .git/hooks/pre-commit
## 🐛 Troubleshooting
-### Database Connection Errors
-
-```bash
-# Check PostgreSQL is running
-pg_isready
-
-# Create test database manually
-createdb codeforge_test
-psql -d codeforge_test -c "CREATE EXTENSION vector"
-```
-
### Ollama Connection Errors
```bash
diff --git a/tests/TEST_COVERAGE.md b/tests/TEST_COVERAGE.md
index 317ef4e..6159c40 100644
--- a/tests/TEST_COVERAGE.md
+++ b/tests/TEST_COVERAGE.md
@@ -6,24 +6,7 @@ Comprehensive test suite for Claude OS RAG system covering all critical componen
## Test Files
-### 1. `test_pg_vector.py` - PostgreSQL Vector Operations
-**Purpose**: Verify pgvector functionality and vector similarity search
-
-**Tests**:
-- ✅ Vector storage and retrieval
-- ✅ Vector similarity queries with cosine distance
-- ✅ KB filtering in vector queries
-- ✅ PostgresManager.query_similar() method
-- ✅ Empty query results handling
-- ✅ Vector dimension validation
-- ✅ Embedding format conversions
-- ✅ Embedding roundtrip (store and retrieve)
-
-**Coverage**: Vector database operations, similarity search, data integrity
-
----
-
-### 2. `test_rag_engine.py` - RAG Engine
+### 1. `test_rag_engine.py` - RAG Engine
**Purpose**: Test RAG query execution and response generation
**Tests**:
@@ -87,23 +70,6 @@ Comprehensive test suite for Claude OS RAG system covering all critical componen
---
-### 6. `test_pg_manager.py` - PostgreSQL Manager
-**Purpose**: Test database operations and schema
-
-**Tests**:
-- ✅ Create collection
-- ✅ List collections
-- ✅ Delete collection
-- ✅ Get collection statistics
-- ✅ Add document
-- ✅ Connection pooling
-- ✅ Schema validation (tables exist)
-- ✅ pgvector extension installed
-
-**Coverage**: Database operations, connection management, schema integrity
-
----
-
## Test Categories (Markers)
### Unit Tests (`@pytest.mark.unit`)
@@ -197,11 +163,6 @@ Tests are designed to run in CI/CD pipelines:
```yaml
# GitHub Actions example
-- name: Setup test database
- run: |
- createdb codeforge_test
- psql -d codeforge_test -c "CREATE EXTENSION vector"
-
- name: Run tests
run: |
pytest --cov=app --cov-report=xml
diff --git a/tests/test_health.py b/tests/test_health.py
index f7700d2..c6dc912 100644
--- a/tests/test_health.py
+++ b/tests/test_health.py
@@ -7,8 +7,7 @@ from unittest.mock import patch, MagicMock
from app.core.health import (
check_ollama_health,
- check_postgres_health,
- check_chroma_health,
+ check_sqlite_health,
wait_for_services
)
@@ -134,12 +133,12 @@ class TestOllamaHealth:
@pytest.mark.unit
-class TestPostgresHealth:
- """Test PostgreSQL health check functionality."""
+class TestSQLiteHealth:
+ """Test SQLite health check functionality."""
@patch('app.core.health.get_sqlite_manager')
- def test_postgres_healthy(self, mock_get_db):
- """Test PostgreSQL health check when service is healthy."""
+ def test_sqlite_healthy(self, mock_get_db):
+ """Test SQLite health check when service is healthy."""
# Mock successful database connection
mock_db = MagicMock()
mock_db.list_collections.return_value = [
@@ -148,73 +147,57 @@ class TestPostgresHealth:
]
mock_get_db.return_value = mock_db
- result = check_postgres_health()
+ result = check_sqlite_health()
assert result["status"] == "healthy"
assert result["collections"] == 2
@patch('app.core.health.get_sqlite_manager')
- def test_postgres_unhealthy(self, mock_get_db):
- """Test PostgreSQL health check when service is unhealthy."""
+ def test_sqlite_unhealthy(self, mock_get_db):
+ """Test SQLite health check when service is unhealthy."""
# Mock database connection error
mock_get_db.side_effect = Exception("Connection failed")
- result = check_postgres_health()
+ result = check_sqlite_health()
assert result["status"] == "unhealthy"
assert "error" in result
assert "Connection refused" in result["error"]
@patch('app.core.health.get_sqlite_manager')
- def test_postgres_empty_database(self, mock_get_db):
- """Test PostgreSQL health check with empty database."""
+ def test_sqlite_empty_database(self, mock_get_db):
+ """Test SQLite health check with empty database."""
# Mock empty database
mock_db = MagicMock()
mock_db.list_collections.return_value = []
mock_get_db.return_value = mock_db
- result = check_postgres_health()
+ result = check_sqlite_health()
assert result["status"] == "healthy"
assert result["collections"] == 0
@patch('app.core.health.get_sqlite_manager')
- def test_postgres_list_collections_error(self, mock_get_db):
- """Test PostgreSQL health check when list_collections fails."""
+ def test_sqlite_list_collections_error(self, mock_get_db):
+ """Test SQLite health check when list_collections fails."""
# Mock list_collections error
mock_db = MagicMock()
mock_db.list_collections.side_effect = Exception("Query failed")
mock_get_db.return_value = mock_db
- result = check_postgres_health()
+ result = check_sqlite_health()
assert result["status"] == "unhealthy"
assert "error" in result
assert "Query failed" in result["error"]
-@pytest.mark.unit
-class TestChromaHealth:
- """Test Chroma health check (backwards compatibility)."""
-
- @patch('app.core.health.check_postgres_health')
- def test_chroma_health_calls_postgres(self, mock_check_pg):
- """Test that Chroma health check calls PostgreSQL health check."""
- mock_check_pg.return_value = {"status": "healthy", "collections": 5}
-
- result = check_chroma_health()
-
- # Should call PostgreSQL health check
- mock_check_pg.assert_called_once()
- assert result == {"status": "healthy", "collections": 5}
-
-
@pytest.mark.unit
class TestWaitForServices:
"""Test wait_for_services functionality."""
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_immediate_success(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test waiting for services when they're immediately healthy."""
@@ -232,7 +215,7 @@ class TestWaitForServices:
mock_pg_check.assert_called_once()
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_eventual_success(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test waiting for services that become healthy after retries."""
@@ -254,7 +237,7 @@ class TestWaitForServices:
assert mock_pg_check.call_count == 3
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_timeout(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test waiting for services that never become healthy."""
@@ -272,11 +255,11 @@ class TestWaitForServices:
assert mock_pg_check.call_count == 3
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_partial_failure(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test waiting when one service is always unhealthy."""
- # Mock Ollama always fails, PostgreSQL succeeds
+ # Mock Ollama always fails, SQLite succeeds
mock_ollama_check.return_value = {"status": "unhealthy"}
mock_pg_check.return_value = {"status": "healthy"}
@@ -288,7 +271,7 @@ class TestWaitForServices:
assert mock_pg_check.call_count == 3
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_default_parameters(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test wait_for_services with default parameters."""
@@ -306,7 +289,7 @@ class TestWaitForServices:
mock_sleep.assert_not_called()
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_logging(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test that wait_for_services logs progress."""
@@ -330,11 +313,11 @@ class TestWaitForServices:
assert "Waiting for services" in log_messages
assert "Attempt 1/2" in log_messages
assert "Ollama=unhealthy" in log_messages
- assert "PostgreSQL=healthy" in log_messages
+ assert "SQLite=healthy" in log_messages
assert "All services healthy" in log_messages
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_error_during_check(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test wait_for_services when health check raises exception."""
@@ -350,7 +333,7 @@ class TestWaitForServices:
assert mock_ollama_check.call_count == 2
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_custom_delay(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test wait_for_services with custom delay."""
@@ -368,7 +351,7 @@ class TestWaitForServices:
mock_sleep.assert_called_once_with(0.5)
@patch('app.core.health.check_ollama_health')
- @patch('app.core.health.check_postgres_health')
+ @patch('app.core.health.check_sqlite_health')
@patch('app.core.health.time.sleep')
def test_wait_for_services_custom_retries(self, mock_sleep, mock_pg_check, mock_ollama_check):
"""Test wait_for_services with custom retry count."""
@@ -432,8 +415,8 @@ class TestHealthIntegration:
assert isinstance(model["name"], str)
@patch('app.core.health.get_sqlite_manager')
- def test_real_postgres_check_structure(self, mock_get_db):
- """Test that PostgreSQL check has correct structure."""
+ def test_real_sqlite_check_structure(self, mock_get_db):
+ """Test that SQLite check has correct structure."""
# Mock realistic database
mock_db = MagicMock()
mock_db.list_collections.return_value = [
@@ -443,7 +426,7 @@ class TestHealthIntegration:
]
mock_get_db.return_value = mock_db
- result = check_postgres_health()
+ result = check_sqlite_health()
# Check response structure
assert "status" in result