From 496d57d63bd7fb6c4c412f66062ce7dfce628ec2 Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@rspamd.com>
Date: Tue, 29 Jul 2025 14:17:00 +0100
Subject: [PATCH] [Test] Add logic to match test id and logs id

---
 .../110_statistics/300-multiclass-redis.robot   |  1 +
 .../cases/110_statistics/multiclass_lib.robot   |  8 ++++----
 test/functional/lib/rspamd.robot                | 17 +++++++++++++++--
 3 files changed, 20 insertions(+), 6 deletions(-)

diff --git a/test/functional/cases/110_statistics/300-multiclass-redis.robot b/test/functional/cases/110_statistics/300-multiclass-redis.robot
index 12ec34d628..aa9e0425d6 100644
--- a/test/functional/cases/110_statistics/300-multiclass-redis.robot
+++ b/test/functional/cases/110_statistics/300-multiclass-redis.robot
@@ -2,6 +2,7 @@
 Documentation   Multiclass Bayes Classification Tests with Redis Backend
 Suite Setup     Rspamd Redis Setup
 Suite Teardown  Rspamd Redis Teardown
+Test Setup      Set Test Hash Documentation
 Resource        multiclass_lib.robot
 
 *** Variables ***
diff --git a/test/functional/cases/110_statistics/multiclass_lib.robot b/test/functional/cases/110_statistics/multiclass_lib.robot
index e8b0b3b64a..9f70e05fb6 100644
--- a/test/functional/cases/110_statistics/multiclass_lib.robot
+++ b/test/functional/cases/110_statistics/multiclass_lib.robot
@@ -21,9 +21,9 @@ Learn Multiclass
     # Extract filename from message path for queue-id
     ${path}  ${filename} =  Split Path  ${message}
     IF  "${user}"
-        ${result} =  Run Rspamc  -d  ${user}  --queue-id  ${filename}  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_class:${class}  ${message}
+        ${result} =  Run Rspamc  -d  ${user}  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_class:${class}  ${message}
     ELSE
-        ${result} =  Run Rspamc  --queue-id  ${filename}  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_class:${class}  ${message}
+        ${result} =  Run Rspamc  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_class:${class}  ${message}
     END
     Check Rspamc  ${result}
 
@@ -33,9 +33,9 @@ Learn Multiclass Legacy
     # Extract filename from message path for queue-id
     ${path}  ${filename} =  Split Path  ${message}
     IF  "${user}"
-        ${result} =  Run Rspamc  -d  ${user}  --queue-id  ${filename}  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_${class}  ${message}
+        ${result} =  Run Rspamc  -d  ${user}  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_${class}  ${message}
     ELSE
-        ${result} =  Run Rspamc  --queue-id  ${filename}  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_${class}  ${message}
+        ${result} =  Run Rspamc  -h  ${RSPAMD_LOCAL_ADDR}:${RSPAMD_PORT_CONTROLLER}  learn_${class}  ${message}
     END
     Check Rspamc  ${result}
 
diff --git a/test/functional/lib/rspamd.robot b/test/functional/lib/rspamd.robot
index 5d23e3cebb..f61998f46e 100644
--- a/test/functional/lib/rspamd.robot
+++ b/test/functional/lib/rspamd.robot
@@ -419,10 +419,23 @@ Run Nginx
   ${nginx_log} =  Get File  ${RSPAMD_TMPDIR}/nginx.log
   Log  ${nginx_log}
 
+Set Test Hash Documentation
+  ${log_tag} =  Evaluate  __import__('hashlib').md5('${TEST NAME}'.encode()).hexdigest()[:8]
+  Log    TEST CONTEXT: [${log_tag}] ${TEST NAME}    console=True
+
 Run Rspamc
   [Arguments]  @{args}
-  ${result} =  Run Process  ${RSPAMC}  -t  60  --header  Queue-ID\=${TEST NAME}
-  ...  @{args}  env:LD_LIBRARY_PATH=${RSPAMD_TESTDIR}/../../contrib/aho-corasick
+  ${log_tag} =  Evaluate  __import__('hashlib').md5('${TEST NAME}'.encode()).hexdigest()[:8]
+  # Check if --queue-id is already provided in the arguments
+  ${args_str} =  Evaluate  ' '.join(@{args})
+  ${has_queue_id} =  Evaluate  '--queue-id' in '${args_str}'
+  IF  ${has_queue_id}
+    ${result} =  Run Process  ${RSPAMC}  -t  60  --log-tag  ${log_tag}
+    ...  @{args}  env:LD_LIBRARY_PATH=${RSPAMD_TESTDIR}/../../contrib/aho-corasick
+  ELSE
+    ${result} =  Run Process  ${RSPAMC}  -t  60  --queue-id  ${TEST NAME}  --log-tag  ${log_tag}
+    ...  @{args}  env:LD_LIBRARY_PATH=${RSPAMD_TESTDIR}/../../contrib/aho-corasick
+  END
   Log  ${result.stdout}
   [Return]  ${result}
 
-- 
2.47.3