"""Tests for contract_exclusion model."""

from abacus_contract.models.contract_exclusion import ContractExclusion
from tests.utils.factories import ContractFactory


def test_create_contract_exclusion():
    """Test to create country record for contract."""
    contract = ContractFactory.create()
    exclusion = ContractExclusion.create(
        contract_id=contract.contract_id,
        exclusions={
            'countries': ['USA', 'AFG'],
            'stores': ['Spotify']
        }
    )
    result = exclusion.query.all()
    assert len(result) == 1