"""Tests for the Sentry Connector."""

from unittest.mock import patch

from sales.connectors import sentry


@patch('sales.connectors.sentry.raven.Client',
       return_value='sentry_client')
@patch('sales.connectors.sentry.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'
    sentry_client = sentry.get_client()
    assert sentry.sentry_client
    mock_raven_client.assert_called_with('SENTRY_DSN')
    assert sentry_client == 'sentry_client'


@patch('raven.Client', return_value='sentry_client')
@patch('sales.connectors.sentry.config')
def test_sentry_disabled(mock_config, mock_raven_client):
    """Test sentry.sentry_client initialized with dsn of None."""
    mock_config.SENTRY_DSN = None
    sentry_client = sentry.get_client()
    # last call is None
    mock_raven_client.assert_called_with(None)
    # sentry_client is returned
    assert sentry_client == 'sentry_client'