21 lines
843 B
Python
21 lines
843 B
Python
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
from git_commit_ai.core.ollama_client import generate_commit_message
|
|
|
|
def test_generate_commit_message_success():
|
|
"""Test successful commit message generation."""
|
|
mock_response = MagicMock()
|
|
mock_response.json.return_value = {'response': 'fix: resolve bug'}
|
|
mock_response.raise_for_status = MagicMock()
|
|
|
|
with patch('requests.post', return_value=mock_response):
|
|
result = generate_commit_message("test prompt")
|
|
assert result == 'fix: resolve bug'
|
|
|
|
def test_generate_commit_message_connection_error():
|
|
"""Test connection error handling."""
|
|
with patch('requests.post') as mock_post:
|
|
mock_post.side_effect = Exception("Connection failed")
|
|
with pytest.raises(ConnectionError):
|
|
generate_commit_message("test prompt")
|