From 734d16e1f43e17a2aabe10aa12fddf149a762bb6 Mon Sep 17 00:00:00 2001
From: Andrew Lewis <nerf@judo.za.org>
Date: Tue, 9 May 2017 12:40:33 +0200
Subject: [PATCH] [Test] Test dkim_signing with redis

---
 .../001_simple.robot}                         |  2 +-
 .../cases/131_dkim_signing/002_redis.robot    | 39 +++++++++++++++++++
 .../configs/dkim_signing/redis.conf           |  8 ++++
 .../simple.conf}                              |  0
 4 files changed, 48 insertions(+), 1 deletion(-)
 rename test/functional/cases/{131_dkim_signing.robot => 131_dkim_signing/001_simple.robot} (92%)
 create mode 100644 test/functional/cases/131_dkim_signing/002_redis.robot
 create mode 100644 test/functional/configs/dkim_signing/redis.conf
 rename test/functional/configs/{dkim_signing.conf => dkim_signing/simple.conf} (100%)

diff --git a/test/functional/cases/131_dkim_signing.robot b/test/functional/cases/131_dkim_signing/001_simple.robot
similarity index 92%
rename from test/functional/cases/131_dkim_signing.robot
rename to test/functional/cases/131_dkim_signing/001_simple.robot
index 58f608b8f..c715b4723 100644
--- a/test/functional/cases/131_dkim_signing.robot
+++ b/test/functional/cases/131_dkim_signing/001_simple.robot
@@ -25,7 +25,7 @@ TEST NOT SIGNED - USERNAME WRONG DOMAIN
 
 *** Keywords ***
 DKIM Signing Setup
-  ${PLUGIN_CONFIG} =  Get File  ${TESTDIR}/configs/dkim_signing.conf
+  ${PLUGIN_CONFIG} =  Get File  ${TESTDIR}/configs/dkim_signing/simple.conf
   Set Suite Variable  ${PLUGIN_CONFIG}
   Generic Setup  PLUGIN_CONFIG
 
diff --git a/test/functional/cases/131_dkim_signing/002_redis.robot b/test/functional/cases/131_dkim_signing/002_redis.robot
new file mode 100644
index 000000000..f46bfbc33
--- /dev/null
+++ b/test/functional/cases/131_dkim_signing/002_redis.robot
@@ -0,0 +1,39 @@
+*** Settings ***
+Suite Setup     DKIM Signing Setup
+Suite Teardown  DKIM Signing Teardown
+Library         ${TESTDIR}/lib/rspamd.py
+Resource        ${TESTDIR}/lib/rspamd.robot
+Variables       ${TESTDIR}/lib/vars.py
+
+*** Variables ***
+${CONFIG}       ${TESTDIR}/configs/plugins.conf
+${MESSAGE}      ${TESTDIR}/messages/dmarc/fail_none.eml
+${REDIS_SCOPE}  Suite
+${RSPAMD_SCOPE}  Suite
+${URL_TLD}      ${TESTDIR}/../lua/unit/test_tld.dat
+
+*** Test Cases ***
+TEST SIGNED
+  ${result} =  Scan Message With Rspamc  ${MESSAGE}  -u  bob@rspamd.tk
+  Check Rspamc  ${result}  DKIM-Signature:
+  Should Contain  ${result.stdout}  DKIM_SIGNED (1.00)
+
+TEST NOT SIGNED - USERNAME WRONG DOMAIN
+  ${result} =  Scan Message With Rspamc  ${MESSAGE}  -u  bob@example.tk
+  Check Rspamc  ${result}  DKIM-Signature:  inverse=1
+  Should Not Contain  ${result.stdout}  DKIM_SIGNED (1.00)
+
+*** Keywords ***
+DKIM Signing Setup
+  ${PLUGIN_CONFIG} =  Get File  ${TESTDIR}/configs/dkim_signing/redis.conf
+  Set Suite Variable  ${PLUGIN_CONFIG}
+  Generic Setup  PLUGIN_CONFIG
+  Run Redis
+  Redis HSET  TEST_DKIM_SELECTORS  rspamd.tk  test
+  ${key} =  Get File  ${TESTDIR}/configs/dkim.key
+  Redis HSET  TEST_DKIM_KEYS  test.rspamd.tk  ${key}
+
+DKIM Signing Teardown
+  Normal Teardown
+  Shutdown Process With Children  ${REDIS_PID}
+  Wait For Port  ${SOCK_STREAM}  ${LOCAL_ADDR}  ${REDIS_PORT}
diff --git a/test/functional/configs/dkim_signing/redis.conf b/test/functional/configs/dkim_signing/redis.conf
new file mode 100644
index 000000000..903a13cf4
--- /dev/null
+++ b/test/functional/configs/dkim_signing/redis.conf
@@ -0,0 +1,8 @@
+dkim_signing {
+  use_redis = true;
+  key_prefix = "TEST_DKIM_KEYS";
+  selector_prefix = "TEST_DKIM_SELECTORS";
+}
+redis {
+  servers = "${REDIS_ADDR}:${REDIS_PORT}";
+}
diff --git a/test/functional/configs/dkim_signing.conf b/test/functional/configs/dkim_signing/simple.conf
similarity index 100%
rename from test/functional/configs/dkim_signing.conf
rename to test/functional/configs/dkim_signing/simple.conf
-- 
2.39.5