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

from unittest.mock import MagicMock, patch

from payee.connectors import sentry


@patch('payee.connectors.sentry.SentryScrubber')
@patch('sentry_sdk.init')
def test_sentry_init_sentry_enabled(init_sentry_mock, scrubber_mock):
    """Test sentry.sentry_client is setup when config.SENTRY is set."""
    sentry_config = MagicMock()
    sentry_config.SENTRY = 'SENTRY_DSN'
    sentry_config.SENTRY_ENVIRONMENT = 'test'

    sentry.setup_sentry(sentry_config)

    init_sentry_mock.assert_called_once()
    scrubber_mock.assert_called_once()


@patch('sentry_sdk.init')
def test_sentry_init_sentry_disabled(init_sentry_mock):
    """Test sentry.sentry_client is setup when config.SENTRY is NOT set."""
    sentry_config = MagicMock()
    sentry_config.SENTRY = ''
    sentry_config.SENTRY_ENVIRONMENT = 'test'

    sentry.setup_sentry(sentry_config)

    init_sentry_mock.assert_not_called()