dnsdist: Add HTTPStatusAction to return a specific HTTP response

[thirdparty/pdns.git] / pdns / test-misc_hh.cc

diff --git a/pdns/test-misc_hh.cc b/pdns/test-misc_hh.cc
index 0e7f37b70fc6828585c3a547dea2186ffb725276..499d6ecf0f0224fc24b2c9a29c2c5cf11e77b720 100644 (file)
--- a/pdns/test-misc_hh.cc
+++ b/pdns/test-misc_hh.cc
@@ -1,54 +1,58 @@
 #define BOOST_TEST_DYN_LINK
 #define BOOST_TEST_NO_MAIN
 
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
 #include <boost/test/unit_test.hpp>
 #include <boost/assign/list_of.hpp>
-#include <boost/foreach.hpp>
+
 #include <boost/tuple/tuple.hpp>
 #include "misc.hh"
 #include "dns.hh"
+#include <arpa/inet.h>
 #include <utility>
 
 using std::string;
 
-BOOST_AUTO_TEST_SUITE(misc_hh)
+BOOST_AUTO_TEST_SUITE(test_misc_hh)
 typedef pair<std::string, uint16_t> typedns_t;
 
 BOOST_AUTO_TEST_CASE(test_CIStringCompare) {
-       set<std::string, CIStringCompare> nsset;
-       nsset.insert("abc");
-       nsset.insert("ns.example.com");
-       nsset.insert("");
-       nsset.insert("def");
-       nsset.insert("aBc");
-       nsset.insert("ns.example.com");
-       BOOST_CHECK_EQUAL(nsset.size(), 4);
-
-       ostringstream s;
-       for(set<std::string, CIStringCompare>::const_iterator i=nsset.begin();i!=nsset.end();++i) {
-               s<<"("<<*i<<")";
-       }
-       BOOST_CHECK_EQUAL(s.str(), "()(abc)(def)(ns.example.com)");
+        set<std::string, CIStringCompare> nsset;
+        nsset.insert("abc");
+        nsset.insert("ns.example.com");
+        nsset.insert("");
+        nsset.insert("def");
+        nsset.insert("aBc");
+        nsset.insert("ns.example.com");
+        BOOST_CHECK_EQUAL(nsset.size(), 4);
+
+        ostringstream s;
+        for(set<std::string, CIStringCompare>::const_iterator i=nsset.begin();i!=nsset.end();++i) {
+                s<<"("<<*i<<")";
+        }
+        BOOST_CHECK_EQUAL(s.str(), "()(abc)(def)(ns.example.com)");
 }
 
 BOOST_AUTO_TEST_CASE(test_CIStringPairCompare) {
-       set<typedns_t, CIStringPairCompare> nsset2;
-       nsset2.insert(make_pair("ns.example.com", 1));
-       nsset2.insert(make_pair("abc", 1));
-       nsset2.insert(make_pair("", 1));
-       nsset2.insert(make_pair("def", 1));
-       nsset2.insert(make_pair("abc", 2));
-       nsset2.insert(make_pair("abc", 1));
-       nsset2.insert(make_pair("ns.example.com", 0));
-       nsset2.insert(make_pair("abc", 2));
-       nsset2.insert(make_pair("ABC", 2));
-       BOOST_CHECK_EQUAL(nsset2.size(), 6);
-
-       ostringstream s;
-       for(set<typedns_t, CIStringPairCompare>::const_iterator i=nsset2.begin();i!=nsset2.end();++i) {
-               s<<"("<<i->first<<"|"<<i->second<<")";
-       }
-       BOOST_CHECK_EQUAL(s.str(), "(|1)(abc|1)(abc|2)(def|1)(ns.example.com|0)(ns.example.com|1)");
+        set<typedns_t, CIStringPairCompare> nsset2;
+        nsset2.insert(make_pair("ns.example.com", 1));
+        nsset2.insert(make_pair("abc", 1));
+        nsset2.insert(make_pair("", 1));
+        nsset2.insert(make_pair("def", 1));
+        nsset2.insert(make_pair("abc", 2));
+        nsset2.insert(make_pair("abc", 1));
+        nsset2.insert(make_pair("ns.example.com", 0));
+        nsset2.insert(make_pair("abc", 2));
+        nsset2.insert(make_pair("ABC", 2));
+        BOOST_CHECK_EQUAL(nsset2.size(), 6);
+
+        ostringstream s;
+        for(set<typedns_t, CIStringPairCompare>::const_iterator i=nsset2.begin();i!=nsset2.end();++i) {
+                s<<"("<<i->first<<"|"<<i->second<<")";
+        }
+        BOOST_CHECK_EQUAL(s.str(), "(|1)(abc|1)(abc|2)(def|1)(ns.example.com|0)(ns.example.com|1)");
 }
 
 BOOST_AUTO_TEST_CASE(test_pdns_ilexicographical_compare) {
@@ -67,7 +71,7 @@ BOOST_AUTO_TEST_CASE(test_pdns_ilexicographical_compare) {
     (case_t(std::string("Abc"), std::string("abc"), false))
   ;
 
-  BOOST_FOREACH(const case_t& val, cases) {
+  for(const case_t& val :  cases) {
     bool res;
     res = pdns_ilexicographical_compare(val.get<0>(), val.get<1>());
     BOOST_CHECK_EQUAL(res, val.get<2>());
@@ -90,7 +94,7 @@ BOOST_AUTO_TEST_CASE(test_pdns_iequals) {
     (case_t(std::string("Abc"), std::string("abc"), true))
   ;
 
-  BOOST_FOREACH(const case_t& val, cases) {
+  for(const case_t& val :  cases) {
     bool res;
     res = pdns_iequals(val.get<0>(), val.get<1>());
     BOOST_CHECK_EQUAL(res, val.get<2>());
@@ -105,15 +109,12 @@ BOOST_AUTO_TEST_CASE(test_stripDot) {
 }
 
 BOOST_AUTO_TEST_CASE(test_labelReverse) {
-    BOOST_CHECK_EQUAL(labelReverse("www.powerdns.com"), "com powerdns www");
+  BOOST_CHECK_EQUAL(DNSName("www.powerdns.com").labelReverse().toString(" ", false), "com powerdns www");
 }
 
-BOOST_AUTO_TEST_CASE(test_makeRelative) {
-    BOOST_CHECK_EQUAL(makeRelative("www.powerdns.com", "powerdns.com"), "www");
-}
 
 BOOST_AUTO_TEST_CASE(test_AtomicCounter) {
-    AtomicCounter ac;
+    AtomicCounter ac(0);
     ++ac;
     ++ac;
     BOOST_CHECK_EQUAL(ac, 2);
@@ -140,5 +141,66 @@ BOOST_AUTO_TEST_CASE(test_parseService) {
     BOOST_CHECK_EQUAL(tp.port, 25);
 }
 
+BOOST_AUTO_TEST_CASE(test_ternary) {
+  int maxqps=1024;
+  BOOST_CHECK_EQUAL(defTer(maxqps, 16384), maxqps);
+  BOOST_CHECK_EQUAL(defTer(0, 16384), 16384);
+
+  int* qps=0;
+  BOOST_CHECK_EQUAL(*defTer(qps, &maxqps), 1024);
+}
+
+BOOST_AUTO_TEST_CASE(test_SimpleMatch) {
+  BOOST_CHECK_EQUAL(SimpleMatch("").match(std::string("")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("?").match(std::string("")), false);
+  BOOST_CHECK_EQUAL(SimpleMatch("*").match(std::string("")), true);
+
+  BOOST_CHECK_EQUAL(SimpleMatch("abc").match(std::string("abc")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("abc").match(std::string("ab")), false);
+  BOOST_CHECK_EQUAL(SimpleMatch("abc").match(std::string("bc")), false);
+
+  BOOST_CHECK_EQUAL(SimpleMatch("?").match(std::string("a")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("a?c").match(std::string("abc")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("a?c").match(std::string("ab")), false);
+  BOOST_CHECK_EQUAL(SimpleMatch("a?c").match(std::string("bc")), false);
+
+  BOOST_CHECK_EQUAL(SimpleMatch("*").match(std::string("*")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("a*c").match(std::string("abc")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("a*c").match(std::string("ab")), false);
+  BOOST_CHECK_EQUAL(SimpleMatch("a*c").match(std::string("bc")), false);
+
+  BOOST_CHECK_EQUAL(SimpleMatch("*").match(std::string("abcdefghj")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("*a").match(std::string("abca")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("*a").match(std::string("abcb")), false);
+  BOOST_CHECK_EQUAL(SimpleMatch("abc*").match(std::string("abcabcabcabacabac")), true);
+  BOOST_CHECK_EQUAL(SimpleMatch("abc*").match(std::string("abc")), true);
+}
+
+template<typename T> bool rfc1982check(T x, T y) {
+  return rfc1982LessThan(x, y);
+}
+
+BOOST_AUTO_TEST_CASE(test_rfc1982LessThan) {
+  // The test cases from rfc1982 section 5.2
+  BOOST_CHECK(rfc1982check<uint8_t>(0, 1));
+  BOOST_CHECK(rfc1982check<uint8_t>(0, 44));
+  BOOST_CHECK(rfc1982check<uint8_t>(0, 100));
+  BOOST_CHECK(rfc1982check<uint8_t>(44, 100));
+  BOOST_CHECK(rfc1982check<uint8_t>(100, 200));
+  BOOST_CHECK(rfc1982check<uint8_t>(200, 255));
+  BOOST_CHECK(rfc1982check<uint8_t>(255, 0));
+  BOOST_CHECK(rfc1982check<uint8_t>(255, 100));
+  BOOST_CHECK(rfc1982check<uint8_t>(200, 0));
+  BOOST_CHECK(rfc1982check<uint8_t>(200, 44));
+
+  BOOST_CHECK(rfc1982check<uint32_t>(0, 1));
+  BOOST_CHECK(rfc1982check<uint32_t>(UINT32_MAX-10, 1));
+  BOOST_CHECK(rfc1982check<uint32_t>(UINT32_MAX/2, UINT32_MAX-10));
+
+  BOOST_CHECK(rfc1982check<uint64_t>(0, 1));
+  BOOST_CHECK(rfc1982check<uint64_t>(UINT64_MAX-10, 1));
+  BOOST_CHECK(rfc1982check<uint64_t>(UINT64_MAX/2, UINT64_MAX-10));
+}
+
 BOOST_AUTO_TEST_SUITE_END()