"""Tests for Reference Agreement type logic."""

from unittest.mock import patch

from abacus_account.logic import reference_agreement_type as logic
from abacus_account.schemas import ReferenceAgreementTypeDetailSchema
from tests.utils.factories import ReferenceAgreementTypeFactory


@patch('abacus_account.logic.reference_agreement_type.models')
def test_get_reference_agreement_types(mock_model):
    """Test get_agreement_types function."""
    reference_agreement_type = ReferenceAgreementTypeFactory.build()
    mock_model.ReferenceAgreementType.query.all.return_value = [
        reference_agreement_type
    ]

    res = logic.get_agreement_types()

    assert res.status == 200
    assert res.message == [
        ReferenceAgreementTypeDetailSchema().dump(reference_agreement_type)
    ]