"""Reference Transaction Type marshmallow schema tests.""" from abacus_contract.schemas.reference_transaction_type \ import ReferenceTransactionTypeSchema from tests.utils.factories import ReferenceTransactionTypeFactory def test_reference_transaction_type_schema(): """Test reference transaction type schema.""" txn_type = ReferenceTransactionTypeFactory.create( transaction_type_name='Rip Van Winkle', transaction_type_code='ZZZ', ) res = ReferenceTransactionTypeSchema().dump(txn_type) assert res['txn_type_code'] == txn_type.transaction_type_code assert res['txn_type_id'] == txn_type.reference_transaction_type_id assert res['txn_type_name'] == txn_type.transaction_type_name