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

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

Browse Source
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
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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