[Test] Phishing

diff --git a/test/functional/cases/114_phishing.robot b/test/functional/cases/114_phishing.robot
new file mode 100644 (file)
index 0000000..dd2d3c8
--- /dev/null
+++ b/test/functional/cases/114_phishing.robot
@@ -0,0 +1,33 @@
+*** Settings ***
+Suite Setup     Phishing Setup
+Suite Teardown  Simple Teardown
+Library         ${TESTDIR}/lib/rspamd.py
+Resource        ${TESTDIR}/lib/rspamd.robot
+Variables       ${TESTDIR}/lib/vars.py
+
+*** Variables ***
+${CONFIG}        ${TESTDIR}/configs/plugins.conf
+${MESSAGE1}       ${TESTDIR}/messages/phishing1.eml
+${MESSAGE2}      ${TESTDIR}/messages/phishing2.eml
+${MESSAGE3}      ${TESTDIR}/messages/phishing3.eml
+${RSPAMD_SCOPE}  Suite
+${URL_TLD}       ${TESTDIR}/../../contrib/publicsuffix/effective_tld_names.dat
+
+*** Test Cases ***
+TEST PHISHING
+  ${result} =  Scan Message With Rspamc  ${MESSAGE1}
+  Check Rspamc  ${result}  ${SPACE}PHISHING
+
+TEST PHISHING STRICT ONE
+  ${result} =  Scan Message With Rspamc  ${MESSAGE2}
+  Check Rspamc  ${result}  STRICT_PHISHING
+
+TEST PHISHING STRICT TWO
+  ${result} =  Scan Message With Rspamc  ${MESSAGE3}
+  Check Rspamc  ${result}  STRICTER_PHISHING
+
+*** Keywords ***
+Phishing Setup
+  ${PLUGIN_CONFIG} =  Get File  ${TESTDIR}/configs/phishing.conf
+  Set Suite Variable  ${PLUGIN_CONFIG}
+  Generic Setup  PLUGIN_CONFIG

diff --git a/test/functional/configs/maps/strict.phishing b/test/functional/configs/maps/strict.phishing
new file mode 100644 (file)
index 0000000..af2de57
--- /dev/null
+++ b/test/functional/configs/maps/strict.phishing
@@ -0,0 +1 @@
+myspace.com

diff --git a/test/functional/configs/maps/stricter.phishing b/test/functional/configs/maps/stricter.phishing
new file mode 100644 (file)
index 0000000..a1240ea
--- /dev/null
+++ b/test/functional/configs/maps/stricter.phishing
@@ -0,0 +1 @@
+bank.com

diff --git a/test/functional/configs/phishing.conf b/test/functional/configs/phishing.conf
new file mode 100644 (file)
index 0000000..3ea53a0
--- /dev/null
+++ b/test/functional/configs/phishing.conf
@@ -0,0 +1,7 @@
+phishing {
+  symbol = "PHISHING";
+  strict_domains = [
+    "${TESTDIR}/configs/maps/strict.phishing:STRICT_PHISHING",
+    "${TESTDIR}/configs/maps/stricter.phishing:STRICTER_PHISHING"
+  ];
+}

diff --git a/test/functional/messages/phishing1.eml b/test/functional/messages/phishing1.eml
new file mode 100644 (file)
index 0000000..d168068
--- /dev/null
+++ b/test/functional/messages/phishing1.eml
@@ -0,0 +1,3 @@
+Content-type: text/html
+
+lol <a href="http://www.example.net">http://www.cnn.com</a>

diff --git a/test/functional/messages/phishing2.eml b/test/functional/messages/phishing2.eml
new file mode 100644 (file)
index 0000000..575dd0b
--- /dev/null
+++ b/test/functional/messages/phishing2.eml
@@ -0,0 +1,3 @@
+Content-type: text/html
+
+lol <a href="http://www.example.net">http://www.myspace.com</a>

diff --git a/test/functional/messages/phishing3.eml b/test/functional/messages/phishing3.eml
new file mode 100644 (file)
index 0000000..8ec9af5
--- /dev/null
+++ b/test/functional/messages/phishing3.eml
@@ -0,0 +1,3 @@
+Content-type: text/html
+
+lol <a href="http://www.example.net">http://www.bank.com</a>