"""Tests for reference_agreement_type model."""

from abacus_account.models.reference_agreement_type import ReferenceAgreementType


def test_get_reference_agreement_types():
    """Test to get reference_agreement_type's."""
    reference_agreement_types = ReferenceAgreementType.query.all()
    assert len(reference_agreement_types) == 7

    expected = {
        'AWAL Core',
        'AWAL +',
        'AWAL Recordings',
        'Courage',
        'Mass Appeal',
        'Relapse',
        'O&O',
    }
    assert {t.agreement_type for t in reference_agreement_types} == expected