"""Unit tests for reference_payment_type handler."""

from payment.logic import reference_payment_type as logic
from tests.utils.factories import ReferencePaymentTypeFactory


def test_dataload_by_ids():
    """Test dataload_by_ids method."""
    reference_payment_type_list = ReferencePaymentTypeFactory.create_batch(8)

    items_ids = (2, 3, 5, 7)

    result = logic.dataload_by_ids(
        [reference_payment_type_list[i].reference_payment_type_id for i in items_ids]
    )

    assert result == {
        'items': [{'data': reference_payment_type_list[i]} for i in items_ids]
    }