BLDX-608 | Fixed flaky tests - SSO group mapping update API now required name

Author: Aryamanz29

Dockerfile

@@ -118,11 +118,11 @@ try:
     import pyatlan
     print(f"✅ PyAtlan imported successfully")
     print(f"Installed version: {pyatlan.__version__}")
-    
+
     # Try importing AtlanClient
     from pyatlan.client.atlan import AtlanClient
     print(f"✅ AtlanClient imported successfully")
-    
+
     # Version validation (skip if 'latest' was requested)
     if expected_version != "latest" and expected_version != "unknown":
         if pyatlan.__version__ != expected_version:
@@ -134,9 +134,9 @@ try:
             print("✅ Version verified")
     else:
         print("✅ Version validation skipped (latest or unknown)")
-        
+
     print("=== Build verification passed ===")
-    
+
 except ImportError as e:
     print(f"❌ Import error: {e}")
     sys.exit(1)

HISTORY.md

@@ -15,7 +15,7 @@
 - Updated Docker docs to reference Harbor pulls.
 - Added local Snyk scan instructions.
 - Limited uv resolution to CPython and bump urllib3 to fix CVEs.
-  
+
 ## 8.5.1 (January 29, 2026)
 
 ### Bug Fixes
@@ -26,7 +26,7 @@
 
 - Enhanced SDK client retry mechanism to automatically retry HTTP `302` redirect responses, improving reliability when handling temporary redirects.
 - Updated admin integration tests to align with latest backend API changes.
-  
+
 
 ## 8.5.0 (January 14, 2026)
 

pyatlan/client/aio/sso.py

@@ -78,6 +78,7 @@ async def update_group_mapping(
         sso_alias: str,
         atlan_group: AtlanGroup,
         group_map_id: str,
+        group_map_name: str,
         sso_group_name: str,
     ) -> SSOMapper:
         """
@@ -86,12 +87,13 @@ async def update_group_mapping(
         :param sso_alias: name of the SSO provider.
         :param atlan_group: existing Atlan group.
         :param group_map_id: existing SSO group map identifier.
+        :param group_map_name: existing SSO group map name.
         :param sso_group_name: new SSO group name.
         :raises AtlanError: on any error during API invocation.
         :returns: updated SSO group mapping instance.
         """
         endpoint, request_obj = SSOUpdateGroupMapping.prepare_request(
-            sso_alias, atlan_group, group_map_id, sso_group_name
+            sso_alias, atlan_group, group_map_id, group_map_name, sso_group_name
         )
         raw_json = await self._client._call_api(endpoint, request_obj=request_obj)
         return SSOUpdateGroupMapping.process_response(raw_json)

pyatlan/client/common/sso.py

@@ -92,6 +92,7 @@ def prepare_request(
         sso_alias: str,
         atlan_group: AtlanGroup,
         group_map_id: str,
+        group_map_name: str,
         sso_group_name: str,
     ) -> tuple:
         """
@@ -100,6 +101,7 @@ def prepare_request(
         :param sso_alias: name of the SSO provider
         :param atlan_group: existing Atlan group
         :param group_map_id: existing SSO group map identifier
+        :param group_map_name: existing SSO group map name
         :param sso_group_name: new SSO group name
         :returns: tuple of (endpoint, request_obj)
         """
@@ -111,9 +113,9 @@ def prepare_request(
             attribute_value=sso_group_name,
         )  # type: ignore[call-arg]
 
-        # NOTE: Updates don't require a group map name; group map ID works fine
         group_mapper = SSOMapper(
             id=group_map_id,
+            name=group_map_name,
             config=group_mapper_config,
             identity_provider_alias=sso_alias,
             identity_provider_mapper=IDP_GROUP_MAPPER,

pyatlan/client/sso.py

@@ -70,6 +70,7 @@ def update_group_mapping(
         sso_alias: str,
         atlan_group: AtlanGroup,
         group_map_id: str,
+        group_map_name: str,
         sso_group_name: str,
     ) -> SSOMapper:
         """
@@ -78,12 +79,13 @@ def update_group_mapping(
         :param sso_alias: name of the SSO provider.
         :param atlan_group: existing Atlan group.
         :param group_map_id: existing SSO group map identifier.
+        :param group_map_name: existing SSO group map name.
         :param sso_group_name: new SSO group name.
         :raises AtlanError: on any error during API invocation.
         :returns: updated SSO group mapping instance.
         """
         endpoint, request_obj = SSOUpdateGroupMapping.prepare_request(
-            sso_alias, atlan_group, group_map_id, sso_group_name
+            sso_alias, atlan_group, group_map_id, group_map_name, sso_group_name
         )
         raw_json = self._client._call_api(endpoint, request_obj=request_obj)
         return SSOUpdateGroupMapping.process_response(raw_json)

tests/integration/aio/test_sso_client.py

@@ -92,9 +92,10 @@ def _assert_sso_group_mapping(
     assert sso_mapping.config.sync_mode == GROUP_MAPPER_SYNC_MODE
     assert sso_mapping.config.attribute_name == GROUP_MAPPER_ATTRIBUTE
     if is_updated:
-        assert sso_mapping.name is None
+        assert sso_mapping.name
         assert sso_mapping.config.attribute_value == SSO_GROUP_NAME_UPDATED
     else:
+        assert sso_mapping.name
         assert group.id and (group.id in str(sso_mapping.name))
         assert sso_mapping.config.attribute_value == SSO_GROUP_NAME
 
@@ -189,11 +190,13 @@ async def test_update_group_mapping(
     assert group
     assert sso_mapping
     assert sso_mapping.id
+    assert sso_mapping.name
 
     updated_mapping = await client.sso.update_group_mapping(
         sso_alias=AtlanSSO.JUMPCLOUD,
         atlan_group=group,
         group_map_id=sso_mapping.id,
+        group_map_name=sso_mapping.name,
         sso_group_name=SSO_GROUP_NAME_UPDATED,
     )
     _assert_sso_group_mapping(group, updated_mapping, True)

tests/integration/test_sso_client.py

@@ -92,9 +92,10 @@ def _assert_sso_group_mapping(
     assert sso_mapping.config.sync_mode == GROUP_MAPPER_SYNC_MODE
     assert sso_mapping.config.attribute_name == GROUP_MAPPER_ATTRIBUTE
     if is_updated:
-        assert sso_mapping.name is None
+        assert sso_mapping.name
         assert sso_mapping.config.attribute_value == SSO_GROUP_NAME_UPDATED
     else:
+        assert sso_mapping.name
         assert group.id and (group.id in str(sso_mapping.name))
         assert sso_mapping.config.attribute_value == SSO_GROUP_NAME
 
@@ -187,11 +188,13 @@ def test_update_group_mapping(
     assert group
     assert sso_mapping
     assert sso_mapping.id
+    assert sso_mapping.name
 
     updated_mapping = client.sso.update_group_mapping(
         sso_alias=AtlanSSO.JUMPCLOUD,
         atlan_group=group,
         group_map_id=sso_mapping.id,
+        group_map_name=sso_mapping.name,
         sso_group_name=SSO_GROUP_NAME_UPDATED,
     )
     _assert_sso_group_mapping(group, updated_mapping, True)

tests/unit/aio/test_sso_client.py

@@ -274,6 +274,7 @@ async def test_sso_update_group_mapping(
         sso_alias="auth0",
         atlan_group=AtlanGroup(),
         group_map_id="group-map-id",
+        group_map_name="group-map-name",
         sso_group_name="sso-group",
     )
     assert response == SSOMapper(**update_group_mapping_json)

tests/unit/test_sso_client.py

@@ -265,6 +265,7 @@ def test_sso_update_group_mapping(mock_api_caller, update_group_mapping_json):
         sso_alias="auth0",
         atlan_group=AtlanGroup(),
         group_map_id="group-map-id",
+        group_map_name="group-map-name",
         sso_group_name="sso-group",
     )
     assert response == SSOMapper(**update_group_mapping_json)