"""Tests for contract_template model."""

from abacus_contract.models.contract_template import ContractTemplate
from abacus_contract.tests.utils.factories import (
    ReferenceSapProfitCenterFactory,
    ReferenceSigningEntityFactory,
    SigningEntitySapProfitCenterFactory,
)


def test_create_contract_template(create_mock_run_controller):
    """Test to create contract_template."""
    reference_signing_entity = ReferenceSigningEntityFactory.create()
    sap_profit_center = ReferenceSapProfitCenterFactory.create(
        company_code=reference_signing_entity.company_code
    )
    SigningEntitySapProfitCenterFactory.create(
        reference_signing_entity=reference_signing_entity,
        reference_sap_profit_center=sap_profit_center,
    )
    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,
        reference_sap_profit_center_id=sap_profit_center.reference_sap_profit_center_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