- Initial agent development, starting by PDF

fix:
- Fixed test discovery on subdirectories

test:
- Started tests for PDF agent
This commit is contained in:
2025-03-19 21:57:08 -03:00
parent 5238d73942
commit a7b292f437
12 changed files with 110 additions and 16 deletions

View File

@@ -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")