"""Unit tests for contract_term_schedule schemas."""

from abacus_contract.schemas.contract_term_schedule import \
    ContractTermScheduleDetailSchema
from tests.utils.factories import ContractTermFactory
from tests.utils.factories import ContractTermScheduleFactory


def test_contract_term_schedule_detail_schema(create_mock_schedule):
    """Test ContractTermScheduleDetailSchema."""
    mock_contract_term = ContractTermFactory.create()
    mock_contract_term_schedule = ContractTermScheduleFactory.create(
        contract_term=mock_contract_term
    )
    res = ContractTermScheduleDetailSchema().dump(mock_contract_term_schedule)

    assert res == {
        'contract_term_schedule_id':
            mock_contract_term_schedule.contract_term_schedule_id,
        'contract_term_id': mock_contract_term_schedule.contract_term_id,
        'schedule_id': mock_contract_term_schedule.schedule_id
    }