"""Unit Tests for legacy_contract model."""

from abacus_contract.models.legacy_contract import LegacyContract
from tests.utils.factories import ContractFactory
from tests.utils.factories import LegacyContractFactory


def test_create_legacy_contract():
    """Test to create a legacy_contract."""
    assert LegacyContract.count() == 0
    contract = ContractFactory.create()
    vendor_contract_id = 1
    LegacyContract.create(
        contract_id=contract.contract_id,
        oa_contract_id=vendor_contract_id
    )
    result = LegacyContract.query.all()
    assert len(result) == 1
    assert result[0].contract_id == contract.contract_id
    assert result[0].oa_contract_id == vendor_contract_id


def test_get_by_contract_id():
    """Test to get a legacy_contract by contract_id."""
    contract = ContractFactory.create()
    vendor_contract_id = 1
    LegacyContractFactory.create(
        contract=contract,
        oa_contract_id=vendor_contract_id
    )
    result = LegacyContract.get_by_contract_id(contract.contract_id)
    assert result.contract_id == contract.contract_id
    assert result.oa_contract_id == vendor_contract_id