Add Django model tests (#812)

Author: thekaveman

tests/pytest/core/test_models.py

@@ -1,22 +1,231 @@
 import pytest
 
+from benefits.core.models import EligibilityType, EligibilityVerifier, TransitAgency
+
+
+@pytest.mark.django_db
+def test_PemData_str(model_PemData):
+    assert str(model_PemData) == model_PemData.label
+
+
+@pytest.mark.django_db
+def test_EligibilityType_str(model_EligibilityType):
+    assert str(model_EligibilityType) == model_EligibilityType.label
+
+
+@pytest.mark.django_db
+def test_EligibilityType_get_matching(model_EligibilityType):
+    eligibility = EligibilityType.get(model_EligibilityType.id)
+
+    assert eligibility == model_EligibilityType
+
+
+@pytest.mark.django_db
+def test_EligibilityType_get_nonmatching():
+    with pytest.raises(EligibilityType.DoesNotExist):
+        EligibilityType.get(99999)
+
+
+@pytest.mark.django_db
+def test_EligibilityType_get_many_matching(model_EligibilityType):
+    new_type = EligibilityType.get(model_EligibilityType.id)
+    new_type.pk = None
+    new_type.save()
+
+    result = EligibilityType.get_many([model_EligibilityType.id, new_type.id])
+
+    assert len(result) == 2
+    assert model_EligibilityType in result
+    assert new_type in result
+
+
+@pytest.mark.django_db
+def test_EligibilityType_get_many_nonmatching():
+    result = EligibilityType.get_many([99998, 99999])
+
+    assert len(result) == 0
+
+
+@pytest.mark.django_db
+def test_EligibilityType_get_many_somematching(model_EligibilityType):
+    result = EligibilityType.get_many([model_EligibilityType.id, 99999])
+
+    assert len(result) == 1
+    assert model_EligibilityType in result
+
 
 @pytest.mark.django_db
-def test_EligibilityVerifier_with_auth_verification(model_EligibilityVerifier, model_AuthProvider_with_verification):
+def test_EligibilityVerifier_str(model_EligibilityVerifier):
+    assert str(model_EligibilityVerifier) == model_EligibilityVerifier.name
+
+
+@pytest.mark.django_db
+def test_EligibilityVerifier_by_id_matching(model_EligibilityVerifier):
+    verifier = EligibilityVerifier.by_id(model_EligibilityVerifier.id)
+
+    assert verifier == model_EligibilityVerifier
+
+
+@pytest.mark.django_db
+def test_EligibilityVerifier_by_id_nonmatching():
+    with pytest.raises(EligibilityVerifier.DoesNotExist):
+        EligibilityVerifier.by_id(99999)
+
+
+@pytest.mark.django_db
+def test_EligibilityVerifier_with_AuthProvider_with_verification(
+    model_EligibilityVerifier, model_AuthProvider_with_verification
+):
     model_EligibilityVerifier.auth_provider = model_AuthProvider_with_verification
 
+    assert model_EligibilityVerifier.is_auth_required
     assert model_EligibilityVerifier.uses_auth_verification
 
 
 @pytest.mark.django_db
-def test_EligibilityVerifier_without_auth_verification(model_EligibilityVerifier, model_AuthProvider_without_verification):
+def test_EligibilityVerifier_with_AuthProvider_without_verification(
+    model_EligibilityVerifier, model_AuthProvider_without_verification
+):
     model_EligibilityVerifier.auth_provider = model_AuthProvider_without_verification
 
+    assert model_EligibilityVerifier.is_auth_required
     assert not model_EligibilityVerifier.uses_auth_verification
 
 
 @pytest.mark.django_db
 def test_EligibilityVerifier_without_AuthProvider(model_EligibilityVerifier):
     model_EligibilityVerifier.auth_provider = None
 
+    assert not model_EligibilityVerifier.is_auth_required
     assert not model_EligibilityVerifier.uses_auth_verification
+
+
+@pytest.mark.django_db
+def test_PaymentProcessor_str(model_PaymentProcessor):
+    assert str(model_PaymentProcessor) == model_PaymentProcessor.name
+
+
+@pytest.mark.django_db
+def test_TransitAgency_str(model_TransitAgency):
+    assert str(model_TransitAgency) == model_TransitAgency.long_name
+
+
+@pytest.mark.django_db
+def test_TransitAgency_get_type_id_matching(model_TransitAgency):
+    eligibility = model_TransitAgency.eligibility_types.first()
+    result = model_TransitAgency.get_type_id(eligibility.name)
+
+    assert result == eligibility.id
+
+
+@pytest.mark.django_db
+def test_TransitAgency_get_type_id_manymatching(model_TransitAgency):
+    eligibility = model_TransitAgency.eligibility_types.first()
+    new_eligibility = EligibilityType.get(eligibility.id)
+    new_eligibility.pk = None
+    new_eligibility.save()
+    model_TransitAgency.eligibility_types.add(new_eligibility)
+
+    with pytest.raises(Exception, match=r"name"):
+        model_TransitAgency.get_type_id(eligibility.name)
+
+
+@pytest.mark.django_db
+def test_TransitAgency_get_type_id_nonmatching(model_TransitAgency):
+    with pytest.raises(Exception, match=r"name"):
+        model_TransitAgency.get_type_id("something")
+
+
+@pytest.mark.django_db
+def test_TransitAgency_supports_type_matching(model_TransitAgency):
+    eligibility = model_TransitAgency.eligibility_types.first()
+
+    assert model_TransitAgency.supports_type(eligibility)
+
+
+@pytest.mark.django_db
+def test_TransitAgency_supports_type_nonmatching(model_TransitAgency):
+    eligibility = model_TransitAgency.eligibility_types.first()
+    new_eligibility = EligibilityType.get(eligibility.id)
+    new_eligibility.pk = None
+    new_eligibility.save()
+
+    assert not model_TransitAgency.supports_type(new_eligibility)
+
+
+@pytest.mark.django_db
+def test_TransitAgency_supports_type_wrongtype(model_TransitAgency):
+    eligibility = model_TransitAgency.eligibility_types.first()
+
+    assert not model_TransitAgency.supports_type(eligibility.name)
+
+
+@pytest.mark.django_db
+def test_TransitAgency_types_to_verify(model_TransitAgency):
+    eligibility = model_TransitAgency.eligibility_types.first()
+    new_eligibility = EligibilityType.get(eligibility.id)
+    new_eligibility.pk = None
+    new_eligibility.save()
+
+    assert eligibility != new_eligibility
+
+    model_TransitAgency.eligibility_types.add(new_eligibility)
+    assert model_TransitAgency.eligibility_types.count() == 2
+
+    verifier = model_TransitAgency.eligibility_verifiers.first()
+    assert verifier.eligibility_type == eligibility
+
+    result = model_TransitAgency.types_to_verify(verifier)
+    assert len(result) == 1
+    assert eligibility in result
+
+
+@pytest.mark.django_db
+def test_TransitAgency_index_url(model_TransitAgency):
+    result = model_TransitAgency.index_url
+
+    assert result.endswith(model_TransitAgency.slug)
+
+
+@pytest.mark.django_db
+def test_TransitAgency_by_id_matching(model_TransitAgency):
+    result = TransitAgency.by_id(model_TransitAgency.id)
+
+    assert result == model_TransitAgency
+
+
+@pytest.mark.django_db
+def test_TransitAgency_by_id_nonmatching():
+    with pytest.raises(TransitAgency.DoesNotExist):
+        TransitAgency.by_id(99999)
+
+
+@pytest.mark.django_db
+def test_TransitAgency_by_slug_matching(model_TransitAgency):
+    result = TransitAgency.by_slug(model_TransitAgency.slug)
+
+    assert result == model_TransitAgency
+
+
+@pytest.mark.django_db
+def test_TransitAgency_by_slug_nonmatching():
+    result = TransitAgency.by_slug("nope")
+
+    assert not result
+
+
+@pytest.mark.django_db
+def test_TransitAgency_all_active(model_TransitAgency):
+    assert TransitAgency.objects.count() == 1
+
+    inactive_agency = TransitAgency.by_id(model_TransitAgency.id)
+    inactive_agency.pk = None
+    inactive_agency.active = False
+    inactive_agency.save()
+
+    assert TransitAgency.objects.count() == 2
+
+    result = TransitAgency.all_active()
+
+    assert len(result) == 1
+    assert model_TransitAgency in result