From 2dfe84afe3b9d77cab09581d7094e3a1c9a0c834 Mon Sep 17 00:00:00 2001 From: RyanHolstien Date: Tue, 6 May 2025 11:15:05 -0500 Subject: [PATCH] fix(smoke-test): fix flakiness of audit smoke test (#13429) --- smoke-test/tests/audit_events/audit_events_test.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/smoke-test/tests/audit_events/audit_events_test.py b/smoke-test/tests/audit_events/audit_events_test.py index 8a390d9a36..3cffb918f6 100644 --- a/smoke-test/tests/audit_events/audit_events_test.py +++ b/smoke-test/tests/audit_events/audit_events_test.py @@ -306,7 +306,7 @@ def test_policy_events(auth_exclude_filter): wait_for_writes_to_sync(consumer_group="datahub-usage-event-consumer-job-client") res_data = searchForAuditEvents( user_session, - 2, + 3, ["CreatePolicyEvent", "UpdatePolicyEvent"], ["urn:li:corpuser:datahub", "urn:li:corpuser:admin"], [], @@ -314,11 +314,19 @@ def test_policy_events(auth_exclude_filter): print(res_data) assert res_data assert res_data["usageEvents"] - assert len(res_data["usageEvents"]) == 2 + assert len(res_data["usageEvents"]) == 3 assert res_data["usageEvents"][0]["eventType"] == "UpdatePolicyEvent" assert res_data["usageEvents"][0]["entityUrn"] == new_urn - assert res_data["usageEvents"][1]["eventType"] == "CreatePolicyEvent" + assert ( + res_data["usageEvents"][1]["eventType"] == "CreatePolicyEvent" + or res_data["usageEvents"][1]["eventType"] == "UpdatePolicyEvent" + ) assert res_data["usageEvents"][1]["entityUrn"] == new_urn + assert ( + res_data["usageEvents"][2]["eventType"] == "CreatePolicyEvent" + or res_data["usageEvents"][2]["eventType"] == "UpdatePolicyEvent" + ) + assert res_data["usageEvents"][2]["entityUrn"] == new_urn user_session.cookies.clear()