Add unit tests for the new mempool variables.

1 files changed, 37 insertions, 0 deletions

diff --git a/test/lua/unit/mempool.lua b/test/lua/unit/mempool.lua
new file mode 100644
index 000000000..2afaad7ec
--- /dev/null
+++ b/test/lua/unit/mempool.lua
@@ -0,0 +1,37 @@
+context("Memory pool unit tests", function()
+  test("Mempool variables", function()
+    local mempool = require "rspamd_mempool"
+
+    local pool = mempool.create()
+
+    assert_not_nil(pool)
+
+    -- string
+    pool:set_variable('a', 'bcd')
+    local var = pool:get_variable('a')
+    assert_equal(var, 'bcd')
+
+    -- integer
+    pool:set_variable('a', 1)
+    var = pool:get_variable('a', 'double')
+    assert_equal(var, 1)
+
+    -- float
+    pool:set_variable('a', 1.01)
+    var = pool:get_variable('a', 'double')
+    assert_equal(var, 1.01)
+
+    -- boolean
+    pool:set_variable('a', false)
+    var = pool:get_variable('a', 'bool')
+    assert_equal(var, false)
+
+    -- multiple
+    pool:set_variable('a', 'bcd', 1, 1.01, false)
+    local v1, v2, v3, v4 = pool:get_variable('a', 'string,double,double,bool')
+    assert_equal(v1, 'bcd')
+    assert_equal(v2, 1)
+    assert_equal(v3, 1.01)
+    assert_equal(v4, false)
+  end)
+end)
\ No newline at end of file