]> source.dussan.org Git - rspamd.git/commitdiff
[Test] Add a simple config helper for the unit tests
authorVsevolod Stakhov <vsevolod@highsecure.ru>
Wed, 24 Nov 2021 09:45:52 +0000 (09:45 +0000)
committerVsevolod Stakhov <vsevolod@highsecure.ru>
Wed, 24 Nov 2021 09:45:52 +0000 (09:45 +0000)
test/lua/rspamd_test_helper.lua
test/lua/unit/lua_util.extract_specific_urls.lua

index 80ca2c8f765b0645f4a5dad6a38fbedf5a3ea1d6..4d58d261a516951bcd973793286cc1cbe372aa06 100644 (file)
@@ -15,4 +15,32 @@ function exports.init_url_parser(file)
   ffi.C.rspamd_url_init(file)
 end
 
+function exports.default_config()
+  local test_dir = string.gsub(debug.getinfo(1).source, "^@(.+/)[^/]+$", "%1")
+  local tld_file = string.format('%s/%s', test_dir, "test_tld.dat")
+
+  local config = {
+    options = {
+      filters = {'spf', 'dkim', 'regexp'},
+      url_tld = tld_file,
+      dns = {
+        nameserver = {'8.8.8.8'}
+      },
+    },
+    logging = {
+      type = 'console',
+      level = 'debug'
+    },
+    metric = {
+      name = 'default',
+      actions = {
+        reject = 100500,
+      },
+      unknown_weight = 1
+    }
+  }
+
+  return config
+end
+
 return exports
\ No newline at end of file
index b1a9b07d21b40f7bd0c0d463625be696d2afa61b..a8471ae23d39cf13091b54299ae0f7cba21be276 100644 (file)
@@ -203,31 +203,10 @@ end)
 context("Lua util - extract_specific_urls message", function()
 
 --[[ ******************* kinda functional *************************************** ]]
-  local test_dir = string.gsub(debug.getinfo(1).source, "^@(.+/)[^/]+$", "%1")
-  local tld_file = string.format('%s/%s', test_dir, "test_tld.dat")
-
-  local config = {
-    options = {
-      filters = {'spf', 'dkim', 'regexp'},
-      url_tld = tld_file,
-      dns = {
-        nameserver = {'8.8.8.8'}
-      },
-    },
-    logging = {
-      type = 'console',
-      level = 'debug'
-    },
-    metric = {
-      name = 'default',
-      actions = {
-        reject = 100500,
-      },
-      unknown_weight = 1
-    }
-  }
 
-  local cfg = rspamd_util.config_from_ucl(config, "INIT_URL,INIT_LIBS,INIT_SYMCACHE,INIT_VALIDATE,INIT_PRELOAD_MAPS")
+
+  local cfg = rspamd_util.config_from_ucl(rspamd_test_helper.default_config(),
+      "INIT_URL,INIT_LIBS,INIT_SYMCACHE,INIT_VALIDATE,INIT_PRELOAD_MAPS")
   local res,task = rspamd_task.load_from_string(msg, cfg)
 
   if not res then