split audit messages by transport type

Signed-off-by: Maciej Mierzwa <[email protected]>
MaciejMierzwa · Nov 7, 2023 · 893f8b9 · 893f8b9
1 parent bac341d
commit 893f8b9
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/integrationTest/java/org/opensearch/test/framework/audit/AuditLogsRule.java b/src/integrationTest/java/org/opensearch/test/framework/audit/AuditLogsRule.java
@@ -139,7 +139,16 @@ private void beforeTest(String methodName) {
     }
 
     public void onAuditMessage(AuditMessage auditMessage) {
-        currentTestAuditMessages.add(auditMessage);
-        log.debug("New audit message received '{}', total number of audit messages '{}'.", auditMessage, currentTestAuditMessages.size());
+        if(auditMessage.getAsMap().keySet().contains("audit_transport_headers")) {
+            if (log.isDebugEnabled()) {
+                log.debug("New transport audit message received '{}', total number of transport audit messages '{}'.", auditMessage, currentTransportTestAuditMessages.size());
+            }
+            currentTransportTestAuditMessages.add(auditMessage);
+        } else {
+            if (log.isDebugEnabled()) {
+                log.debug("New audit message received '{}', total number of audit messages '{}'.", auditMessage, currentTestAuditMessages.size());
+            }
+            currentTestAuditMessages.add(auditMessage);
+        }
     }
 }