from unittest.mock import patch, MagicMock

from src.connectors import sentry


@patch('src.connectors.sentry.raven.Client', return_value='client')
@patch('src.connectors.sentry.config')
def test_get_client_enabled(mock_config, mock_client):
    mock_config.SENTRY = 'DSN'
    c = sentry.get_client()
    assert c == 'client'
    mock_client.assert_called_with('DSN')


@patch('src.connectors.sentry.raven.Client', return_value='client')
@patch('src.connectors.sentry.config')
def test_get_client_disabled(mock_config, mock_client):
    mock_config.SENTRY = None
    c = sentry.get_client()
    assert c == 'client'
    mock_client.assert_called_with(None)


@patch('src.connectors.sentry.sentry_client')
def test_send_response_to_sentry(mock_client):
    mock_client.__bool__.return_value = True
    resp = MagicMock(message='m', errors={'e': 'x'}, status=200)
    sentry.send_response_to_sentry(resp, 'msg')
    mock_client.captureMessage.assert_called()