From e5f163a041d5a45ea72448e11cfc30abb16f10b6 Mon Sep 17 00:00:00 2001
From: Gavin Andresen <gavinandresen@gmail.com>
Date: Tue, 19 Mar 2013 14:08:21 -0400
Subject: -alertnotify=<cmd>

Runs a shell command when an AppliesToMe() alert is received.
%s in the <cmd> string is replaced with the alert.strStatusBar
message.
---
 src/test/alert_tests.cpp | 111 ++++++++++++++++++++++++++++++++++-------------
 src/test/data/alertTests | Bin 1123 -> 1283 bytes
 2 files changed, 82 insertions(+), 29 deletions(-)

(limited to 'src/test')

diff --git a/src/test/alert_tests.cpp b/src/test/alert_tests.cpp
index c8d409db9e..f7a11376d3 100644
--- a/src/test/alert_tests.cpp
+++ b/src/test/alert_tests.cpp
@@ -4,13 +4,12 @@
 
 #include <boost/foreach.hpp>
 #include <boost/test/unit_test.hpp>
+#include <fstream>
 
 #include "alert.h"
 #include "serialize.h"
 #include "util.h"
 
-BOOST_AUTO_TEST_SUITE(Alert_tests)
-
 #if 0
 //
 // alertTests contains 7 alerts, generated with this code:
@@ -39,64 +38,89 @@ BOOST_AUTO_TEST_SUITE(Alert_tests)
     SignAndSave(alert, "test/alertTests");
 
     alert.setSubVer.clear();
-    alert.nID = 2;
+    ++alert.nID;
     alert.nCancel = 1;
+    alert.nPriority = 100;
     alert.strStatusBar  = "Alert 2, cancels 1";
     SignAndSave(alert, "test/alertTests");
 
     alert.nExpiration += 60;
+    ++alert.nID;
     SignAndSave(alert, "test/alertTests");
 
+    ++alert.nID;
     alert.nMinVer = 11;
     alert.nMaxVer = 22;
     SignAndSave(alert, "test/alertTests");
 
+    ++alert.nID;
     alert.strStatusBar  = "Alert 2 for Satoshi 0.1.0";
     alert.setSubVer.insert(std::string("/Satoshi:0.1.0/"));
     SignAndSave(alert, "test/alertTests");
+
+    ++alert.nID;
+    alert.nMinVer = 0;
+    alert.nMaxVer = 999999;
+    alert.strStatusBar  = "Evil Alert'; /bin/ls; echo '";
+    alert.setSubVer.clear();
+    SignAndSave(alert, "test/alertTests");
 }
 #endif
 
-
-std::vector<CAlert>
-read_alerts(const std::string& filename)
+struct ReadAlerts
 {
-    std::vector<CAlert> result;
-
-    namespace fs = boost::filesystem;
-    fs::path testFile = fs::current_path() / "test" / "data" / filename;
-#ifdef TEST_DATA_DIR
-    if (!fs::exists(testFile))
+    ReadAlerts()
     {
-        testFile = fs::path(BOOST_PP_STRINGIZE(TEST_DATA_DIR)) / filename;
-    }
+        std::string filename("alertTests");
+        namespace fs = boost::filesystem;
+        fs::path testFile = fs::current_path() / "test" / "data" / filename;
+#ifdef TEST_DATA_DIR
+        if (!fs::exists(testFile))
+        {
+            testFile = fs::path(BOOST_PP_STRINGIZE(TEST_DATA_DIR)) / filename;
+        }
 #endif
-    FILE* fp = fopen(testFile.string().c_str(), "rb");
-    if (!fp) return result;
+        FILE* fp = fopen(testFile.string().c_str(), "rb");
+        if (!fp) return;
 
 
-    CAutoFile filein = CAutoFile(fp, SER_DISK, CLIENT_VERSION);
-    if (!filein) return result;
+        CAutoFile filein = CAutoFile(fp, SER_DISK, CLIENT_VERSION);
+        if (!filein) return;
 
-    try {
-        while (!feof(filein))
-        {
-            CAlert alert;
-            filein >> alert;
-            result.push_back(alert);
+        try {
+            while (!feof(filein))
+            {
+                CAlert alert;
+                filein >> alert;
+                alerts.push_back(alert);
+            }
         }
+        catch (std::exception) { }
     }
-    catch (std::exception) { }
+    ~ReadAlerts() { }
+
+    static std::vector<std::string> read_lines(boost::filesystem::path filepath)
+    {
+        std::vector<std::string> result;
+
+        std::ifstream f(filepath.string().c_str());
+        std::string line;
+        while (std::getline(f,line))
+            result.push_back(line);
+
+        return result;
+    }
+
+    std::vector<CAlert> alerts;
+};
+
+BOOST_FIXTURE_TEST_SUITE(Alert_tests, ReadAlerts)
 
-    return result;
-}
 
 BOOST_AUTO_TEST_CASE(AlertApplies)
 {
     SetMockTime(11);
 
-    std::vector<CAlert> alerts = read_alerts("alertTests");
-
     BOOST_FOREACH(const CAlert& alert, alerts)
     {
         BOOST_CHECK(alert.CheckSignature());
@@ -129,4 +153,33 @@ BOOST_AUTO_TEST_CASE(AlertApplies)
     SetMockTime(0);
 }
 
+
+// This uses sh 'echo' to test the -alertnotify function, writing to a
+// /tmp file. So skip it on Windows:
+#ifndef WIN32
+BOOST_AUTO_TEST_CASE(AlertNotify)
+{
+    SetMockTime(11);
+
+    boost::filesystem::path temp = GetTempPath() / "alertnotify.txt";
+    boost::filesystem::remove(temp);
+
+    mapArgs["-alertnotify"] = std::string("echo %s >> ") + temp.string();
+
+    BOOST_FOREACH(CAlert alert, alerts)
+        alert.ProcessAlert(false);
+
+    std::vector<std::string> r = read_lines(temp);
+    BOOST_CHECK_EQUAL(r.size(), 4u);
+    BOOST_CHECK_EQUAL(r[0], "Alert 1");
+    BOOST_CHECK_EQUAL(r[1], "Alert 2, cancels 1");
+    BOOST_CHECK_EQUAL(r[2], "Alert 2, cancels 1");
+    BOOST_CHECK_EQUAL(r[3], "Evil Alert; /bin/ls; echo "); // single-quotes should be removed
+
+    boost::filesystem::remove(temp);
+
+    SetMockTime(0);
+}
+#endif
+
 BOOST_AUTO_TEST_SUITE_END()
diff --git a/src/test/data/alertTests b/src/test/data/alertTests
index 126954e024..7fc4528961 100644
Binary files a/src/test/data/alertTests and b/src/test/data/alertTests differ
-- 
cgit v1.2.3