From 4f3008f9682d4edb741913b2ee2aa9cd831f7d04 Mon Sep 17 00:00:00 2001
From: Andrew Lewis <nerf@judo.za.org>
Date: Wed, 10 Apr 2024 16:30:56 +0200
Subject: [PATCH] [Test] Add tests for grow_factor

---
 test/functional/cases/450_grow_factor.robot   | 37 +++++++++++++++++++
 .../functional/configs/grow_factor-local.conf |  8 ++++
 test/functional/configs/grow_factor.conf      |  7 ++++
 test/functional/lua/simple_plus.lua           | 28 ++++++++++++++
 4 files changed, 80 insertions(+)
 create mode 100644 test/functional/cases/450_grow_factor.robot
 create mode 100644 test/functional/configs/grow_factor-local.conf
 create mode 100644 test/functional/configs/grow_factor.conf
 create mode 100644 test/functional/lua/simple_plus.lua

diff --git a/test/functional/cases/450_grow_factor.robot b/test/functional/cases/450_grow_factor.robot
new file mode 100644
index 000000000..518c3ed82
--- /dev/null
+++ b/test/functional/cases/450_grow_factor.robot
@@ -0,0 +1,37 @@
+*** Settings ***
+Suite Setup     Rspamd Setup
+Suite Teardown  Rspamd Teardown
+Library         ${RSPAMD_TESTDIR}/lib/rspamd.py
+Resource        ${RSPAMD_TESTDIR}/lib/rspamd.robot
+Variables       ${RSPAMD_TESTDIR}/lib/vars.py
+
+*** Variables ***
+${CONFIG}               ${RSPAMD_TESTDIR}/configs/grow_factor.conf
+${HAM_MESSAGE}          ${RSPAMD_TESTDIR}/messages/ham.eml
+${RSPAMD_SCOPE}         Suite
+&{RESCORED_SYMBOLS}
+...  SIMPLE_TEST_001=0.013067
+...  SIMPLE_TEST_002=14.374194
+...  SIMPLE_TEST_003=6.533724
+...  SIMPLE_TEST_004=13.067449
+...  SIMPLE_TEST_005=0.013067
+...  SIMPLE_TEST_006=0.130674
+...  SIMPLE_TEST_007=0.143741
+...  SIMPLE_TEST_008=0.156809
+...  SIMPLE_TEST_009=0.169876
+...  SIMPLE_TEST_010=0.182944
+...  SIMPLE_TEST_011=-0.010000
+...  SIMPLE_TEST_012=-0.100000
+...  SIMPLE_TEST_013=-10.000000
+
+*** Test Cases ***
+CHECK BASIC
+  Scan File  ${HAM_MESSAGE}
+  ...  Settings={groups_enabled = [simple_tests]}
+  Expect Required Score  15
+  Expect Symbols With Scores  &{RESCORED_SYMBOLS}
+
+CHECK NOREJECT
+  Scan File  ${HAM_MESSAGE}
+  ...  Settings={actions { reject = null, "add header" = 15 }, groups_enabled = [simple_tests]}
+  Expect Symbols With Scores  &{RESCORED_SYMBOLS}
diff --git a/test/functional/configs/grow_factor-local.conf b/test/functional/configs/grow_factor-local.conf
new file mode 100644
index 000000000..cdf1bf436
--- /dev/null
+++ b/test/functional/configs/grow_factor-local.conf
@@ -0,0 +1,8 @@
+metric {
+  actions {
+    grow_factor = 1.1;
+    reject = 15;
+    "add header" = 6;
+    greylist = 4;
+  }
+}
diff --git a/test/functional/configs/grow_factor.conf b/test/functional/configs/grow_factor.conf
new file mode 100644
index 000000000..e26faa54b
--- /dev/null
+++ b/test/functional/configs/grow_factor.conf
@@ -0,0 +1,7 @@
+.include "{= env.TESTDIR =}/../../conf/rspamd.conf"
+
+lua = "{= env.TESTDIR =}/lua/simple_plus.lua"
+
+.include(priority=1,duplicate=merge) "{= env.TESTDIR =}/configs/grow_factor-local.conf"
+.include(priority=1,duplicate=merge) "{= env.TESTDIR =}/configs/merged-local.conf"
+.include(priority=2,duplicate=replace) "{= env.TESTDIR =}/configs/merged-override.conf"
diff --git a/test/functional/lua/simple_plus.lua b/test/functional/lua/simple_plus.lua
new file mode 100644
index 000000000..57aee6ae3
--- /dev/null
+++ b/test/functional/lua/simple_plus.lua
@@ -0,0 +1,28 @@
+local test_symbols = {
+  SIMPLE_TEST_001 = 0.01,
+  SIMPLE_TEST_002 = 11,
+  SIMPLE_TEST_003 = 5.0,
+  SIMPLE_TEST_004 = 10.0,
+  SIMPLE_TEST_005 = 0.01,
+  SIMPLE_TEST_006 = 0.10,
+  SIMPLE_TEST_007 = 0.11,
+  SIMPLE_TEST_008 = 0.12,
+  SIMPLE_TEST_009 = 0.13,
+  SIMPLE_TEST_010 = 0.14,
+  SIMPLE_TEST_011 = -0.01,
+  SIMPLE_TEST_012 = -0.1,
+  SIMPLE_TEST_013 = -10.0,
+}
+
+for k, v in pairs(test_symbols) do
+
+  rspamd_config:register_symbol({
+    name = k,
+    group = 'simple_tests',
+    score = v,
+    callback = function()
+      return true
+    end
+  })
+
+end
-- 
2.39.5