clients: add dig
Change-Id: I78822e9d68b36973362f0790f3c2f6e676cc6b94
diff --git a/tests/unit/clients/iterative-query-controller.cpp b/tests/unit/clients/iterative-query-controller.cpp
new file mode 100644
index 0000000..0346419
--- /dev/null
+++ b/tests/unit/clients/iterative-query-controller.cpp
@@ -0,0 +1,113 @@
+/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
+/**
+ * Copyright (c) 2014, Regents of the University of California.
+ *
+ * This file is part of NDNS (Named Data Networking Domain Name Service).
+ * See AUTHORS.md for complete list of NDNS authors and contributors.
+ *
+ * NDNS is free software: you can redistribute it and/or modify it under the terms
+ * of the GNU General Public License as published by the Free Software Foundation,
+ * either version 3 of the License, or (at your option) any later version.
+ *
+ * NDNS is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
+ * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
+ * PURPOSE. See the GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along with
+ * NDNS, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "clients/iterative-query-controller.hpp"
+#include "daemon/name-server.hpp"
+#include "logger.hpp"
+#include "../database-test-data.hpp"
+#include "../../boost-test.hpp"
+
+#include <ndn-cxx/util/dummy-client-face.hpp>
+
+#include <boost/asio.hpp>
+
+namespace ndn {
+namespace ndns {
+namespace tests {
+NDNS_LOG_INIT("IterativeQueryControllerTest");
+
+class QueryControllerFixture : public DbTestData
+{
+public:
+ QueryControllerFixture()
+ : producerFace(ndn::util::makeDummyClientFace(io, { false, true }))
+ , consumerFace(ndn::util::makeDummyClientFace(io, { false, true }))
+ , validator(*producerFace)
+ , top(m_root.getName(), m_certName, *producerFace, m_session, m_keyChain, validator)
+ , net(m_net.getName(), m_certName, *producerFace, m_session, m_keyChain, validator)
+ , ndnsim(m_ndnsim.getName(), m_certName, *producerFace, m_session, m_keyChain, validator)
+ {
+ run();
+ producerFace->onInterest += [&] (const Interest& interest) { consumerFace->receive(interest); };
+ consumerFace->onInterest += [&] (const Interest& interest) { producerFace->receive(interest); };
+ producerFace->onData += [&] (const Data& data) { consumerFace->receive(data); };
+ consumerFace->onData += [&] (const Data& data) { producerFace->receive(data); };
+ }
+
+ void
+ run()
+ {
+ io.poll();
+ io.reset();
+ }
+
+public:
+ boost::asio::io_service io;
+ shared_ptr<ndn::util::DummyClientFace> producerFace;
+ shared_ptr<ndn::util::DummyClientFace> consumerFace;
+
+ Name hint;
+ Validator validator;
+ ndns::NameServer top;
+ ndns::NameServer net;
+ ndns::NameServer ndnsim;
+};
+
+
+BOOST_AUTO_TEST_SUITE(IterativeQueryController)
+
+BOOST_FIXTURE_TEST_CASE(Basic, QueryControllerFixture)
+{
+ using std::string;
+ using ndns::NameServer;
+
+ string hint;
+ Name name = m_ndnsim.getName();
+ Name dstLabel(name.append("www"));
+ name::Component rrType("TXT");
+ time::milliseconds lifetime(4000);
+
+ bool hasDataBack = false;
+ auto ctr = make_shared<ndns::IterativeQueryController>(
+ dstLabel, rrType, lifetime,
+ [&hasDataBack] (const Data&, const Response&) {
+ hasDataBack = true;
+ BOOST_CHECK(true);
+ },
+ [&hasDataBack] (uint32_t errCode, const std::string& errMsg) {
+ BOOST_CHECK(false);
+ },
+ *consumerFace);
+
+ // IterativeQueryController is a whole process
+ // the tester should not send Interest one by one
+ // instead of starting it and letting it handle Interest/Data automatically
+ ctr->setNFinishedComps(1);
+
+ ctr->start();
+
+ run();
+ BOOST_CHECK_EQUAL(hasDataBack, true);
+}
+
+BOOST_AUTO_TEST_SUITE_END()
+
+} // namespace tests
+} // namespace ndns
+} // namespace ndn