add tests

Author: saxix

tests/admin/test_admin_cleaners.py

@@ -0,0 +1,22 @@
+from unittest import mock
+from unittest.mock import Mock
+
+import pytest
+
+from country_workspace.workspaces.admin.cleaners.validate import validate_program
+
+
+@pytest.fixture
+def household():
+    from testutils.factories import CountryHouseholdFactory, ProgramFactory
+
+    program = ProgramFactory()
+    return CountryHouseholdFactory(batch__program=program, batch__country_office=program.country_office)
+
+
+def test_validate_program(household):
+    assert validate_program(Mock(program=household.program)) == {"invalid": 0, "total": 0, "valid": 1}
+    with mock.patch(
+        "country_workspace.workspaces.admin.cleaners.validate.Household.validate_with_checker", return_value=False
+    ):
+        assert validate_program(Mock(program=household.program)) == {"invalid": 1, "total": 0, "valid": 0}

tests/models/test_m_household.py

@@ -1,9 +1,13 @@
 from typing import TYPE_CHECKING
+from unittest import mock
+from unittest.mock import Mock
 
 import pytest
 
+
 if TYPE_CHECKING:
-    from country_workspace.workspaces.models import CountryHousehold
+    from country_workspace.workspaces.models import CountryHousehold, CountryIndividual
+    from country_workspace.models import Household
 
 
 @pytest.fixture
@@ -13,6 +17,22 @@ def household() -> "CountryHousehold":
     return CountryHouseholdFactory()
 
 
+@pytest.fixture
+def individual(household) -> "CountryIndividual":
+    from testutils.factories import CountryIndividualFactory
+
+    return CountryIndividualFactory(household=household)
+
+
 def test_properties(household: "CountryHousehold"):
     assert household.program == household.batch.program
     assert household.country_office == household.batch.country_office
+
+
+def test_validate_with_checker(individual: "CountryHousehold"):
+    household: Household = individual.household
+    assert household.validate_with_checker()
+    assert household.errors == {}
+    with mock.patch.object(household.program.beneficiary_validator, "validate", Mock(return_value=["Error"])):
+        assert not household.validate_with_checker()
+        assert household.errors == {"dct": ["Error"]}

tests/workspace/actions/test_ws_validate.py

@@ -4,7 +4,6 @@
 import freezegun
 import pytest
 from django.urls import reverse
-
 from testutils.utils import select_office
 
 from country_workspace.state import state