branch-4.1: [test](fe) Cover internal query audit failure in FE unit test #65696

fe/fe-core/src/test/java/org/apache/doris/qe/StmtExecutorInternalQueryTest.java

@@ -18,13 +18,19 @@
 package org.apache.doris.qe;
 
 import org.apache.doris.analysis.StatementBase;
+import org.apache.doris.catalog.Env;
 import org.apache.doris.common.ErrorCode;
+import org.apache.doris.common.jmockit.Deencapsulation;
 import org.apache.doris.nereids.NereidsPlanner;
+import org.apache.doris.plugin.AuditEvent;
+import org.apache.doris.resource.workloadschedpolicy.WorkloadRuntimeStatusMgr;
 
 import mockit.Mock;
 import mockit.MockUp;
 import org.junit.Assert;
 import org.junit.Test;
+import org.mockito.ArgumentCaptor;
+import org.mockito.Mockito;
 
 public class StmtExecutorInternalQueryTest {
     @Test
@@ -73,4 +79,43 @@ public void plan(StatementBase queryStmt, org.apache.doris.thrift.TQueryOptions
             ConnectContext.remove();
         }
     }
+
+    @Test
+    public void testExecuteInternalQuerySubmitsErrorAuditEventOnFailure() {
+        ConnectContext ctx = new ConnectContext();
+        ctx.setThreadLocalInfo();
+        StmtExecutor executor = new StmtExecutor(ctx, "select * from table1");
+        Env env = Env.getCurrentEnv();
+        WorkloadRuntimeStatusMgr originalWorkloadRuntimeStatusMgr = env.getWorkloadRuntimeStatusMgr();
+        WorkloadRuntimeStatusMgr workloadRuntimeStatusMgr = Mockito.mock(WorkloadRuntimeStatusMgr.class);
+        ArgumentCaptor<AuditEvent> auditEventCaptor = ArgumentCaptor.forClass(AuditEvent.class);
+
+        Deencapsulation.setField(env, "workloadRuntimeStatusMgr", workloadRuntimeStatusMgr);
+        new MockUp<NereidsPlanner>() {
+            @Mock
+            public void plan(StatementBase queryStmt, org.apache.doris.thrift.TQueryOptions queryOptions) {
+                throw new RuntimeException("mock plan failure");
+            }
+        };
+        try {
+            Assert.assertThrows(RuntimeException.class, executor::executeInternalQuery);
+            Mockito.verify(workloadRuntimeStatusMgr).submitFinishQueryToAudit(auditEventCaptor.capture());
+        } finally {
+            Deencapsulation.setField(env, "workloadRuntimeStatusMgr", originalWorkloadRuntimeStatusMgr);
+            ConnectContext.remove();
+        }
+
+        AuditEvent auditEvent = auditEventCaptor.getValue();
+        Assert.assertEquals(AuditEvent.EventType.AFTER_QUERY, auditEvent.type);
+        Assert.assertEquals("ERR", auditEvent.state);
+        Assert.assertEquals(ErrorCode.ERR_INTERNAL_ERROR.getCode(), auditEvent.errorCode);
+        Assert.assertNotNull(auditEvent.errorMessage);
+        Assert.assertTrue("error message should mention root cause, got: " + auditEvent.errorMessage,
+                auditEvent.errorMessage.contains("mock plan failure"));
+        Assert.assertTrue("audit event should be marked as internal", auditEvent.isInternal);
+        Assert.assertTrue("audit event should be marked as query", auditEvent.isQuery);
+        Assert.assertTrue("audit event should be marked as nereids", auditEvent.isNereids);
+        Assert.assertEquals("select * from table1", auditEvent.stmt);
+        Assert.assertEquals("a8ec30e5ad0820f8c5bd16a82a4491ca", auditEvent.sqlHash);
+    }
 }