update organization filter test case

the-deep · Jan 8, 2024 · de28ff5 · de28ff5
1 parent 2ab83db
commit de28ff5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/apps/assessment_registry/filter_set.py b/apps/assessment_registry/filter_set.py
@@ -38,6 +38,12 @@ def filter_focuses(self, qs, _, value):
     def filter_sectors(self, qs, _, value):
         return qs if value is None else qs.filter(sectors__overlap=value)
 
+    def filter_details_type(self, qs, _, value):
+        return qs if value is None else qs.filter(details_type__overlap=value)
+
+    def filter_coordinated_joint(self, qs, _, value):
+        return qs if value is None else qs.filter(coordinated_joint__overlap=value)
+
 
 AssessmentDashboardFilterDataType, AssessmentDashboardFilterDataInputType = generate_type_for_filter_set(
     AssessmentDashboardFilterSet,

diff --git a/apps/organization/tests/test_schemas.py b/apps/organization/tests/test_schemas.py
@@ -42,12 +42,12 @@ def test_organization_query(self):
             query MyQuery (
                 $verified: Boolean
                 $search: String
-                $usedInProject: ID
+                $usedInProjectByLead: ID
             ) {
               organizations(
                   search: $search
                   verified: $verified
-                  usedInProject: $usedInProject
+                  usedInProjectByLead: $usedInProjectByLead
                   ordering: DESC_TITLE
               ) {
                 results {
@@ -96,10 +96,10 @@ def test_organization_query(self):
                     'verified': False,
                 }, [org6, org5, org3, org1]),
                 (user, {
-                    'usedInProject': str(project.id),
+                    'usedInProjectByLead': str(project.id),
                 }, [org6, org5, org3, org2, org1]),
                 (non_member_user, {
-                    'usedInProject': str(project.id),
+                    'usedInProjectByLead': str(project.id),
                     # Return all the organizations (Filter not applied)
                 }, [org6, org5, org4, org3, org2, org1]),
         ]: