1 files changed, 18 insertions, 0 deletions
diff --git a/test/lua/rspamd_test_helper.lua b/test/lua/rspamd_test_helper.lua
new file mode 100644
index 000000000..80ca2c8f7
--- /dev/null
+++ b/test/lua/rspamd_test_helper.lua
@@ -0,0 +1,18 @@
+local ffi = require "ffi"
+local cfg = rspamd_config
+
+ffi.cdef[[
+void rspamd_url_init (const char *tld_file);
+]]
+local exports = {}
+
+function exports.init_url_parser(file)
+  if not file then
+    local test_dir = string.gsub(debug.getinfo(1).source, "^@(.+/)[^/]+$", "%1")
+    file = string.format('%s/unit/%s', test_dir, "test_tld.dat")
+  end
+
+  ffi.C.rspamd_url_init(file)
+end
+
+return exports
+\ No newline at end of file