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