"""Unit tests for reference_mechanical_rate schema."""

from abacus_common_logic.utils import dates

from abacus_contract.schemas.reference_mechanical_rate \
    import ReferenceMechanicalRateSchema
from tests.utils.factories import ReferenceMechanicalRateFactory


def test_reference_mechanical_rate_schema():
    """Test reference_mechanical_rate schema."""
    mock_reference_mechanical_rate = ReferenceMechanicalRateFactory.create()
    res = ReferenceMechanicalRateSchema().dump(mock_reference_mechanical_rate)

    assert res == {
        'reference_mechanical_rate_id':
            mock_reference_mechanical_rate.reference_mechanical_rate_id,
        'country_code': mock_reference_mechanical_rate.country_code,
        'base_rate': str(mock_reference_mechanical_rate.base_rate),
        'minute_rate': str(mock_reference_mechanical_rate.minute_rate),
        'ringtone_rate': str(mock_reference_mechanical_rate.ringtone_rate),
        'effective_start_date': dates.safe_format_datetime(
            mock_reference_mechanical_rate.effective_start_date
        )
    }