"""Tests for contract_template model."""

from abacus_contract.models.contract_template import ContractTemplate
from tests.utils.factories import ReferenceSigningEntityFactory


def test_create_contract_template(create_mock_run_controller):
    """Test to create contract_template."""
    reference_signing_entity = ReferenceSigningEntityFactory.create()
    reference_signing_entity_id = reference_signing_entity.reference_signing_entity_id
    ContractTemplate.create(
        template_name='Test Template 1',
        run_controller_id=1,
        reference_signing_entity_id=reference_signing_entity_id,
        term_duration_year=1,
        contract_type='distribution',
        contract_terms=[{'term_type': 'label'}],
        contract_exclusions={'countries': [], 'stores': []}
    )

    response = ContractTemplate.query.all()
    assert len(response) == 1