[Test] Suppress bogus lookup

diff --git a/test/functional/lua/dns.lua b/test/functional/lua/dns.lua
index 702b985506938dc9b78bf4e21b2de70ee11dcf0b..b4750cd7b2528431f2ddb0d64f50ae6afdd4f06e 100644 (file)
--- a/test/functional/lua/dns.lua
+++ b/test/functional/lua/dns.lua
@@ -2,7 +2,9 @@ local rspamd_dns = require "rspamd_dns"
 local logger = require "rspamd_logger"
 
 local function dns_sync_symbol(task)
-  local to_resolve = tostring(task:get_request_header('to-resolve'))
+  local to_resolve = task:get_request_header('to-resolve')
+  if not to_resolve then return end
+  to_resolve = tostring(to_resolve)
   local is_ok, results = rspamd_dns.request({
     task = task,
     type = 'a',
@@ -37,7 +39,9 @@ local function dns_symbol(task)
       task:insert_result('DNS', 1.0, tostring(results[1]))
     end
   end
-  local to_resolve = tostring(task:get_request_header('to-resolve'))
+  local to_resolve = task:get_request_header('to-resolve')
+  if not to_resolve then return end
+  to_resolve = tostring(to_resolve)
 
   task:get_resolver():resolve_a({
     task = task,
@@ -50,4 +54,4 @@ rspamd_config:register_symbol({
   name = 'SIMPLE_DNS',
   score = 1.0,
   callback = dns_symbol,
-})
\ No newline at end of file
+})