aboutsummaryrefslogtreecommitdiff
path: root/src/test
diff options
context:
space:
mode:
authorGavin Andresen <gavinandresen@gmail.com>2013-03-19 14:08:21 -0400
committerGavin Andresen <gavinandresen@gmail.com>2013-03-19 15:16:30 -0400
commite5f163a041d5a45ea72448e11cfc30abb16f10b6 (patch)
tree9cf3a9a396bac00f28466421393deb19db8ab372 /src/test
parent1472308d679fe7ed165c203070335fd58024d4da (diff)
-alertnotify=<cmd>
Runs a shell command when an AppliesToMe() alert is received. %s in the <cmd> string is replaced with the alert.strStatusBar message.
Diffstat (limited to 'src/test')
-rw-r--r--src/test/alert_tests.cpp111
-rw-r--r--src/test/data/alertTestsbin1123 -> 1283 bytes
2 files changed, 82 insertions, 29 deletions
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
--- a/src/test/data/alertTests
+++ b/src/test/data/alertTests
Binary files differ