Skip to content

Commit

Permalink
[regression-test](case) audit log case (#47456)
Browse files Browse the repository at this point in the history
shuke987 authored Jan 29, 2025

Verified

This commit was signed with the committer’s verified signature.
zhengruifeng Ruifeng Zheng
1 parent 60910a4 commit 9e57929
Showing 1 changed file with 23 additions and 29 deletions.
52 changes: 23 additions & 29 deletions regression-test/suites/audit/test_audit_log_behavior.groovy
Original file line number Diff line number Diff line change
@@ -73,39 +73,33 @@ suite("test_audit_log_behavior") {

qt_audit_log_schema """desc internal.__internal_schema.audit_log"""

for (def on : [true, false]) {
sql "set enable_nereids_planner=${on}"
sql "truncate table __internal_schema.audit_log"
// run queries
for (int i = 0; i < cnt; i++) {
def tuple2 = sqls.get(i)
sql tuple2[0]
}
sql "truncate table __internal_schema.audit_log"
// run queries
for (int i = 0; i < cnt; i++) {
def tuple2 = sqls.get(i)
sql tuple2[0]
}

if (on == true) {
// only new planner supports call flush_audit_log
// make sure audit event is created.
// see WorkloadRuntimeStatusMgr.getQueryNeedAudit()
Thread.sleep(6000)
sql """call flush_audit_log()"""
}
// check result
for (int i = 0; i < cnt; i++) {
def tuple2 = sqls.get(i)
def retry = 180
def query = "select stmt from __internal_schema.audit_log where stmt like 'insert%3F6B9A_${i}%' order by time asc limit 1"
def res = sql "${query}"
while (res.isEmpty()) {
if (retry-- < 0) {
logger.warn("It has retried a few but still failed, you need to check it")
return
}
sleep(1000)
res = sql "${query}"
Thread.sleep(6000)
sql """call flush_audit_log()"""

// check result
for (int i = 0; i < cnt; i++) {
def tuple2 = sqls.get(i)
def retry = 180
def query = "select stmt from __internal_schema.audit_log where stmt like 'insert%3F6B9A_${i}%' order by time asc limit 1"
def res = sql "${query}"
while (res.isEmpty()) {
if (retry-- < 0) {
logger.warn("It has retried a few but still failed, you need to check it")
return
}
assertEquals(res[0][0].toString(), tuple2[1].toString())
sleep(1000)
res = sql "${query}"
}
assertEquals(res[0][0].toString(), tuple2[1].toString())
}

// do not turn off
sql "set global enable_audit_plugin = false"
sql "set global audit_plugin_max_sql_length = 4096"

0 comments on commit 9e57929

Please sign in to comment.