Merge branch 'dev' into LTD-5763-bulk-approval

api/cases/managers.py

@@ -323,6 +323,7 @@ def search(  # noqa
                 "queues",
                 "queues__team",
                 "baseapplication__licences",
+                "flags",
                 Prefetch(
                     "baseapplication__parties",
                     to_attr="end_user_parties",

api/cases/serializers.py

@@ -33,6 +33,8 @@
 from api.compliance.serializers.ComplianceVisitCaseSerializers import ComplianceVisitSerializer
 from api.core.serializers import KeyValueChoiceField, PrimaryKeyRelatedSerializerField
 from api.documents.libraries.process_document import process_document
+from api.flags.serializers import CaseListFlagSerializer
+from api.flags.models import Flag
 from api.gov_users.serializers import GovUserSimpleSerializer
 from api.organisations.models import Organisation
 from api.organisations.serializers import TinyOrganisationViewSerializer
@@ -122,6 +124,7 @@ class CaseListSerializer(serializers.Serializer):
     intended_end_use = serializers.SerializerMethodField()
     end_users = serializers.SerializerMethodField()
     sub_status = CaseSubStatusSerializer()
+    flags = PrimaryKeyRelatedSerializerField(many=True, queryset=Flag.objects.all(), serializer=CaseListFlagSerializer)
 
     def __init__(self, *args, **kwargs):
         self.team = kwargs.pop("team", None)

api/cases/views/search/service.py

@@ -16,6 +16,7 @@
 from api.applications.serializers.advice import AdviceSearchViewSerializer
 from api.cases.models import Case, EcjuQuery, Advice
 from api.common.dates import working_days_in_range, number_of_days_since
+from api.flags.models import Flag
 from api.flags.serializers import CaseListFlagSerializer
 from api.organisations.models import Organisation
 from api.staticdata.statuses.enums import CaseStatusEnum
@@ -50,27 +51,7 @@ def get_advice_types_list():
     return AdviceType.to_representation()
 
 
-def populate_other_flags(cases: List[Dict]):
-    from api.flags.models import Flag
-
-    case_ids = [case["id"] for case in cases]
-
-    union_flags = set(
-        [
-            *Flag.objects.filter(cases__id__in=case_ids).annotate(case_id=F("cases__id")),
-            *Flag.objects.filter(organisations__cases__id__in=case_ids).annotate(case_id=F("organisations__cases__id")),
-        ]
-    )
-
-    for case in cases:
-        case_id = str(case["id"])
-        flags = [flag for flag in union_flags if str(flag.case_id) == case_id]
-        case["flags"] = CaseListFlagSerializer(flags, many=True).data
-
-
 def populate_goods_flags(cases: List[Dict]):
-    from api.flags.models import Flag
-
     case_ids = [case["id"] for case in cases]
     qs1 = Flag.objects.filter(goods__goods_on_application__application_id__in=case_ids).annotate(
         case_id=F("goods__goods_on_application__application_id"),

api/cases/tests/test_case_search.py → api/cases/views/search/tests/test_views.py

@@ -910,7 +910,7 @@ def setUp(self):
         super().setUp()
 
         self.other_team = self.create_team("other team")
-        self.other_team_gov_user = self.create_gov_user("new_user@digital.trade.gov.uk", self.other_team)
+        self.other_team_gov_user = self.create_gov_user("new_user@digital.trade.gov.uk", self.other_team)  # /PS-IGNORE
         self.queue = self.create_queue("my new queue", self.team)
         self.case = self.create_standard_application_case(self.organisation)
         self.case.queues.set([self.queue])
@@ -1011,7 +1011,7 @@ def test_api_success(self):
         self._create_data()
         self.advice = FinalAdviceFactory(user=self.gov_user, case=self.case, type=AdviceType.APPROVE, good=self.good)
         self.gov_user2 = self.create_gov_user(
-            team=self.create_team(name="other_team"), email="new_user2@digital.trade.gov.uk"
+            team=self.create_team(name="other_team"), email="new_user2@digital.trade.gov.uk"  # /PS-IGNORE
         )
         self.group_advice = FinalAdviceFactory(
             user=self.gov_user2, case=self.case, type=AdviceType.REFUSE, good=self.good
@@ -1148,12 +1148,33 @@ def test_api_success(self):
             ],
         )
 
-        # Reflect rest framework's way of rendering datetime objects... https://github.com/encode/django-rest-framework/blob/c9e7b68a4c1db1ac60e962053380acda549609f3/rest_framework/utils/encoders.py#L29
+        # Reflect rest framework's way of rendering datetime objects... https://github.com/encode/django-rest-framework/blob/c9e7b68a4c1db1ac60e962053380acda549609f3/rest_framework/utils/encoders.py#L29  /PS-IGNORE
         expected_submitted_at = self.case.submitted_at.isoformat()
         if expected_submitted_at.endswith("+00:00"):
             expected_submitted_at = expected_submitted_at[:-6] + "Z"
         self.assertEqual(case_api_result["submitted_at"], expected_submitted_at)
 
+    def test_api_multiple_cases_flags_correct(self):
+        # Create two cases..
+        self._create_data()
+        self._create_data()
+        # Add a case flag to each case..
+        flag_alias = "REFER_TO_FCDO_MEUC_CONCERNS"
+        flag = Flag.objects.get(alias=flag_alias)
+        all_cases = Case.objects.all()
+        for case in all_cases:
+            case.flags.add(flag)
+
+        # Perform the search
+        response = self.client.get(self.url, **self.gov_headers)
+        response_data = response.json()["results"]
+
+        self.assertEqual(response.status_code, status.HTTP_200_OK)
+        self.assertEqual(len(response_data["cases"]), 2)
+        for search_result in response_data["cases"]:
+            self.assertEqual(len(search_result["flags"]), 1)
+            self.assertEqual(search_result["flags"][0]["alias"], flag_alias)
+
     def test_api_no_advice(self):
         self._create_data()
         response = self.client.get(self.url, **self.gov_headers)

api/cases/views/search/views.py

@@ -59,7 +59,6 @@ def get(self, request, *args, **kwargs):
         # Populate certain fields outside of the serializer for performance improvements
         service.populate_goods_flags(cases)
         service.populate_destinations_flags(cases)
-        service.populate_other_flags(cases)
         service.populate_organisation(cases)
         service.populate_is_recently_updated(cases)
         service.populate_activity_updates(case_map)