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

from unittest.mock import patch

from feed_sender.util import sentry


@patch(
    'feed_sender.util.sentry.raven.Client',
    return_value='sentry_client'
)
@patch('feed_sender.util.sentry.os')
def test_sentry_enabled(mock_os, mock_raven_client):
    """Test sentry.sentry_client is setup when config.SENTRY is set."""
    mock_os.environ = {'SENTRY_DSN': 'test_dsn'}
    sentry.get_client()
    mock_raven_client.assert_called_with('test_dsn')


@patch('raven.Client', return_value='sentry_client')
@patch('feed_sender.util.sentry.os')
def test_sentry_disabled(mock_os, mock_raven_client):
    """Test sentry.sentry_client initialized with dsn of None."""
    mock_os.environ = {'SENTRY_DSN': None}
    sentry.get_client()
    # last call is None
    mock_raven_client.assert_called_with(None)