Merge pull request #5104 from rspamd/vstakhov-transparent-ucl

[Project] Allow manipulations with opaque UCL objects

1 files changed, 75 insertions, 0 deletions

diff --git a/test/lua/unit/ucl.lua b/test/lua/unit/ucl.lua
new file mode 100644
index 000000000..1b975d390
--- /dev/null
+++ b/test/lua/unit/ucl.lua
@@ -0,0 +1,75 @@
+-- Test some UCL stuff
+
+context("UCL manipulation", function()
+  local ucl = require "ucl"
+
+  local parser = ucl.parser()
+  local res, err = parser:parse_string('{"key":"val"}')
+  assert(res)
+
+  local reply = parser:get_object_wrapped()
+  local expected = {
+    key = 'ohlol',
+    ololo = 'ohlol'
+  }
+
+  test("UCL transparent test: object", function()
+    assert_equal(tostring(reply), '{"key":"val"}')
+    assert_equal(reply:type(), 'object')
+    assert_equal(reply:at('key'):unwrap(), 'val')
+    reply.ololo = 'ohlol'
+    reply.ololo = 'ohlol'
+    reply.key = 'ohlol'
+    assert_equal(reply:at('key'):unwrap(), 'ohlol')
+
+    for k, v in reply:pairs() do
+      assert_equal(expected[k], v:unwrap())
+    end
+  end)
+
+  test("UCL transparent test: array", function()
+    parser = ucl.parser()
+    res, err = parser:parse_string('["e1","e2"]')
+    assert(res)
+    local ireply = parser:get_object_wrapped()
+
+    assert_equal(tostring(ireply), '["e1","e2"]')
+    assert_equal(ireply:type(), 'array')
+    ireply[1] = 1
+    ireply[1] = 1
+    ireply[1] = 1
+    ireply[1] = 1
+    ireply[1] = 1
+    ireply[ireply:len() + 1] = 100500
+    local iexpected = { 1, "e2", 100500 }
+    for k, v in ireply:ipairs() do
+      assert_equal(v:unwrap(), iexpected[k])
+    end
+  end)
+
+  test("UCL transparent test: concat", function()
+    reply.tbl = ireply
+    expected.tbl = iexpected
+    for k, v in reply:pairs() do
+      if type(expected[k]) == 'table' then
+        for kk, vv in v:ipairs() do
+          assert_equal(expected[k][kk], vv:unwrap())
+        end
+      else
+        assert_equal(expected[k], v:unwrap())
+      end
+    end
+  end)
+
+  test("UCL transparent test: implicit conversion array->object", function()
+    -- Assign empty table, so it'll be an array
+    reply.t = {}
+    assert_equal(reply.t:type(), 'array')
+    -- We can convert empty table to object
+    reply.t.test = 'test'
+    assert_equal(reply.t:type(), 'object')
+    assert_equal(reply.t.test:unwrap(), 'test')
+  end)
+
+  collectgarbage() -- To ensure we don't crash with asan
+end)
\ No newline at end of file