From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Mon, 20 Jul 2015 21:52:20 +0000 (+0100)
Subject: Add unit tests for the new mempool variables.
X-Git-Tag: 1.0.0~327
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=39f580451550ff4f08fc3a26a0b2455675a27ab9;p=thirdparty%2Frspamd.git

Add unit tests for the new mempool variables.
---

diff --git a/test/lua/unit/mempool.lua b/test/lua/unit/mempool.lua
new file mode 100644
index 0000000000..2afaad7ecc
--- /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