"""Unit tests for reference_mechanical_rate model."""

from abacus_contract.models import ReferenceMechanicalRate


def test_create_reference_mechanical_rate():
    """Test creating a reference_mechanical_rate."""
    args = {
        'country_code': 'USA',
        'base_rate': '0.1240',
        'minute_rate': '0.0238',
        'ringtone_rate': '0.2400',
        'effective_start_date': '2004-01-01 00:00:00'
    }
    ReferenceMechanicalRate.create(**args)
    res = ReferenceMechanicalRate.query.all()

    assert len(res) == 1
    assert res[0].country_code == args['country_code']
    assert str(res[0].base_rate) == args['base_rate']
    assert str(res[0].minute_rate) == args['minute_rate']
    assert str(res[0].ringtone_rate) == args['ringtone_rate']
    assert str(res[0].effective_start_date) == args['effective_start_date']