pdns/test-zoneparser_tng_cc.cc

   1 #define BOOST_TEST_DYN_LINK
   2 #define BOOST_TEST_NO_MAIN
   3 #include <boost/test/unit_test.hpp>
   4 #include <boost/assign/list_of.hpp>
   5 #include <boost/foreach.hpp>
   6 #include <boost/tuple/tuple.hpp>
   7 #include <boost/iostreams/stream.hpp>
   8 #include <boost/iostreams/device/file.hpp>
   9 #include <boost/lexical_cast.hpp>
  10 #include "dns.hh"
  11 #include "zoneparser-tng.hh"
  12 #include "dnsrecords.hh"
  13 #include <fstream>
  14 #include <cstdlib>
  15
  16 BOOST_AUTO_TEST_SUITE(test_zoneparser_tng_cc)
  17
  18 BOOST_AUTO_TEST_CASE(test_tng_record_types) {
  19   reportAllTypes();
  20
  21   std::ostringstream pathbuf;
  22   const char* p = std::getenv("SRCDIR");
  23   if(!p)
  24     p = ".";
  25   pathbuf << p << "/../regression-tests/zones/unit.test";
  26   ZoneParserTNG zp(pathbuf.str(), "unit.test");
  27   DNSResourceRecord rr;
  28
  29   ifstream ifs(pathbuf.str());
  30
  31   while(zp.get(rr)) {
  32     // make sure these concur.
  33     std::string host, type, data;
  34     int ttl;
  35     std::getline(ifs, host, ' ');
  36     std::getline(ifs, type, ' ');
  37     ttl = boost::lexical_cast<int>(type);
  38     std::getline(ifs, type, ' ');
  39     std::getline(ifs, type, ' ');
  40     std::getline(ifs, data, '\n');
  41     // see if these agree
  42     BOOST_CHECK_EQUAL(rr.qname, host);
  43     BOOST_CHECK_EQUAL(rr.ttl, ttl);
  44     BOOST_CHECK_EQUAL(rr.qtype.getName(), type);
  45     if (*(rr.content.rbegin()) != '.' && *(data.rbegin()) == '.')
  46       BOOST_CHECK_EQUAL(rr.content, std::string(data.begin(),data.end()-1));
  47     else
  48       BOOST_CHECK_EQUAL(rr.content, data);
  49   }
  50
  51 }
  52
  53 BOOST_AUTO_TEST_SUITE_END();