"""Test payee KYC notifications logic."""

from unittest.mock import patch

from payee.logic import payee_kyc_notification as logic
from tests.utils.factories import (
    PayeeKycNotificationFactory,
)


@patch('payee.logic.payee_kyc_notification.PayeeKycNotification')
def test_get_payee_kyc_notifications_dataloader(model_mock):
    """Test getting payee KYC notifications."""

    data = [
        PayeeKycNotificationFactory.build(),
    ]

    model_mock.get_latest_by_payee_ids.return_value = data
    res = logic.get_payee_kyc_notification_dataloader([1, 2])

    model_mock.get_latest_by_payee_ids.assert_called_once_with([1, 2])
    assert res == {
        'payee_kyc_notifications': [
            {'data': data[0]},
            {'data': None},
        ]
    }