"""Test for Transaction model."""

from datetime import date, datetime
from decimal import Decimal

from collaborator.constants import transaction as constants
from collaborator.models.rds.transaction import Transaction


def test_to_dict():
    """Test converting a Transaction object to a dictionary."""
    item = Transaction(
        transaction_id=1,
        collaborator_id=1,
        date=date(2019, 9, 8),
        transaction_type=constants.TYPE_REVENUE,
        description="Manual revenue",
        original_amount=Decimal(123),
        collaborator_share=0.813,
        chargeable_amount=Decimal(100),
        transferwise_transaction_id=None,
        report_id=None,
        created_date=datetime(2019, 9, 8, 7, 6, 5),
        voided_transaction_id=None,
        currency="USD",
        statement_period_id=1,
    )

    assert item.to_dict() == {
        "id": 1,
        "collaborator_id": 1,
        "date": "2019-09-08",
        "type": constants.TYPE_REVENUE,
        "description": "Manual revenue",
        "original_amount": 123.0,
        "collaborator_share": 0.813,
        "chargeable_amount": 100.0,
        "transferwise_transaction_id": None,
        "report_id": None,
        "created_date": "2019-09-08T07:06:05",
        "voided_transaction_id": None,
        "credited_payment_id": None,
        "currency": "USD",
        "deleted_date": None,
        "current_balance": 0,
        "statement_period_id": 1,
        "creation_batch_uuid": None,
    }