"""Sentry module unit tests."""

from unittest.mock import patch

import sentry


@patch('sentry.AwsLambdaIntegration', return_value='integration')
@patch('sentry.sentry_sdk.init', return_value='sentry_client')
@patch('sentry.config')
def test_sentry_enabled(mock_config, mock_raven_client, mock_integration):
    """Test sentry.sentry_client is setup when config.SENTRY is set."""
    mock_config.SENTRY_DSN = 'SENTRY'
    mock_config.ENVIRONMENT = 'TEST'
    sentry_client = sentry.init_client()

    mock_raven_client.assert_called_once_with(
        dsn=mock_config.SENTRY_DSN,
        environment=mock_config.ENVIRONMENT,
        integrations=[
            mock_integration(),
        ],

    )
    assert sentry_client is None  # we don't need client with sentry-sdk