"""Functional tests for contract_term_schedule API."""

from abacus_contract.tests.utils.factories import (
    ContractTermFactory,
    ContractTermScheduleFactory,
)


def test_get_contract_term_schedules_by_term_id(create_mock_schedule, fixture_client):
    """Test to get a list of 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)
    ContractTermScheduleFactory.create(contract_term=mock_contract_term, schedule_id=3)

    res = fixture_client.get(
        f'/contract-term/{contract_term_id}/contract-term-schedules'
    )

    assert res.status_code == 200
    assert len(res.json) == 2
    assert res.json == [
        {
            'contract_term_schedule_id': 2,
            'contract_term_id': contract_term_id,
            'schedule_id': 2,
        },
        {
            'contract_term_schedule_id': 3,
            'contract_term_id': contract_term_id,
            'schedule_id': 3,
        },
    ]