From 296ba4423cf767224c61cd96f315e6bec2c64564 Mon Sep 17 00:00:00 2001 From: Paulina Kujawa Date: Fri, 31 Jan 2025 15:01:29 +0100 Subject: [PATCH] fix tests - messages query, rdi query --- src/hct_mis_api/apps/core/backends.py | 1 + .../unit/apps/accountability/test_list_query_messages.py | 8 ++------ .../test_registration_data_import_query.py | 8 ++++++-- 3 files changed, 9 insertions(+), 8 deletions(-) diff --git a/src/hct_mis_api/apps/core/backends.py b/src/hct_mis_api/apps/core/backends.py index 2db986f86a..0c1635f34e 100644 --- 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 index ccca647775..2f1f5b5910 100644 --- 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 index 4fa2daf7d7..2763464701 100644 --- 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},