"""Sentry module unit tests."""

from unittest.mock import patch

from lambdacommon import sentry


@patch('sentry_sdk.init')
@patch('lambdacommon.sentry.common_config')
def test_sentry_enabled(mock_config, mock_sentry_client):
    """Test sentry.sentry_client is setup when config.SENTRY is set."""
    mock_config.SENTRY_DSN = 'SENTRY'
    sentry.init()

    mock_sentry_client.assert_called()
    assert mock_sentry_client.call_args[1]['dsn'] == 'SENTRY'


@patch('sentry_sdk.init')
@patch('lambdacommon.sentry.common_config')
def test_sentry_disabled(mock_config, mock_sentry_client):
    """Test sentry.sentry_client initialized with dsn of None."""
    mock_config.SENTRY_DSN = None
    sentry.init()

    mock_sentry_client.assert_called()
    assert mock_sentry_client.call_args[1]['dsn'] is None