Fix failing test.

tests/integration/custom_metadata_test.py

@@ -33,7 +33,10 @@
 from tests.integration.admin_test import create_group, delete_group
 from tests.integration.client import TestId, delete_asset
 from tests.integration.glossary_test import create_glossary, create_term
-from tests.integration.utils import wait_for_successful_custometadatadef_purge
+from tests.integration.utils import (
+    wait_for_successful_custometadatadef_purge,
+    wait_for_successful_enumadef_purge,
+)
 
 MODULE_NAME = TestId.make_unique("CM")
 
@@ -310,7 +313,7 @@ def cm_enum(
 ) -> Generator[EnumDef, None, None]:
     enum_def = create_enum(client, name=DQ_ENUM, values=DQ_TYPE_LIST)
     yield enum_def
-    wait_for_successful_custometadatadef_purge(DQ_ENUM, client=client)
+    wait_for_successful_enumadef_purge(DQ_ENUM, client=client)
 
 
 def test_cm_enum(

tests/integration/utils.py

@@ -13,7 +13,7 @@
 from pyatlan.model.assets import Asset
 from pyatlan.model.enums import EntityStatus
 from pyatlan.model.response import AssetMutationResponse
-from pyatlan.model.typedef import AtlanTagDef, CustomMetadataDef
+from pyatlan.model.typedef import AtlanTagDef, CustomMetadataDef, EnumDef
 
 
 def block(
@@ -65,3 +65,12 @@ def wait_for_successful_tagdef_purge(name: str, client: AtlanClient):
 )
 def wait_for_successful_custometadatadef_purge(name: str, client: AtlanClient):
     client.typedef.purge(name, typedef_type=CustomMetadataDef)
+
+
+@retry(
+    retry=retry_if_exception_type(AtlanError),
+    wait=wait_random_exponential(multiplier=1, max=5),
+    stop=stop_after_attempt(3),
+)
+def wait_for_successful_enumadef_purge(name: str, client: AtlanClient):
+    client.typedef.purge(name, typedef_type=EnumDef)