From bd01177c1364399ef29ca6671669a271af967998 Mon Sep 17 00:00:00 2001
From: Andrew Lewis <nerf@judo.za.org>
Date: Tue, 9 Aug 2016 13:58:33 +0100
Subject: [PATCH] [Test] Add simple test for proxying

---
 test/functional/cases/140_proxy.robot | 38 +++++++++++++++++++++++++++
 test/functional/configs/proxy.conf    | 18 +++++++++++++
 test/functional/lib/vars.py           |  1 +
 3 files changed, 57 insertions(+)
 create mode 100644 test/functional/cases/140_proxy.robot
 create mode 100644 test/functional/configs/proxy.conf

diff --git a/test/functional/cases/140_proxy.robot b/test/functional/cases/140_proxy.robot
new file mode 100644
index 000000000..6ad308692
--- /dev/null
+++ b/test/functional/cases/140_proxy.robot
@@ -0,0 +1,38 @@
+*** Settings ***
+Suite Setup     Proxy Setup
+Suite Teardown  Proxy Teardown
+Library         ${TESTDIR}/lib/rspamd.py
+Resource        ${TESTDIR}/lib/rspamd.robot
+Variables       ${TESTDIR}/lib/vars.py
+
+*** Variables ***
+${LUA_SCRIPT}   ${TESTDIR}/lua/simple.lua
+${MESSAGE}      ${TESTDIR}/messages/spam_message.eml
+
+*** Test Cases ***
+Scan Message
+  ${result} =  Run Rspamc  -h  ${LOCAL_ADDR}:${PORT_PROXY}  -p  ${MESSAGE}
+  ${PROXY_LOGPOS} =  Log Logs  ${PROXY_TMPDIR}/rspamd.log  ${PROXY_LOGPOS}
+  Set Suite Variable  ${PROXY_LOGPOS}
+  ${SLAVE_LOGPOS} =  Log Logs  ${SLAVE_TMPDIR}/rspamd.log  ${SLAVE_LOGPOS}
+  Set Suite Variable  ${SLAVE_LOGPOS}
+  Run Keyword If  ${result.rc} != 0  Log  ${result.stderr}
+  Should Contain  ${result.stdout}  SIMPLE_TEST
+  Should Be Equal As Integers  ${result.rc}  0
+
+*** Keywords ***
+Proxy Setup
+  &{d} =  Run Rspamd  CONFIG=${TESTDIR}/configs/lua_test.conf
+  Set Suite Variable  ${SLAVE_LOGPOS}  &{d}[RSPAMD_LOGPOS]
+  Set Suite Variable  ${SLAVE_PID}  &{d}[RSPAMD_PID]
+  Set Suite Variable  ${SLAVE_TMPDIR}  &{d}[TMPDIR]
+  &{d} =  Run Rspamd  CONFIG=${TESTDIR}/configs/proxy.conf
+  Set Suite Variable  ${PROXY_LOGPOS}  &{d}[RSPAMD_LOGPOS]
+  Set Suite Variable  ${PROXY_PID}  &{d}[RSPAMD_PID]
+  Set Suite Variable  ${PROXY_TMPDIR}  &{d}[TMPDIR]
+
+Proxy Teardown
+  Shutdown Process With Children  ${PROXY_PID}
+  Shutdown Process With Children  ${SLAVE_PID}
+  Cleanup Temporary Directory  ${PROXY_TMPDIR}
+  Cleanup Temporary Directory  ${SLAVE_TMPDIR}
diff --git a/test/functional/configs/proxy.conf b/test/functional/configs/proxy.conf
new file mode 100644
index 000000000..5f40d1593
--- /dev/null
+++ b/test/functional/configs/proxy.conf
@@ -0,0 +1,18 @@
+options = {
+	filters = "spf"
+	pidfile = "${TMPDIR}/rspamd.pid"
+}
+logging = {
+	type = "file",
+	level = "debug"
+	filename = "${TMPDIR}/rspamd.log"
+}
+worker "rspamd_proxy" {
+    bind_socket = "${LOCAL_ADDR}:${PORT_PROXY}";
+    upstream {
+        name = "${LOCAL_ADDR}";
+        default = yes;
+        hosts = "${LOCAL_ADDR}:${PORT_NORMAL}";
+    }
+    count = 1;
+}
diff --git a/test/functional/lib/vars.py b/test/functional/lib/vars.py
index 170b7c5c3..f260dc1b4 100644
--- a/test/functional/lib/vars.py
+++ b/test/functional/lib/vars.py
@@ -4,6 +4,7 @@ LOCAL_ADDR = '127.0.0.1'
 PORT_CONTROLLER = 56790
 PORT_FUZZY = 56791
 PORT_NORMAL = 56789
+PORT_PROXY = 56795
 REDIS_ADDR = '127.0.0.1'
 REDIS_PORT = 56379
 RSPAMD_GROUP = 'nogroup'
-- 
2.39.5