"""Test for TransferWise Profile model."""

from datetime import datetime

from collaborator.constants.transferwise_profile import TransferwiseProfileStatus
from collaborator.models.rds.transferwise_profile import TransferwiseProfile


def test_to_dict():
    """Test getting dictionary of model fields."""
    item = TransferwiseProfile(
        transferwise_profile_id=123,
        vendor_id=24601,
        subaccount_id=None,
        profile_id=1,
        access_token="letmein",
        refresh_token="rememberme",
        refresh_token_created_at=datetime(2024, 1, 15, 10, 30, 0),
        status=TransferwiseProfileStatus.UNVERIFIED,
        active=True,
    )

    assert item.to_dict() == {
        "id": 123,
        "vendor_id": 24601,
        "subaccount_id": None,
        "profile_id": 1,
        "access_token": "letmein",
        "refresh_token": "rememberme",
        "refresh_token_created_at": "2024-01-15T10:30:00",
        "status": TransferwiseProfileStatus.UNVERIFIED,
        "active": True,
        "deleted_by": None,
        "deleted_date": None,
    }