tests/unit-tests/update-handler.t.cpp - nTorrent - Gitiles

 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
 /**
 * Copyright (c) 2016 Regents of the University of California.
 *
 * This file is part of the nTorrent codebase.
 *
 * nTorrent is free software: you can redistribute it and/or modify it under the
 * terms of the GNU Lesser General Public License as published by the Free Software
 * Foundation, either version 3 of the License, or (at your option) any later version.
 *
 * nTorrent 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 Lesser General Public License for more details.
 *
 * You should have received copies of the GNU General Public License and GNU Lesser
 * General Public License along with nTorrent, e.g., in COPYING.md file. If not, see
 * <http://www.gnu.org/licenses/>.
 *
 * See AUTHORS for complete list of nTorrent authors and contributors.
 */

 #include "boost-test.hpp"
 #include "update-handler.hpp"
 #include "unit-test-time-fixture.hpp"
 #include "dummy-parser-fixture.hpp"

 #include <ndn-cxx/name.hpp>
 #include <ndn-cxx/util/dummy-client-face.hpp>
 #include <ndn-cxx/util/io.hpp>

 namespace ndn {
 namespace ntorrent {
 namespace tests {

 using util::DummyClientFace;
 using std::vector;

 class TestUpdateHandler : public UpdateHandler {
 public:
   TestUpdateHandler(Name torrentName, shared_ptr<KeyChain> keyChain,
                     shared_ptr<StatsTable> statsTable, shared_ptr<Face> face)
   : UpdateHandler(torrentName, keyChain, statsTable, face)
   {
   }

   ~TestUpdateHandler()
   {
   }

   Name
   getOwnRoutablePrefix()
   {
     return UpdateHandler::getOwnRoutablePrefix();
   }
 };

 class FaceFixture : public UnitTestTimeFixture
 {
 public:
   explicit
   FaceFixture()
   : face1(new DummyClientFace(io, {true, true}))
   , face2(new DummyClientFace(io, {true, true}))
   {
   }

 public:
   shared_ptr<DummyClientFace> face1;
   shared_ptr<DummyClientFace> face2;
 };

 BOOST_FIXTURE_TEST_SUITE(TestUpdateHandlerClass, FaceFixture)

 BOOST_AUTO_TEST_CASE(TestInitialSetup)
 {
   shared_ptr<StatsTable> table1 = make_shared<StatsTable>(Name("linux15.01"));

   shared_ptr<KeyChain> keyChain = make_shared<KeyChain>();

   TestUpdateHandler handler1(Name("linux15.01"), keyChain, table1, face1);
   advanceClocks(time::milliseconds(1), 10);
   // Create a data packet containing one name as content
   shared_ptr<Data> d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
                                                       { Name("ucla") });
   keyChain->sign(*d);
   face1->receive(*d);

   BOOST_CHECK_EQUAL(handler1.getOwnRoutablePrefix().toUri(), "/ucla");

   shared_ptr<StatsTable> table2 = make_shared<StatsTable>(Name("linux15.01"));
   TestUpdateHandler handler2(Name("linux15.01"), keyChain, table2, face2);
   advanceClocks(time::milliseconds(1), 10);
   // Create a data packet containing one name as content
   d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
                                      { Name("arizona") });
   keyChain->sign(*d);
   face2->receive(*d);

   BOOST_CHECK_EQUAL(handler2.getOwnRoutablePrefix().toUri(), "/arizona");
 }

 BOOST_AUTO_TEST_CASE(TestAliveInterestExchange)
 {
   shared_ptr<StatsTable> table1 = make_shared<StatsTable>(Name("linux15.01"));
   table1->insert(Name("isp1"));
   table1->insert(Name("isp2"));
   table1->insert(Name("isp3"));

   shared_ptr<KeyChain> keyChain = make_shared<KeyChain>();

   TestUpdateHandler handler1(Name("linux15.01"), keyChain, table1, face1);
   advanceClocks(time::milliseconds(1), 10);
   // Create a data packet containing one name as content
   shared_ptr<Data> d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
                                                       { Name("ucla") });
   keyChain->sign(*d);
   face1->receive(*d);

   shared_ptr<StatsTable> table2 = make_shared<StatsTable>(Name("linux15.01"));
   table2->insert(Name("ucla"));
   TestUpdateHandler handler2(Name("linux15.01"), keyChain, table2, face2);
   advanceClocks(time::milliseconds(1), 10);
   // Create a data packet containing one name as content
   d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
                                      { Name("arizona") });
   keyChain->sign(*d);
   face2->receive(*d);

   handler2.sendAliveInterest(table2->begin());

   advanceClocks(time::milliseconds(1), 40);
   Interest interest(Name("/NTORRENT/linux15.01/ALIVE/test"));
   face1->receive(interest);

   advanceClocks(time::milliseconds(1), 10);
   std::vector<Data> dataVec = face1->sentData;

   BOOST_CHECK_EQUAL(dataVec.size(), 1);
   BOOST_CHECK_EQUAL(dataVec[0].getName().toUri(), "/NTORRENT/linux15.01/ALIVE/test");

   auto block = dataVec[0].getContent();
   shared_ptr<vector<Name>> nameVec = DummyParser::decodeContent(block);
   auto it = nameVec->begin();

   BOOST_CHECK_EQUAL(it->toUri(), "/test");
   ++it;
   BOOST_CHECK_EQUAL(it->toUri(), "/isp3");
   ++it;
   BOOST_CHECK_EQUAL(it->toUri(), "/isp2");
   ++it;
   BOOST_CHECK_EQUAL(it->toUri(), "/isp1");

   BOOST_CHECK_EQUAL((table1->end() - 1)->getRecordName().toUri(), "/test");

   d = DummyParser::createDataPacket(Name("/NTORRENT/linux15.01/ALIVE/arizona"),
                                      { Name("isp1"), Name("isp2"), Name("isp3") });
   keyChain->sign(*d);

   advanceClocks(time::milliseconds(1), 40);
   face2->receive(*d);

   auto i = table2->begin();
   ++i;
   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp1");
   ++i;
   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp2");
   ++i;
   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp3");

   table1->erase(Name("isp1"));
   table1->erase(Name("isp2"));

   table1->insert(Name("isp4"));
   table1->insert(Name("isp5"));
   table1->insert(Name("isp6"));
   table1->insert(Name("isp7"));
   table1->insert(Name("isp8"));
   table1->insert(Name("isp9"));

   handler2.sendAliveInterest(table2->begin());

   advanceClocks(time::milliseconds(1), 40);
   Interest interest2(Name("/NTORRENT/linux15.01/ALIVE/arizona"));
   face1->receive(interest2);

   advanceClocks(time::milliseconds(1), 10);

   dataVec = face1->sentData;
   BOOST_CHECK_EQUAL(dataVec.size(), 2);

   auto iter = dataVec.begin() + 1;
   advanceClocks(time::milliseconds(1), 30);
   face2->receive(*iter);

   i = table2->begin();
   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/ucla");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp1");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp2");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp3");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/test");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp4");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp5");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp6");
   BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
   BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
   BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
   ++i;

   BOOST_CHECK(i == table2->end());
 }

 BOOST_AUTO_TEST_CASE(TestNeedsUpdate)
 {
   shared_ptr<StatsTable> table1 = make_shared<StatsTable>(Name("linux15.01"));
   table1->insert(Name("isp1"));
   table1->insert(Name("isp2"));
   table1->insert(Name("isp3"));

   shared_ptr<KeyChain> keyChain = make_shared<KeyChain>();

   UpdateHandler handler1(Name("linux15.01"), keyChain, table1, face1);

   BOOST_CHECK(handler1.needsUpdate());

   auto i = table1->begin() + 1;
   i->incrementSentInterests();
   i->incrementSentInterests();
   i->incrementReceivedData();

   BOOST_CHECK(handler1.needsUpdate());

   table1->insert(Name("isp4"));
   table1->insert(Name("isp5"));

   BOOST_CHECK(!handler1.needsUpdate());
 }

 BOOST_AUTO_TEST_SUITE_END()

 } // namespace tests
 } // namespace ntorrent
 } // namespace ndn
	/* -- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -- */
	/**
	* Copyright (c) 2016 Regents of the University of California.
	*
	* This file is part of the nTorrent codebase.
	*
	* nTorrent is free software: you can redistribute it and/or modify it under the
	* terms of the GNU Lesser General Public License as published by the Free Software
	* Foundation, either version 3 of the License, or (at your option) any later version.
	*
	* nTorrent 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 Lesser General Public License for more details.
	*
	* You should have received copies of the GNU General Public License and GNU Lesser
	* General Public License along with nTorrent, e.g., in COPYING.md file. If not, see
	* <http://www.gnu.org/licenses/>.
	*
	* See AUTHORS for complete list of nTorrent authors and contributors.
	*/

	#include "boost-test.hpp"
	#include "update-handler.hpp"
	#include "unit-test-time-fixture.hpp"
	#include "dummy-parser-fixture.hpp"

	#include <ndn-cxx/name.hpp>
	#include <ndn-cxx/util/dummy-client-face.hpp>
	#include <ndn-cxx/util/io.hpp>

	namespace ndn {
	namespace ntorrent {
	namespace tests {

	using util::DummyClientFace;
	using std::vector;

	class TestUpdateHandler : public UpdateHandler {
	public:
	TestUpdateHandler(Name torrentName, shared_ptr<KeyChain> keyChain,
	shared_ptr<StatsTable> statsTable, shared_ptr<Face> face)
	: UpdateHandler(torrentName, keyChain, statsTable, face)
	{
	}

	~TestUpdateHandler()
	{
	}

	Name
	getOwnRoutablePrefix()
	{
	return UpdateHandler::getOwnRoutablePrefix();
	}
	};

	class FaceFixture : public UnitTestTimeFixture
	{
	public:
	explicit
	FaceFixture()
	: face1(new DummyClientFace(io, {true, true}))
	, face2(new DummyClientFace(io, {true, true}))
	{
	}

	public:
	shared_ptr<DummyClientFace> face1;
	shared_ptr<DummyClientFace> face2;
	};

	BOOST_FIXTURE_TEST_SUITE(TestUpdateHandlerClass, FaceFixture)

	BOOST_AUTO_TEST_CASE(TestInitialSetup)
	{
	shared_ptr<StatsTable> table1 = make_shared<StatsTable>(Name("linux15.01"));

	shared_ptr<KeyChain> keyChain = make_shared<KeyChain>();

	TestUpdateHandler handler1(Name("linux15.01"), keyChain, table1, face1);
	advanceClocks(time::milliseconds(1), 10);
	// Create a data packet containing one name as content
	shared_ptr<Data> d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
	{ Name("ucla") });
	keyChain->sign(*d);
	face1->receive(*d);

	BOOST_CHECK_EQUAL(handler1.getOwnRoutablePrefix().toUri(), "/ucla");

	shared_ptr<StatsTable> table2 = make_shared<StatsTable>(Name("linux15.01"));
	TestUpdateHandler handler2(Name("linux15.01"), keyChain, table2, face2);
	advanceClocks(time::milliseconds(1), 10);
	// Create a data packet containing one name as content
	d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
	{ Name("arizona") });
	keyChain->sign(*d);
	face2->receive(*d);

	BOOST_CHECK_EQUAL(handler2.getOwnRoutablePrefix().toUri(), "/arizona");
	}

	BOOST_AUTO_TEST_CASE(TestAliveInterestExchange)
	{
	shared_ptr<StatsTable> table1 = make_shared<StatsTable>(Name("linux15.01"));
	table1->insert(Name("isp1"));
	table1->insert(Name("isp2"));
	table1->insert(Name("isp3"));

	shared_ptr<KeyChain> keyChain = make_shared<KeyChain>();

	TestUpdateHandler handler1(Name("linux15.01"), keyChain, table1, face1);
	advanceClocks(time::milliseconds(1), 10);
	// Create a data packet containing one name as content
	shared_ptr<Data> d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
	{ Name("ucla") });
	keyChain->sign(*d);
	face1->receive(*d);

	shared_ptr<StatsTable> table2 = make_shared<StatsTable>(Name("linux15.01"));
	table2->insert(Name("ucla"));
	TestUpdateHandler handler2(Name("linux15.01"), keyChain, table2, face2);
	advanceClocks(time::milliseconds(1), 10);
	// Create a data packet containing one name as content
	d = DummyParser::createDataPacket(Name("/localhop/nfd/rib/routable-prefixes"),
	{ Name("arizona") });
	keyChain->sign(*d);
	face2->receive(*d);

	handler2.sendAliveInterest(table2->begin());

	advanceClocks(time::milliseconds(1), 40);
	Interest interest(Name("/NTORRENT/linux15.01/ALIVE/test"));
	face1->receive(interest);

	advanceClocks(time::milliseconds(1), 10);
	std::vector<Data> dataVec = face1->sentData;

	BOOST_CHECK_EQUAL(dataVec.size(), 1);
	BOOST_CHECK_EQUAL(dataVec[0].getName().toUri(), "/NTORRENT/linux15.01/ALIVE/test");

	auto block = dataVec[0].getContent();
	shared_ptr<vector<Name>> nameVec = DummyParser::decodeContent(block);
	auto it = nameVec->begin();

	BOOST_CHECK_EQUAL(it->toUri(), "/test");
	++it;
	BOOST_CHECK_EQUAL(it->toUri(), "/isp3");
	++it;
	BOOST_CHECK_EQUAL(it->toUri(), "/isp2");
	++it;
	BOOST_CHECK_EQUAL(it->toUri(), "/isp1");

	BOOST_CHECK_EQUAL((table1->end() - 1)->getRecordName().toUri(), "/test");

	d = DummyParser::createDataPacket(Name("/NTORRENT/linux15.01/ALIVE/arizona"),
	{ Name("isp1"), Name("isp2"), Name("isp3") });
	keyChain->sign(*d);

	advanceClocks(time::milliseconds(1), 40);
	face2->receive(*d);

	auto i = table2->begin();
	++i;
	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp1");
	++i;
	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp2");
	++i;
	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp3");

	table1->erase(Name("isp1"));
	table1->erase(Name("isp2"));

	table1->insert(Name("isp4"));
	table1->insert(Name("isp5"));
	table1->insert(Name("isp6"));
	table1->insert(Name("isp7"));
	table1->insert(Name("isp8"));
	table1->insert(Name("isp9"));

	handler2.sendAliveInterest(table2->begin());

	advanceClocks(time::milliseconds(1), 40);
	Interest interest2(Name("/NTORRENT/linux15.01/ALIVE/arizona"));
	face1->receive(interest2);

	advanceClocks(time::milliseconds(1), 10);

	dataVec = face1->sentData;
	BOOST_CHECK_EQUAL(dataVec.size(), 2);

	auto iter = dataVec.begin() + 1;
	advanceClocks(time::milliseconds(1), 30);
	face2->receive(*iter);

	i = table2->begin();
	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/ucla");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp1");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp2");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp3");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/test");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp4");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp5");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK_EQUAL(i->getRecordName().toUri(), "/isp6");
	BOOST_CHECK_EQUAL(i->getRecordSuccessRate(), 0);
	BOOST_CHECK_EQUAL(i->getRecordSentInterests(), 0);
	BOOST_CHECK_EQUAL(i->getRecordReceivedData(), 0);
	++i;

	BOOST_CHECK(i == table2->end());
	}

	BOOST_AUTO_TEST_CASE(TestNeedsUpdate)
	{
	shared_ptr<StatsTable> table1 = make_shared<StatsTable>(Name("linux15.01"));
	table1->insert(Name("isp1"));
	table1->insert(Name("isp2"));
	table1->insert(Name("isp3"));

	shared_ptr<KeyChain> keyChain = make_shared<KeyChain>();

	UpdateHandler handler1(Name("linux15.01"), keyChain, table1, face1);

	BOOST_CHECK(handler1.needsUpdate());

	auto i = table1->begin() + 1;
	i->incrementSentInterests();
	i->incrementSentInterests();
	i->incrementReceivedData();

	BOOST_CHECK(handler1.needsUpdate());

	table1->insert(Name("isp4"));
	table1->insert(Name("isp5"));

	BOOST_CHECK(!handler1.needsUpdate());
	}

	BOOST_AUTO_TEST_SUITE_END()

	} // namespace tests
	} // namespace ntorrent
	} // namespace ndn