"""Test legacy_contract schema."""

from abacus_contract.schemas import LegacyContractSchema
from tests.utils.factories import LegacyContractFactory


def test_legacy_contract_schema():
    """Test legacy_contract schema."""
    legacy_contract = LegacyContractFactory.create()
    result = LegacyContractSchema().dump(legacy_contract)

    assert result['legacy_contract_id'] == legacy_contract.legacy_contract_id
    assert result['contract_id'] == legacy_contract.contract_id
    assert result['oa_contract_id'] == legacy_contract.oa_contract_id