"""Unit Tests for contract_term_schedule model."""

from abacus_contract.models.contract_term_schedule import ContractTermSchedule
from tests.utils.factories import ContractTermFactory
from tests.utils.factories import ContractTermScheduleFactory


def test_create_contract_term_schedule(create_mock_schedule):
    """Test to create contract_term_schedule record."""
    mock_contract_term = ContractTermFactory.create()
    mock_contract_term_schedule = ContractTermSchedule.create(
        contract_term_id=mock_contract_term.contract_term_id,
        schedule_id=1
    )
    result = ContractTermSchedule.query.all()
    assert len(result) == 1
    assert result[0] == mock_contract_term_schedule


def test_get_by_term_id(create_mock_schedule):
    """Test to get contract term schedules by contract_term_id."""
    mock_contract_term = ContractTermFactory.create()
    contract_term_id = mock_contract_term.contract_term_id
    ContractTermScheduleFactory.create(
        contract_term=mock_contract_term,
        schedule_id=1,
        deleted_by='test_user',
        deleted_at='2023-08-18',
    )
    ContractTermScheduleFactory.create(
        contract_term=mock_contract_term,
        schedule_id=2
    )

    items = ContractTermSchedule.get_by_term_id(contract_term_id)
    assert len(items) == 1
    assert items[0].schedule_id == 2