refactored test generic import

Author: johniak

tests/extras/test_utils/factories/__init__.py

@@ -8,6 +8,7 @@
     UserFactory,
 )
 from .accountability import CommunicationMessageFactory, FeedbackFactory, FeedbackMessageFactory, SurveyFactory
+from .api import APITokenFactory
 from .changelog import ChangelogFactory
 from .core import (
     BeneficiaryGroupFactory,
@@ -34,6 +35,7 @@
     AccountFactory,
     AccountTypeFactory,
     DeliveryMechanismFactory,
+    FinancialInstitutionFactory,
     FinancialServiceProviderFactory,
     FinancialServiceProviderXlsxTemplateFactory,
     FspXlsxTemplatePerDeliveryMechanismFactory,
@@ -47,11 +49,12 @@
     WesternUnionPaymentPlanReportFactory,
 )
 from .program import ProgramCycleFactory, ProgramFactory
-from .registration_data import RegistrationDataImportFactory
+from .registration_data import ImportDataFactory, RegistrationDataImportFactory
 from .steficon import RuleCommitFactory, RuleFactory
 
 __all__ = [
     "AdminAreaLimitedToFactory",
+    "APITokenFactory",
     "AreaFactory",
     "AreaTypeFactory",
     "BeneficiaryGroupFactory",
@@ -71,11 +74,13 @@
     "DocumentTypeFactory",
     "EntitlementCardFactory",
     "HouseholdFactory",
+    "ImportDataFactory",
     "IndividualFactory",
     "IndividualRoleInHouseholdFactory",
     "AccountFactory",
     "AccountTypeFactory",
     "DeliveryMechanismFactory",
+    "FinancialInstitutionFactory",
     "FinancialServiceProviderFactory",
     "FspXlsxTemplatePerDeliveryMechanismFactory",
     "FinancialServiceProviderXlsxTemplateFactory",

tests/extras/test_utils/factories/api.py

@@ -0,0 +1,21 @@
+"""API-related factories."""
+
+from django.utils import timezone
+from factory import LazyFunction
+from factory.django import DjangoModelFactory
+
+from hope.models import APIToken
+from hope.models.utils import Grant
+
+
+class APITokenFactory(DjangoModelFactory):
+    """Factory for API tokens."""
+
+    class Meta:
+        model = APIToken
+
+    # key is auto-generated by model.save() if not provided
+    allowed_ips = ""
+    grants = [Grant.API_RDI_CREATE.name]
+    valid_from = LazyFunction(timezone.now)
+    valid_to = None

tests/extras/test_utils/factories/payment.py

@@ -10,6 +10,7 @@
     Account,
     AccountType,
     DeliveryMechanism,
+    FinancialInstitution,
     FinancialServiceProvider,
     FinancialServiceProviderXlsxTemplate,
     FspXlsxTemplatePerDeliveryMechanism,
@@ -178,6 +179,14 @@ class Meta:
     name = factory.Sequence(lambda n: f"WU Invoice {n}")
 
 
+class FinancialInstitutionFactory(DjangoModelFactory):
+    class Meta:
+        model = FinancialInstitution
+
+    name = factory.Sequence(lambda n: f"Financial Institution {n}")
+    type = FinancialInstitution.FinancialInstitutionType.BANK
+
+
 class WesternUnionPaymentPlanReportFactory(DjangoModelFactory):
     class Meta:
         model = WesternUnionPaymentPlanReport

tests/extras/test_utils/factories/registration_data.py

@@ -3,13 +3,23 @@
 import factory
 from factory.django import DjangoModelFactory
 
-from hope.models import RegistrationDataImport
+from hope.models import ImportData, RegistrationDataImport
 
 from .account import UserFactory
 from .core import BusinessAreaFactory
 from .program import ProgramFactory
 
 
+class ImportDataFactory(DjangoModelFactory):
+    class Meta:
+        model = ImportData
+
+    status = ImportData.STATUS_PENDING
+    data_type = ImportData.XLSX
+    business_area_slug = ""
+    file = factory.django.FileField(filename="test_data.xlsx")
+
+
 class RegistrationDataImportFactory(DjangoModelFactory):
     class Meta:
         model = RegistrationDataImport