Merge pull request #71 from openimis/feature/wait-for-mutation

Feature/wait for mutation

Author: dragos-dobre

payroll/schema.py

@@ -5,6 +5,7 @@
 from django.db.models import Q, Sum
 
 from core.schema import OrderedDjangoFilterConnectionField
+from core.services import wait_for_mutation
 from core.utils import append_validity_filter
 from invoice.gql.gql_types.bill_types import BillGQLType
 from invoice.models import Bill
@@ -120,6 +121,7 @@ def resolve_bill_by_payroll(self, info, **kwargs):
 
         client_mutation_id = kwargs.get("client_mutation_id", None)
         if client_mutation_id:
+            wait_for_mutation(client_mutation_id)
             filters.append(Q(mutations__mutation__client_mutation_id=client_mutation_id))
 
         subject_type = kwargs.get("subject_type", None)
@@ -142,6 +144,7 @@ def resolve_benefit_consumption_by_payroll(self, info, **kwargs):
 
         client_mutation_id = kwargs.get("client_mutation_id", None)
         if client_mutation_id:
+            wait_for_mutation(client_mutation_id)
             filters.append(Q(mutations__mutation__client_mutation_id=client_mutation_id))
 
         return gql_optimizer.query(BenefitConsumption.objects.filter(*filters), info)
@@ -156,6 +159,7 @@ def resolve_benefit_attachment_by_payroll(self, info, **kwargs):
 
         client_mutation_id = kwargs.get("client_mutation_id", None)
         if client_mutation_id:
+            wait_for_mutation(client_mutation_id)
             filters.append(Q(mutations__mutation__client_mutation_id=client_mutation_id))
 
         return gql_optimizer.query(BenefitAttachment.objects.filter(*filters), info)
@@ -166,6 +170,7 @@ def resolve_payment_point(self, info, **kwargs):
 
         client_mutation_id = kwargs.get("client_mutation_id")
         if client_mutation_id:
+            wait_for_mutation(client_mutation_id)
             filters.append(Q(mutations__mutation__client_mutation_id=client_mutation_id))
 
         parent_location = kwargs.get('parent_location')
@@ -181,6 +186,7 @@ def resolve_payroll(self, info, **kwargs):
 
         client_mutation_id = kwargs.get("client_mutation_id")
         if client_mutation_id:
+            wait_for_mutation(client_mutation_id)
             filters.append(Q(mutations__mutation__client_mutation_id=client_mutation_id))
 
         query = Payroll.objects.filter(*filters)
@@ -192,6 +198,7 @@ def resolve_payroll_benefit_consumption(self, info, **kwargs):
 
         client_mutation_id = kwargs.get("client_mutation_id")
         if client_mutation_id:
+            wait_for_mutation(client_mutation_id)
             filters.append(Q(mutations__mutation__client_mutation_id=client_mutation_id))
 
         benefit_plan_name = kwargs.get("benefitPlanName")
@@ -216,6 +223,7 @@ def resolve_benefit_consumption(self, info, **kwargs):
 
         client_mutation_id = kwargs.get("client_mutation_id")
         if client_mutation_id:
+            wait_for_mutation(client_mutation_id)
             filters.append(Q(mutations__mutation__client_mutation_id=client_mutation_id))
 
         query = BenefitConsumption.objects.filter(*filters)