ut

unicef · Dec 11, 2024 · a87f49e · a87f49e
1 parent d9353a3
commit a87f49e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/hct_mis_api/apps/reporting/services/generate_report_service.py b/src/hct_mis_api/apps/reporting/services/generate_report_service.py
@@ -26,7 +26,6 @@
 from django.template.loader import render_to_string
 
 import openpyxl
-from apps.payment.models import Payment
 from openpyxl.utils import get_column_letter
 from openpyxl.worksheet.worksheet import Worksheet
 
@@ -45,6 +44,7 @@
 )
 from hct_mis_api.apps.payment.delivery_mechanisms import DeliveryMechanismChoices
 from hct_mis_api.apps.payment.models import (
+    Payment,
     PaymentPlan,
     PaymentVerification,
     PaymentVerificationPlan,

diff --git a/tests/unit/apps/grievance/test_filter_already_existing_tickets.py b/tests/unit/apps/grievance/test_filter_already_existing_tickets.py
@@ -130,7 +130,7 @@ def setUpTestData(cls) -> None:
         cls.ticket = SensitiveGrievanceTicketWithoutExtrasFactory(
             household=cls.household_1,
             individual=cls.individuals_1[0],
-            payment_obj=cls.payment,
+            payment=cls.payment,
             ticket=grievance_1,
         )
         SensitiveGrievanceTicketWithoutExtrasFactory(
@@ -141,7 +141,7 @@ def setUpTestData(cls) -> None:
         SensitiveGrievanceTicketWithoutExtrasFactory(
             household=cls.household_3,
             individual=cls.individuals_3[0],
-            payment_obj=cls.payment_record2,
+            payment=cls.payment_record2,
             ticket=grievance_3,
         )
         GrievanceComplaintTicketFactory.create_batch(5)

diff --git a/tests/unit/apps/payment/test_verification_plan_status_change_services.py b/tests/unit/apps/payment/test_verification_plan_status_change_services.py
@@ -125,7 +125,7 @@ def setUpTestData(cls) -> None:
 
             PaymentVerificationFactory(
                 payment_verification_plan=other_payment_plan_payment_verification,
-                payment_obj=other_payment_record,
+                payment=other_payment_record,
                 status=PaymentVerification.STATUS_PENDING,
             )
             EntitlementCardFactory(household=other_household)