Initial upload: Local AI Commit Reviewer CLI with CI/CD workflow
This commit is contained in:
54
tests/unit/test_llm.py
Normal file
54
tests/unit/test_llm.py
Normal file
@@ -0,0 +1,54 @@
|
||||
import pytest
|
||||
from unittest.mock import MagicMock, patch
|
||||
from src.llm.provider import LLMProvider, LLMResponse, ModelInfo
|
||||
|
||||
|
||||
class MockLLMProvider(LLMProvider):
|
||||
def __init__(self, available: bool = True):
|
||||
self._available = available
|
||||
self._models = []
|
||||
|
||||
def is_available(self) -> bool:
|
||||
return self._available
|
||||
|
||||
def generate(self, prompt: str, **kwargs) -> LLMResponse:
|
||||
return LLMResponse(
|
||||
text="Mock review response",
|
||||
model="mock-model",
|
||||
tokens_used=100,
|
||||
finish_reason="stop"
|
||||
)
|
||||
|
||||
async def agenerate(self, prompt: str, **kwargs) -> LLMResponse:
|
||||
return self.generate(prompt, **kwargs)
|
||||
|
||||
def stream_generate(self, prompt: str, **kwargs):
|
||||
yield "Mock"
|
||||
|
||||
def list_models(self) -> list[ModelInfo]:
|
||||
return self._models
|
||||
|
||||
def health_check(self) -> bool:
|
||||
return self._available
|
||||
|
||||
|
||||
class TestLLMProvider:
|
||||
def test_mock_provider_is_available(self):
|
||||
provider = MockLLMProvider(available=True)
|
||||
assert provider.is_available() is True
|
||||
|
||||
def test_mock_provider_not_available(self):
|
||||
provider = MockLLMProvider(available=False)
|
||||
assert provider.is_available() is False
|
||||
|
||||
def test_mock_generate(self):
|
||||
provider = MockLLMProvider()
|
||||
response = provider.generate("test prompt")
|
||||
assert isinstance(response, LLMResponse)
|
||||
assert response.text == "Mock review response"
|
||||
assert response.model == "mock-model"
|
||||
|
||||
def test_mock_list_models(self):
|
||||
provider = MockLLMProvider()
|
||||
models = provider.list_models()
|
||||
assert isinstance(models, list)
|
||||
Reference in New Issue
Block a user