BLDX-460: Fixed iterative pagination for User and Role caches

Author: Aryamanz29

pyatlan/cache/aio/group_cache.py

@@ -82,13 +82,11 @@ async def refresh_cache(self) -> None:
 
     async def _refresh_cache(self) -> None:
         async with self.lock:
-            groups = await self.client.group.get_all()
+            groups = [group async for group in await self.client.group.get_all()]
             if not groups:
                 return
-            # Process response using shared logic - extract records from response
-            group_list = groups.records or []
             (self.map_id_to_name, self.map_name_to_id, self.map_alias_to_id) = (
-                GroupCacheCommon.refresh_cache_data(group_list)
+                GroupCacheCommon.refresh_cache_data(groups)
             )
 
     async def _get_id_for_name(self, name: str) -> Optional[str]:

pyatlan/cache/aio/user_cache.py

@@ -74,13 +74,11 @@ async def refresh_cache(self) -> None:
 
     async def _refresh_cache(self) -> None:
         async with self.lock:
-            users = await self.client.user.get_all()
+            users = [user async for user in await self.client.user.get_all()]
             if not users:
                 return
-            # Process response using shared logic - extract records from response
-            user_list = users.records or []
             (self.map_id_to_name, self.map_name_to_id, self.map_email_to_id) = (
-                UserCacheCommon.refresh_cache_data(user_list)
+                UserCacheCommon.refresh_cache_data(users)
             )
 
     async def _get_id_for_name(self, name: str) -> Optional[str]:

pyatlan/cache/group_cache.py

@@ -64,13 +64,11 @@ def validate_aliases(self, aliases: Iterable[str]):
 
     def _refresh_cache(self) -> None:
         with self.lock:
-            groups = self.client.group.get_all()
+            groups = [group for group in self.client.group.get_all()]
             if not groups:
                 return
-            # Process response using shared logic - extract records from response
-            group_list = groups.records or []
             (self.map_id_to_name, self.map_name_to_id, self.map_alias_to_id) = (
-                GroupCacheCommon.refresh_cache_data(group_list)
+                GroupCacheCommon.refresh_cache_data(groups)
             )
 
     def _get_id_for_name(self, name: str) -> Optional[str]:

pyatlan/cache/user_cache.py

@@ -64,13 +64,11 @@ def validate_names(self, names: Iterable[str]):
 
     def _refresh_cache(self) -> None:
         with self.lock:
-            users = self.client.user.get_all()
+            users = [user for user in self.client.user.get_all()]
             if not users:
                 return
-            # Process response using shared logic - extract records from response
-            user_list = users.records or []
             (self.map_id_to_name, self.map_name_to_id, self.map_email_to_id) = (
-                UserCacheCommon.refresh_cache_data(user_list)
+                UserCacheCommon.refresh_cache_data(users)
             )
 
     def _get_id_for_name(self, name: str) -> Optional[str]: