fix(smoke): prevent audit events smoke flake (#13944)

This commit is contained in:
RyanHolstien 2025-07-02 13:00:17 -05:00 committed by GitHub
parent 77a5671256
commit 87c8378b12
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -19,6 +19,8 @@ os.environ["DATAHUB_TELEMETRY_ENABLED"] = "false"
(admin_user, admin_pass) = get_admin_credentials()
previous_policy_urn = ""
@pytest.fixture()
def auth_exclude_filter():
@ -325,12 +327,17 @@ def test_policy_events(auth_exclude_filter):
or res_data["usageEvents"][1]["eventType"] == "UpdatePolicyEvent"
)
assert res_data["usageEvents"][1]["entityUrn"] == new_urn
global previous_policy_urn
if len(res_data["usageEvents"]) == 3:
assert (
res_data["usageEvents"][2]["eventType"] == "CreatePolicyEvent"
or res_data["usageEvents"][2]["eventType"] == "UpdatePolicyEvent"
)
assert res_data["usageEvents"][2]["entityUrn"] == new_urn
assert (
res_data["usageEvents"][2]["entityUrn"] == new_urn
or res_data["usageEvents"][2]["entityUrn"] == previous_policy_urn
)
previous_policy_urn = new_urn
user_session.cookies.clear()
@ -537,12 +544,17 @@ def test_policy_create_delete(auth_exclude_filter):
or res_data["usageEvents"][1]["eventType"] == "DeletePolicyEvent"
)
assert res_data["usageEvents"][1]["entityUrn"] == new_urn
global previous_policy_urn
if len(res_data["usageEvents"]) == 3:
assert (
res_data["usageEvents"][2]["eventType"] == "CreatePolicyEvent"
or res_data["usageEvents"][2]["eventType"] == "DeletePolicyEvent"
)
assert res_data["usageEvents"][2]["entityUrn"] == new_urn
assert (
res_data["usageEvents"][2]["entityUrn"] == new_urn
or res_data["usageEvents"][2]["entityUrn"] == previous_policy_urn
)
previous_policy_urn = new_urn
user_session.cookies.clear()