"""PayeeKycNotification model tests."""

from datetime import timedelta

from payee.models.payee_kyc_notification import PayeeKycNotification
from payee.utils.models import object_as_dict
from tests.utils.factories import PayeeKycNotificationFactory


def test_get_latest_by_payee_id(payee_fixtures):
    """Check get_latest_by_payee_id."""
    payee_id = 1

    now = PayeeKycNotification.current_timestamp()
    PayeeKycNotificationFactory.create(payee_id=payee_id, created_at=now)
    notification = PayeeKycNotificationFactory.create(
        payee_id=payee_id,
        created_at=now + timedelta(seconds=1),
    )
    PayeeKycNotificationFactory.create(
        payee_id=payee_id,
        created_at=now + timedelta(seconds=2),
        deleted_at=PayeeKycNotification.current_timestamp(),
        deleted_by='test_user_id',
    )

    result = PayeeKycNotification.get_latest_by_payee_id(payee_id)
    assert object_as_dict(result) == object_as_dict(notification)


def test_get_latest_by_payee_ids(payee_fixtures):
    """Check get_latest_by_payee_ids."""
    payee_ids = [1]

    now = PayeeKycNotification.current_timestamp()
    PayeeKycNotificationFactory.create(payee_id=payee_ids[0], created_at=now)
    notification = PayeeKycNotificationFactory.create(
        payee_id=payee_ids[0],
        created_at=now + timedelta(seconds=1),
    )
    PayeeKycNotificationFactory.create(
        payee_id=payee_ids[0],
        created_at=now + timedelta(seconds=2),
        deleted_at=PayeeKycNotification.current_timestamp(),
        deleted_by='test_user_id',
    )

    result = PayeeKycNotification.get_latest_by_payee_ids(payee_ids)
    assert [object_as_dict(result[0])] == [object_as_dict(notification)]