"""Unit tests for reference_transaction_type model."""

from abacus_contract.models import ReferenceTransactionType
from tests.utils.factories import ReferenceTransactionTypeFactory


def test_create_reference_transaction_type():
    """Test creating a ReferenceTransactionType."""
    args = {
        'transaction_type_name': 'Transaction Type Name',
        'transaction_type_code': 'TTN'
    }
    ReferenceTransactionType.create(**args)
    res = ReferenceTransactionType.query.all()

    assert len(res) == 1
    assert res[0].transaction_type_name == args['transaction_type_name']
    assert res[0].transaction_type_code == args['transaction_type_code']


def test_get_all():
    """Test method to get all transaction types ordered by name."""
    tt_1 = ReferenceTransactionTypeFactory.create(
        transaction_type_name='B Transaction Type'
    )
    tt_2 = ReferenceTransactionTypeFactory.create(
        transaction_type_name='A Transaction Type'
    )

    res = ReferenceTransactionType.get_all()

    assert len(res) == 2
    assert res[0] == tt_2
    assert res[1] == tt_1