"""Reference VAT Rate logic tests."""

from unittest.mock import patch

from ledger.logic import reference_vat_rate as logic
from ledger.schemas.reference_vat_rate import ReferenceVatRateListSchema

from tests.utils.factories import ReferenceVatRateFactory


@patch('ledger.logic.reference_vat_rate.ReferenceVatRate')
def test_get_reference_vat_rates(mock_model):
    """Test get_reference_vat_rates method."""
    entry = ReferenceVatRateFactory.create()
    mock_model_response = [entry]
    mock_model.get_all.return_value = mock_model_response
    expected_response = {
        'items': ReferenceVatRateListSchema().dump(mock_model_response, many=True),
        'total_count': 1,
    }
    res = logic.get_reference_vat_rates()

    mock_model.get_all.assert_called_once()
    assert res.status == 200
    assert res.message == expected_response