yujunjun/datahub
2
0
Fork 0
mirror of https://github.com/datahub-project/datahub.git synced 2025-12-12 18:47:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(smoke-test): fix flakiness of audit smoke test (#13429)

Browse Source
This commit is contained in:
RyanHolstien 2025-05-06 11:15:05 -05:00 committed by GitHub
parent e9b867ddd8
commit 2dfe84afe3
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
smoke-test/tests/audit_events/audit_events_test.py
View File

@ -306,7 +306,7 @@ def test_policy_events(auth_exclude_filter):
wait_for_writes_to_sync(consumer_group="datahub-usage-event-consumer-job-client") wait_for_writes_to_sync(consumer_group="datahub-usage-event-consumer-job-client")
res_data = searchForAuditEvents( res_data = searchForAuditEvents(
user_session, user_session,
2, 3,
["CreatePolicyEvent", "UpdatePolicyEvent"], ["CreatePolicyEvent", "UpdatePolicyEvent"],
["urn:li:corpuser:datahub", "urn:li:corpuser:admin"], ["urn:li:corpuser:datahub", "urn:li:corpuser:admin"],
[], [],
@ -314,11 +314,19 @@ def test_policy_events(auth_exclude_filter):
print(res_data) print(res_data)
assert res_data assert res_data
assert res_data["usageEvents"] 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]["eventType"] == "UpdatePolicyEvent"
assert res_data["usageEvents"][0]["entityUrn"] == new_urn 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"][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() user_session.cookies.clear()