"""Test contract_exclusion schema."""

from abacus_contract.schemas import ContractExclusionSchema
from abacus_contract.tests.utils.factories import ContractExclusionFactory


def test_contract_exclusion_schema():
    """Test contract_exclusion schema."""
    contract_exclusion = ContractExclusionFactory.create()
    res = ContractExclusionSchema().dump(contract_exclusion)

    assert res['contract_exclusion_id'] == contract_exclusion.contract_exclusion_id
    assert res['contract_id'] == contract_exclusion.contract_id
    assert res['exclusions'] == contract_exclusion.exclusions