fix tests - messages query, rdi query

unicef · Jan 31, 2025 · 296ba44 · 296ba44
1 parent 3f7e68c
commit 296ba44
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 8 deletions.
diff --git a/src/hct_mis_api/apps/core/backends.py b/src/hct_mis_api/apps/core/backends.py
@@ -111,6 +111,7 @@ def get_all_permissions(self, user: User, obj: Optional[Model] = None) -> set[st
         return permissions_set
 
     def has_perm(self, user_obj: Union[User, AnonymousUser], perm: str, obj: Optional[Model] = None) -> bool:  # type: ignore
+        print(obj)
         if user_obj.is_superuser:
             return True
         if isinstance(user_obj, AnonymousUser):

diff --git a/tests/unit/apps/accountability/test_list_query_messages.py b/tests/unit/apps/accountability/test_list_query_messages.py
@@ -137,7 +137,7 @@ def setUpTestData(cls) -> None:
     def test_list_communication_messages(
         self, _: str, permissions: Sequence[str], extra_filters: Union[Callable[[User], dict], dict]
     ) -> None:
-        self.create_user_role_with_permissions(self.user, permissions, self.business_area)
+        self.create_user_role_with_permissions(self.user, permissions, self.business_area, self.program)
 
         self.snapshot_graphql_request(
             request_string=self.QUERY,
@@ -167,11 +167,7 @@ def test_list_communication_messages(
         )
     )
     def test_list_communication_message_recipients(self, _: str, permissions: Sequence[str], variables: dict) -> None:
-        self.create_user_role_with_permissions(
-            self.user,
-            permissions,
-            self.business_area,
-        )
+        self.create_user_role_with_permissions(self.user, permissions, self.business_area, self.program)
 
         self.snapshot_graphql_request(
             request_string=self.QUERY_RECIPIENTS,

diff --git a/tests/unit/apps/registration_data/test_registration_data_import_query.py b/tests/unit/apps/registration_data/test_registration_data_import_query.py
@@ -138,7 +138,9 @@ def test_registration_data_import_datahub_query_all(
         _: Any,
         permissions: List[Permissions],
     ) -> None:
-        self.create_user_role_with_permissions(self.user, permissions, self.business_area)
+        self.create_user_role_with_permissions(
+            self.user, permissions, self.business_area, whole_business_area_access=True
+        )
         self.snapshot_graphql_request(
             request_string=self.ALL_REGISTRATION_DATA_IMPORT_DATAHUB_QUERY,
             context={"user": self.user},
@@ -159,7 +161,9 @@ def test_registration_data_import_datahub_query_all(
     def test_registration_data_import_datahub_query_single_with_permission(
         self, _: Any, permissions: List[Permissions]
     ) -> None:
-        self.create_user_role_with_permissions(self.user, permissions, self.business_area)
+        self.create_user_role_with_permissions(
+            self.user, permissions, self.business_area, whole_business_area_access=True
+        )
         self.snapshot_graphql_request(
             request_string=self.REGISTRATION_DATA_IMPORT_DATAHUB_QUERY,
             context={"user": self.user},