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