"""Test for TransferWise Batch model."""

from datetime import datetime

from collaborator.models.rds.transferwise_batch import TransferwiseBatch


def test_to_dict():
    """Test getting dictionary of model fields."""
    item = TransferwiseBatch(
        transferwise_batch_id=1,
        vendor_id=123,
        profile_id=444,
        batch_id="abcd-1234",
        version=1,
        batch_name="batch test name",
        amount=10.0,
        currency="USD",
        wire_fee=2.0,
        transferwise_fee=1.0,
        status="Pending",
        batch_type="Payment",
        created_date=datetime(2024, 1, 15, 10, 30, 0),
    )

    assert item.to_dict() == {
        "id": 1,
        "vendor_id": 123,
        "profile_id": 444,
        "batch_id": "abcd-1234",
        "version": 1,
        "batch_name": "batch test name",
        "amount": 10.0,
        "currency": "USD",
        "wire_fee": 2.0,
        "transferwise_fee": 1.0,
        "status": "Pending",
        "type": "Payment",
        "reference": None,
        "payin_name": None,
        "bank_code": None,
        "account_number": None,
        "account_type": None,
        "iban": None,
        "bban": None,
        "bank_address_name": None,
        "bank_address_branch_name": None,
        "bank_address_first_line": None,
        "bank_address_post_code": None,
        "bank_address_city": None,
        "bank_address_state_code": None,
        "bank_address_country": None,
        "tw_address_name": None,
        "tw_address_first_line": None,
        "tw_address_post_code": None,
        "tw_address_city": None,
        "tw_address_state_code": None,
        "tw_address_country": None,
        "created_date": "2024-01-15T10:30:00",
    }