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

from unittest import mock
from unittest.mock import MagicMock

from payee.connectors.kafka import emit_payee_event
from payee.constants.constants import PAYEE_EVENT_NAMES


@mock.patch('payee.connectors.kafka._create_payee_event_key_and_payload')
def test_emit_payee_event_oauth_complete_success(mock_create_key_payload, monkeypatch):
    """Test emit_payee_event success."""
    account_payee_id = 123
    action_name = PAYEE_EVENT_NAMES.VAT_COLLECTED
    mock_kafka_producers = {
        PAYEE_EVENT_NAMES.VAT_COLLECTED: MagicMock(produce_event=MagicMock()),
    }
    monkeypatch.setattr(
        'payee.config.Config.KAFKA_PRODUCERS_BY_ACTION_NAME', mock_kafka_producers
    )
    payload = {
        'timestamp': 'timestamp',
        'accountPayee': {'accountPayeeId': account_payee_id},
    }

    mock_create_key_payload.return_value = (
        f'account_payee_id_{account_payee_id}',
        payload,
    )

    emit_payee_event(account_payee_id, action_name)
    mock_create_key_payload.assert_called_once_with(account_payee_id)