From 6f72e868f5217c0cb7373546a1d6337d5f1494f3 Mon Sep 17 00:00:00 2001
From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Mon, 21 Jan 2019 17:04:05 +0000
Subject: [PATCH] [Test] Add tests for lua_udp

---
 test/functional/cases/310_udp.robot | 46 ++++++++++++++++
 test/functional/lua/udp.lua         | 81 +++++++++++++++++++++++++++++
 test/functional/util/dummy_udp.py   |  5 ++
 3 files changed, 132 insertions(+)
 create mode 100644 test/functional/cases/310_udp.robot
 create mode 100644 test/functional/lua/udp.lua
 mode change 100644 => 100755 test/functional/util/dummy_udp.py

diff --git a/test/functional/cases/310_udp.robot b/test/functional/cases/310_udp.robot
new file mode 100644
index 000000000..2ecd1cfb8
--- /dev/null
+++ b/test/functional/cases/310_udp.robot
@@ -0,0 +1,46 @@
+*** Settings ***
+Test Setup      UDP Setup
+Test Teardown   UDP Teardown
+Library         Process
+Library         ${TESTDIR}/lib/rspamd.py
+Resource        ${TESTDIR}/lib/rspamd.robot
+Variables       ${TESTDIR}/lib/vars.py
+
+*** Variables ***
+${URL_TLD}      ${TESTDIR}/../lua/unit/test_tld.dat
+${CONFIG}       ${TESTDIR}/configs/lua_test.conf
+${MESSAGE}      ${TESTDIR}/messages/spam_message.eml
+${RSPAMD_SCOPE}  Test
+
+*** Test Cases ***
+Simple UDP request
+  ${result} =  Scan Message With Rspamc  ${MESSAGE}
+  Check Rspamc  ${result}  UDP_SUCCESS (0.00)[helloworld]
+
+Sendonly UDP request
+  ${result} =  Scan Message With Rspamc  ${MESSAGE}
+  Check Rspamc  ${result}  UDP_SENDTO
+
+Errored UDP request
+  ${result} =  Scan Message With Rspamc  ${MESSAGE}
+  Check Rspamc  ${result}  UDP_FAIL (0.00)[read timeout]
+
+*** Keywords ***
+Lua Setup
+  [Arguments]  ${LUA_SCRIPT}
+  Set Global Variable  ${LUA_SCRIPT}
+  Generic Setup
+
+UDP Setup
+  Run Dummy UDP
+  Lua Setup  ${TESTDIR}/lua/udp.lua
+
+UDP Teardown
+  ${udp_pid} =  Get File  /tmp/dummy_udp.pid
+  Shutdown Process With Children  ${udp_pid}
+  Normal Teardown
+
+Run Dummy UDP
+  [Arguments]
+  ${result} =  Start Process  ${TESTDIR}/util/dummy_udp.py  5005
+  Wait Until Created  /tmp/dummy_udp.pid
diff --git a/test/functional/lua/udp.lua b/test/functional/lua/udp.lua
new file mode 100644
index 000000000..0ed4b15a8
--- /dev/null
+++ b/test/functional/lua/udp.lua
@@ -0,0 +1,81 @@
+--[[[
+-- Just a test for UDP API
+--]]
+
+local rspamd_udp = require "rspamd_udp"
+local logger = require "rspamd_logger"
+
+-- [[ old fashioned callback api ]]
+local function simple_udp_async_symbol(task)
+  logger.errx(task, 'udp_symbol: begin')
+  local function udp_cb(success, data)
+    logger.errx(task, 'udp_cb: got reply: %s', data)
+
+    if success then
+      task:insert_result('UDP_SUCCESS', 1.0, data)
+    else
+      task:insert_result('UDP_FAIL', 1.0, data)
+    end
+  end
+  rspamd_udp:sendto({
+    task = task,
+    callback = udp_cb,
+    host = '127.0.0.1',
+    data = {'hello', 'world'},
+    port = 5005,
+  })
+end
+
+rspamd_config:register_symbol({
+  name = 'UDP_SUCCESS',
+  score = 0.0,
+  callback = simple_udp_async_symbol,
+})
+
+local function send_only_udp(task)
+  logger.errx(task, 'udp_symbol_sendonly: begin')
+  if rspamd_udp:sendto({
+    task = task,
+    host = '127.0.0.1',
+    data = {'hoho'},
+    port = 5005,
+  }) then
+
+    task:insert_result('UDP_SENDTO', 1.0)
+  end
+end
+
+rspamd_config:register_symbol({
+  name = 'UDP_SENDTO',
+  score = 0.0,
+  callback = send_only_udp,
+})
+
+local function udp_failed_cb(task)
+  logger.errx(task, 'udp_failed_cb: begin')
+  local function udp_cb(success, data)
+    logger.errx(task, 'udp_failed_cb: got reply: %s', data)
+
+    if success then
+      task:insert_result('UDP_SUCCESS', 1.0, data)
+    else
+      task:insert_result('UDP_FAIL', 1.0, data)
+    end
+  end
+  rspamd_udp:sendto({
+    task = task,
+    callback = udp_cb,
+    host = '127.0.0.1',
+    data = {'hello', 'world'},
+    port = 5006,
+    retransmits = 2,
+    timeout = 0.1,
+  })
+end
+
+rspamd_config:register_symbol({
+  name = 'UDP_FAIL',
+  score = 0.0,
+  callback = udp_failed_cb,
+})
+-- ]]
diff --git a/test/functional/util/dummy_udp.py b/test/functional/util/dummy_udp.py
old mode 100644
new mode 100755
index 9946b50f3..4d904e3b1
--- a/test/functional/util/dummy_udp.py
+++ b/test/functional/util/dummy_udp.py
@@ -1,7 +1,11 @@
+#!/usr/bin/env python
+
 import socket
 import sys
+import dummy_killer
 
 UDP_IP = "127.0.0.1"
+PID = "/tmp/dummy_udp.pid"
 
 if __name__ == "__main__":
     alen = len(sys.argv)
@@ -12,6 +16,7 @@ if __name__ == "__main__":
     sock = socket.socket(socket.AF_INET, # Internet
                          socket.SOCK_DGRAM) # UDP
     sock.bind((UDP_IP, port))
+    dummy_killer.write_pid(PID)
 
     while True:
         data, addr = sock.recvfrom(1024) # buffer size is 1024 bytes
-- 
2.39.5