"""Unit tests for Contract Lifecycle Schedule Detail model."""

from abacus_contract.models.contract_lifecycle_schedule_detail\
    import ContractLifecycleScheduleDetail
from tests.utils.factories import ContractLifecycleScheduleDetailFactory


def test_get_contract_lifecycle_schedule_detail_by_ids():
    """Test get Contract Lifecycle Schedule Detail by IDs."""
    mock_contract_lifecycle_schedule_details =\
        ContractLifecycleScheduleDetailFactory.create_batch(2)

    schedule_detail_ids = [
        schedule_detail.contract_lifecycle_schedule_detail_id
        for schedule_detail in mock_contract_lifecycle_schedule_details
    ]
    period_intervals = [
        schedule_detail.period_interval
        for schedule_detail in mock_contract_lifecycle_schedule_details
    ]
    period_types = [
        schedule_detail.period_type
        for schedule_detail in mock_contract_lifecycle_schedule_details
    ]

    res = ContractLifecycleScheduleDetail.get_by_ids(schedule_detail_ids)

    assert len(res) == 2
    assert res[0].contract_lifecycle_schedule_detail_id == schedule_detail_ids[0]
    assert res[0].period_interval == period_intervals[0]
    assert res[0].period_type == period_types[0]
    assert res[1].contract_lifecycle_schedule_detail_id == schedule_detail_ids[1]
    assert res[1].period_interval == period_intervals[1]
    assert res[1].period_type == period_types[1]