[Test] task:remove_symbol() on grouped virtual symbol

diff --git a/test/functional/cases/001_merged/101_lua.robot b/test/functional/cases/001_merged/101_lua.robot
index 23637396aa9078d55496636e0794896c01da6d82..ac4f8677a9ba2633e91e0c9877426a95a16cc412 100644 (file)
--- a/test/functional/cases/001_merged/101_lua.robot
+++ b/test/functional/cases/001_merged/101_lua.robot
@@ -34,6 +34,11 @@ Option Order
   Expect Symbol With Exact Options  OPTION_ORDER  one  two  three  4  5  a
   Expect Symbol With Exact Options  TBL_OPTION_ORDER  one  two  three  4  5  a
 
+Remove Result
+  Scan File  ${MESSAGE}  Settings={symbols_enabled = [REMOVE_RESULT_EXPECTED, REMOVE_RESULT_UNEXPECTED]}
+  Expect Symbol  REMOVE_RESULT_EXPECTED
+  Do Not Expect Symbol  REMOVE_RESULT_UNEXPECTED
+
 Rule conditions
   Scan File  ${MESSAGE}  Settings={symbols_enabled = [ANY_A]}
   Expect Symbol With Option  ANY_A  hello3

diff --git a/test/functional/configs/merged.conf b/test/functional/configs/merged.conf
index bda7044e11c3949c4cc309ea0b2ae49218189a59..60930f657c6cb3fc2c67eea56236d6600560b62e 100644 (file)
--- a/test/functional/configs/merged.conf
+++ b/test/functional/configs/merged.conf
@@ -8,6 +8,7 @@ lua = "{= env.TESTDIR =}/lua/hashes.lua"
 lua = "{= env.TESTDIR =}/lua/maps_kv.lua"
 lua = "{= env.TESTDIR =}/lua/option_order.lua"
 lua = "{= env.TESTDIR =}/lua/recipients.lua"
+lua = "{= env.TESTDIR =}/lua/remove_result.lua"
 lua = "{= env.TESTDIR =}/lua/tlds.lua"
 
 # 104_get_from

diff --git a/test/functional/lua/remove_result.lua b/test/functional/lua/remove_result.lua
new file mode 100644 (file)
index 0000000..106d7f4
--- /dev/null
+++ b/test/functional/lua/remove_result.lua
@@ -0,0 +1,26 @@
+local id = rspamd_config:register_symbol({
+  name = 'REMOVE_RESULT_CB',
+  callback = function(task)
+    task:insert_result('REMOVE_RESULT_UNEXPECTED', 1.0, 'ohno')
+  end,
+  type = 'callback',
+})
+
+rspamd_config:register_symbol({
+  name = 'REMOVE_RESULT_UNEXPECTED',
+  type = 'virtual',
+  score = 0.1,
+  group = 'remove_result_test',
+  parent = id,
+})
+
+rspamd_config:register_symbol({
+  name = 'REMOVE_RESULT_EXPECTED',
+  callback = function(task)
+    return task:remove_result('REMOVE_RESULT_UNEXPECTED') and true or false
+  end,
+  type = 'normal',
+  score = 0.1,
+})
+
+rspamd_config:register_dependency('REMOVE_RESULT_EXPECTED', 'REMOVE_RESULT_UNEXPECTED')