import importlib
from unittest.mock import patch

from masters_registry.connectors import sentry


@patch('raven.Client', return_value='sentry_client')
@patch('masters_registry.config')
def test_sentry_enabled(mock_config, mock_raven_client):
    """Test sentry.sentry_client is setup when config.SENTRY_DSN is set.
    """

    mock_config.SENTRY_DSN = 'SENTRY_DSN'
    importlib.reload(sentry)
    assert sentry.sentry_client
    mock_raven_client.assert_called_with(
        'SENTRY_DSN'
    )
    assert sentry.sentry_client == 'sentry_client'


@patch('raven.Client', return_value='sentry_client')
@patch('masters_registry.config')
def test_sentry_disabled(mock_config, mock_raven_client):
    """Test sentry.sentry_client is not setup if config.SENTRY_DSN is not set.
    """

    mock_config.SENTRY_DSN = None
    importlib.reload(sentry)
    assert not sentry.sentry_client
    assert mock_raven_client.call_count == 0