unittest summarizer

.gitignore

@@ -2,4 +2,6 @@
 __pycache__
 .pytest_cache
 .env*
-!.env.example
+!.env.example
+
+my_chroma_db

poetry.lock

@@ -2239,6 +2239,25 @@ tomli = {version = ">=1", markers = "python_version < \"3.11\""}
 [package.extras]
 dev = ["argcomplete", "attrs (>=19.2)", "hypothesis (>=3.56)", "mock", "pygments (>=2.7.2)", "requests", "setuptools", "xmlschema"]
 
+[[package]]
+name = "pytest-asyncio"
+version = "0.25.2"
+description = "Pytest support for asyncio"
+optional = false
+python-versions = ">=3.9"
+groups = ["dev"]
+files = [
+    {file = "pytest_asyncio-0.25.2-py3-none-any.whl", hash = "sha256:0d0bb693f7b99da304a0634afc0a4b19e49d5e0de2d670f38dc4bfa5727c5075"},
+    {file = "pytest_asyncio-0.25.2.tar.gz", hash = "sha256:3f8ef9a98f45948ea91a0ed3dc4268b5326c0e7bce73892acc654df4262ad45f"},
+]
+
+[package.dependencies]
+pytest = ">=8.2,<9"
+
+[package.extras]
+docs = ["sphinx (>=5.3)", "sphinx-rtd-theme (>=1)"]
+testing = ["coverage (>=6.2)", "hypothesis (>=5.7.1)"]
+
 [[package]]
 name = "python-dateutil"
 version = "2.9.0.post0"
@@ -3121,4 +3140,4 @@ type = ["pytest-mypy"]
 [metadata]
 lock-version = "2.1"
 python-versions = ">=3.10, <4.0"
-content-hash = "8fa4e35e6b978aa0ae1e7c3a5c6f9bb998e79491f717ebec21cca25d5eebc47f"
+content-hash = "e832a3ea167213ca280f213201124d535205b11ddadd8f4affbbdf0431a78906"

pyproject.toml

@@ -28,6 +28,7 @@ packages = [
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^8.3.4"
+pytest-asyncio = "^0.25.2"
 
 [build-system]
 requires = ["poetry-core>=2.0.0,<3.0.0"]

tests/test_summarizer.py

@@ -0,0 +1,86 @@
+import pytest
+import tempfile
+from unittest.mock import Mock, patch, AsyncMock
+from pathlib import Path
+from pydantic_ai import Agent
+from know_lang_bot.code_parser.summarizer import CodeSummarizer
+from know_lang_bot.code_parser.parser import CodeChunk, ChunkType
+from know_lang_bot.config import AppConfig
+
+@pytest.fixture
+def config():
+    """Create a test configuration"""
+    with tempfile.TemporaryDirectory() as temp_dir:
+        yield AppConfig(
+            llm={"model_name": "test-model", "model_provider": "test"},
+            db={"persist_directory": Path(temp_dir), "collection_name": "test_collection"}
+        )
+
+@pytest.fixture
+def sample_chunks():
+    """Create sample code chunks for testing"""
+    return [
+        CodeChunk(
+            type=ChunkType.FUNCTION,
+            content="def hello(): return 'world'",
+            start_line=1,
+            end_line=2,
+            file_path="test.py",
+            name="hello",
+            docstring="Says hello"
+        ),
+        CodeChunk(
+            type=ChunkType.CLASS,
+            content="class TestClass:\n    def __init__(self):\n        pass",
+            start_line=4,
+            end_line=6,
+            file_path="test.py",
+            name="TestClass",
+            docstring="A test class"
+        )
+    ]
+
+@pytest.fixture
+def mock_summary():
+    """Create a sample summary result"""
+    return "This is a test function"
+
+@pytest.fixture
+def mock_run_result(mock_summary):
+    """Create a mock run result"""
+    mock_result = Mock()
+    mock_result.data = mock_summary
+    return mock_result
+
+@pytest.mark.asyncio
+@patch('know_lang_bot.code_parser.summarizer.Agent')
+async def test_summarize_chunk(mock_agent_class, config: AppConfig, sample_chunks: list[CodeChunk], mock_run_result: Mock):
+    """Test summarizing a single chunk"""
+    # Setup the mock agent instance
+    mock_agent = mock_agent_class.return_value
+    mock_agent.run = AsyncMock(return_value=mock_run_result)
+
+    summarizer = CodeSummarizer(config)
+    result = await summarizer.summarize_chunk(sample_chunks[0])
+    
+    # Verify result
+    assert isinstance(result, str)
+    assert result == mock_run_result.data
+    
+    # Verify agent was called with correct prompt
+    call_args = mock_agent.run.call_args[0][0]
+    assert "def hello()" in call_args
+    assert "Says hello" in call_args
+
+@patch('know_lang_bot.code_parser.summarizer.Agent')
+def test_chromadb_initialization(mock_agent_class, config: AppConfig):
+    """Test ChromaDB initialization"""
+    mock_agent = mock_agent_class.return_value
+    
+    summarizer = CodeSummarizer(config)
+    assert summarizer.collection is not None
+    
+    # Verify we can create a new collection
+    summarizer.db_client.delete_collection(config.db.collection_name)
+    new_summarizer = CodeSummarizer(config)
+    assert new_summarizer.collection is not None