summaryrefslogtreecommitdiffstats
path: root/test/lua/rspamd_assertions.lua
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-10 21:30:40 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-10 21:30:40 +0000
commit133a45c109da5310add55824db21af5239951f93 (patch)
treeba6ac4c0a950a0dda56451944315d66409923918 /test/lua/rspamd_assertions.lua
parentInitial commit. (diff)
downloadrspamd-133a45c109da5310add55824db21af5239951f93.tar.xz
rspamd-133a45c109da5310add55824db21af5239951f93.zip
Adding upstream version 3.8.1.upstream/3.8.1upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'test/lua/rspamd_assertions.lua')
-rw-r--r--test/lua/rspamd_assertions.lua138
1 files changed, 138 insertions, 0 deletions
diff --git a/test/lua/rspamd_assertions.lua b/test/lua/rspamd_assertions.lua
new file mode 100644
index 0000000..d0590fd
--- /dev/null
+++ b/test/lua/rspamd_assertions.lua
@@ -0,0 +1,138 @@
+local logger = require "rspamd_logger"
+local telescope = require "telescope"
+local util = require 'lua_util'
+
+local function rspamd_assert_equals(tbl)
+ return tbl.expect == tbl.actual
+end
+
+local function rspamd_assert_equals_msg(_, tbl)
+ return logger.slog(
+ "Failed asserting that \n (actual) : %1 \n equals to\n (expected) : %2",
+ tbl.actual, tbl.expect
+ )
+end
+
+local function rspamd_assert_table_equals(tbl)
+ return util.table_cmp(tbl.expect, tbl.actual)
+end
+
+local function rspamd_assert_table_equals_sorted(tbl)
+ local expect = util.deepcopy(tbl.expect)
+ local actual = util.deepcopy(tbl.actual)
+ util.deepsort(expect)
+ util.deepsort(actual)
+ return util.table_cmp(expect, actual)
+end
+
+local function table_keys_sorted(t)
+ local keys = {}
+
+ for k,_ in pairs(t) do
+ table.insert(keys, k)
+ end
+ table.sort(keys)
+ return keys;
+end
+
+local function format_line(level, key, v_expect, v_actual)
+ local prefix
+ if v_expect == v_actual then
+ prefix = string.rep(' ', level * 2 + 1)
+ return logger.slog("%s[%s] = %s", prefix, key, v_expect)
+ else
+ prefix = string.rep(' ', level * 2)
+ local ret = {}
+ if v_expect then
+ ret[#ret + 1] = logger.slog("-%s[%s] = %s: %s", prefix, key,
+ type(v_expect), v_expect)
+ end
+ if v_actual then
+ ret[#ret + 1] = logger.slog("+%s[%s] = %s: %s", prefix,
+ (key), type(v_actual), (v_actual))
+ end
+ return table.concat(ret, "\n")
+ end
+end
+
+local function format_table_begin(level, key)
+ local prefix = string.rep(' ', level * 2 + 1)
+ return string.format("%s[%s] = {", prefix, tostring(key))
+end
+
+local function format_table_end(level)
+ local prefix = string.rep(' ', level * 2 + 1)
+ return string.format("%s}", prefix)
+end
+
+local function rspamd_assert_table_diff_msg(_, tbl)
+ local avoid_loops = {}
+ local msg = rspamd_assert_equals_msg(_, tbl)
+
+ local diff = {}
+ local function recurse(expect, actual, level)
+ if avoid_loops[actual] then
+ return
+ end
+ avoid_loops[actual] = true
+
+ local keys_expect = table_keys_sorted(expect)
+ local keys_actual = table_keys_sorted(actual)
+
+ local i_k_expect, i_v_expect = next(keys_expect)
+ local i_k_actual, i_v_actual = next(keys_actual)
+
+ while i_k_expect and i_k_actual do
+ local v_expect = expect[i_v_expect]
+ local v_actual = actual[i_v_actual]
+
+ if i_v_expect == i_v_actual then
+ -- table keys are the same: compare values
+ if type(v_expect) == 'table' and type(v_actual) == 'table' then
+ if util.table_cmp(v_expect, v_actual) then
+ -- we use the same value for 'actual' and 'expect' as soon as they're equal and don't bother us
+ diff[#diff + 1] = format_line(level, i_v_expect, v_expect, v_expect)
+ else
+ diff[#diff + 1] = format_table_begin(level, i_v_expect)
+ recurse(v_expect, v_actual, level + 1)
+ diff[#diff + 1] = format_table_end(level)
+ end
+ else
+ diff[#diff + 1] = format_line(level, i_v_expect, v_expect, v_actual)
+ end
+
+ i_k_expect, i_v_expect = next(keys_expect, i_k_expect)
+ i_k_actual, i_v_actual = next(keys_actual, i_k_actual)
+ elseif tostring(v_actual) > tostring(v_expect) then
+ diff[#diff + 1] = format_line(level, i_v_expect, v_expect, nil)
+ i_k_expect, i_v_expect = next(keys_expect, i_k_expect)
+ else
+ diff[#diff + 1] = format_line(level, i_v_actual, nil, v_actual)
+ i_k_actual, i_v_actual = next(keys_actual, i_k_actual)
+ end
+
+ end
+
+ while i_k_expect do
+ local v_expect = expect[i_v_expect]
+ diff[#diff + 1] = format_line(level, i_v_expect, v_expect, nil)
+ i_k_expect, i_v_expect = next(keys_expect, i_k_expect)
+ end
+
+ while i_k_actual do
+ local v_actual = actual[i_v_actual]
+ diff[#diff + 1] = format_line(level, i_v_actual, nil, v_actual)
+ i_k_actual, i_v_actual = next(keys_actual, i_k_actual)
+ end
+ end
+ recurse(tbl.expect, tbl.actual, 0)
+
+ return string.format("%s\n===== diff (-expect, +actual) ======\n%s", msg, table.concat(diff, "\n"))
+end
+
+telescope.make_assertion("rspamd_eq", rspamd_assert_equals_msg, rspamd_assert_equals)
+-- telescope.make_assertion("rspamd_table_eq", rspamd_assert_equals_msg, rspamd_assert_table_equals)
+telescope.make_assertion("rspamd_table_eq", rspamd_assert_table_diff_msg, rspamd_assert_table_equals)
+telescope.make_assertion("rspamd_table_eq_sorted", rspamd_assert_table_diff_msg,
+ rspamd_assert_table_equals_sorted)
+