"""Test suite for TransferwiseTransaction logic."""

from unittest.mock import patch

from collaborator.logic import transferwise_transaction


@patch("collaborator.logic.transferwise_transaction.TransferwiseTransactionPersister")
def test_get_payments(mockTWPersister):
    """Test get_collaborator_payments."""
    mockTWPersister.get_collaborator_payments.return_value = [{"id": 12}, {"id": 34}], 2

    result = transferwise_transaction.get_payments(
        vendor_id=25153,
        collaborator_id=1,
        payment_statuses=["incoming_payment_waiting"],
        limit=5,
        offset=10,
        sort_key="id",
        sort_direction="ASC",
    )

    mockTWPersister.get_collaborator_payments.assert_called_with(
        25153, 1, ["incoming_payment_waiting"], 5, 10, "id", "ASC"
    )

    assert result.message == {
        "items": [{"id": 12}, {"id": 34}],
        "pagination": {"total_records": 2, "type": "standard"},
    }