feat:
- Initial agent development, starting by PDF fix: - Fixed test discovery on subdirectories test: - Started tests for PDF agent
This commit is contained in:
@@ -3,13 +3,14 @@ from unittest.mock import patch, MagicMock
|
||||
|
||||
from src.service.ollama.ollamaModelManager import OllamaModelManager
|
||||
|
||||
|
||||
class TestOllamaModelManager(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
self.model_manager = OllamaModelManager()
|
||||
self.model_manager.base_url = "http://test-url:11434"
|
||||
|
||||
@patch('service.ollama.ollamaModelManager.requests.get')
|
||||
@patch('src.service.ollama.ollamaModelManager.requests.get')
|
||||
def test_get_available_models_success(self, mock_get):
|
||||
# Setup mock response
|
||||
mock_response = MagicMock()
|
||||
@@ -29,7 +30,7 @@ class TestOllamaModelManager(unittest.TestCase):
|
||||
self.assertEqual(len(result), 2)
|
||||
self.assertEqual(result, [{'name': 'model1'}, {'name': 'model2'}])
|
||||
|
||||
@patch('service.ollama.ollamaModelManager.requests.get')
|
||||
@patch('src.service.ollama.ollamaModelManager.requests.get')
|
||||
def test_get_available_models_exception(self, mock_get):
|
||||
# Setup mock to raise exception
|
||||
mock_get.side_effect = Exception("Connection error")
|
||||
@@ -40,7 +41,7 @@ class TestOllamaModelManager(unittest.TestCase):
|
||||
# Assertions
|
||||
self.assertEqual(result, [])
|
||||
|
||||
@patch('service.ollama.ollamaModelManager.requests.post')
|
||||
@patch('src.service.ollama.ollamaModelManager.requests.post')
|
||||
def test_get_model_details_success(self, mock_post):
|
||||
# Setup mock response
|
||||
mock_response = MagicMock()
|
||||
@@ -63,7 +64,7 @@ class TestOllamaModelManager(unittest.TestCase):
|
||||
self.assertEqual(result['context_size'], 4096)
|
||||
self.assertEqual(result['license'], 'Apache 2.0')
|
||||
|
||||
@patch('service.ollama.ollamaModelManager.requests.post')
|
||||
@patch('src.service.ollama.ollamaModelManager.requests.post')
|
||||
def test_get_model_details_exception(self, mock_post):
|
||||
# Setup mock to raise exception
|
||||
mock_post.side_effect = Exception("API error")
|
||||
|
||||
Reference in New Issue
Block a user