From: Mikhail Galanin <mgalanin@mimecast.com>
Date: Wed, 19 Sep 2018 13:51:01 +0000 (+0100)
Subject: [Test] Assertions for table comparsion that are nicer to use
X-Git-Tag: 1.8.0~64^2
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=refs%2Fpull%2F2501%2Fhead;p=thirdparty%2Frspamd.git

[Test] Assertions for table comparsion that are nicer to use
---

diff --git a/test/lua/rspamd_assertions.lua b/test/lua/rspamd_assertions.lua
new file mode 100644
index 0000000000..0f2eb36265
--- /dev/null
+++ b/test/lua/rspamd_assertions.lua
@@ -0,0 +1,21 @@
+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
+
+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)
diff --git a/test/lua/tests.lua b/test/lua/tests.lua
index 8f64bba4ec..99d509de0c 100644
--- a/test/lua/tests.lua
+++ b/test/lua/tests.lua
@@ -1,6 +1,7 @@
 -- Run all unit tests in 'unit' directory
 
 local telescope = require "telescope"
+require "rspamd_assertions"
 
 local contexts = {}