tests/test-lsdb.cpp - NLSR - Gitiles

 /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
 /**
  * Copyright (c) 2014-2015,  The University of Memphis,
  *                           Regents of the University of California,
  *                           Arizona Board of Regents.
  *
  * This file is part of NLSR (Named-data Link State Routing).
  * See AUTHORS.md for complete list of NLSR authors and contributors.
  *
  * NLSR 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.
  *
  * NLSR 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
  * NLSR, e.g., in COPYING.md file.  If not, see <http://www.gnu.org/licenses/>.
  **/

 #include "test-common.hpp"
 #include "dummy-face.hpp"

 #include "lsdb.hpp"
 #include "nlsr.hpp"
 #include "lsa.hpp"
 #include "name-prefix-list.hpp"
 #include <boost/test/unit_test.hpp>

 namespace nlsr {
 namespace test {

 using ndn::DummyFace;
 using ndn::shared_ptr;

 class LsdbFixture : public BaseFixture
 {
 public:
   LsdbFixture()
     : face(ndn::makeDummyFace())
     , nlsr(g_ioService, g_scheduler, ndn::ref(*face))
     , sync(*face, nlsr.getLsdb(), nlsr.getConfParameter(), nlsr.getSequencingManager())
     , lsdb(nlsr.getLsdb())
     , conf(nlsr.getConfParameter())
     , REGISTER_COMMAND_PREFIX("/localhost/nfd/rib")
     , REGISTER_VERB("register")
   {
     conf.setNetwork("/ndn");
     conf.setSiteName("/site");
     conf.setRouterName("/%C1.router/this-router");

     nlsr.initialize();

     face->processEvents(ndn::time::milliseconds(1));
     face->m_sentInterests.clear();

     INIT_LOGGERS("/tmp", "DEBUG");
   }

   void
   extractParameters(ndn::Interest& interest, ndn::Name::Component& verb,
                     ndn::nfd::ControlParameters& extractedParameters)
   {
     const ndn::Name& name = interest.getName();
     verb = name[REGISTER_COMMAND_PREFIX.size()];
     const ndn::Name::Component& parameterComponent = name[REGISTER_COMMAND_PREFIX.size() + 1];

     ndn::Block rawParameters = parameterComponent.blockFromValue();
     extractedParameters.wireDecode(rawParameters);
   }

   void
   areNamePrefixListsEqual(NamePrefixList& lhs, NamePrefixList& rhs)
   {
     typedef std::list<ndn::Name> NameList;

     NameList& lhsList = lhs.getNameList();
     NameList& rhsList = rhs.getNameList();

     BOOST_REQUIRE_EQUAL(lhsList.size(), rhsList.size());

     NameList::iterator i = lhsList.begin();
     NameList::iterator j = rhsList.begin();

     for (; i != lhsList.end(); ++i, ++j) {
       BOOST_CHECK_EQUAL(*i, *j);
     }
   }

 public:
   shared_ptr<DummyFace> face;
   Nlsr nlsr;
   SyncLogicHandler sync;

   Lsdb& lsdb;
   ConfParameter& conf;

   ndn::Name REGISTER_COMMAND_PREFIX;
   ndn::Name::Component REGISTER_VERB;
 };

 BOOST_FIXTURE_TEST_SUITE(TestLsdb, LsdbFixture)

 BOOST_AUTO_TEST_CASE(LsdbSync)
 {
   ndn::Name interestName("/ndn/NLSR/LSA/cs/%C1.Router/router2/name");
   uint64_t oldSeqNo = 82;

   ndn::Name oldInterestName = interestName;
   oldInterestName.appendNumber(oldSeqNo);

   lsdb.expressInterest(oldInterestName, 0);
   face->processEvents(ndn::time::milliseconds(1));

   std::vector<ndn::Interest>& interests = face->m_sentInterests;

   BOOST_REQUIRE(interests.size() > 0);
   std::vector<ndn::Interest>::iterator it = interests.begin();

   BOOST_CHECK_EQUAL(it->getName(), oldInterestName);
   interests.clear();

   steady_clock::TimePoint deadline = steady_clock::now() +
                                      ndn::time::seconds(static_cast<int>(LSA_REFRESH_TIME_MAX));

   // Simulate an LSA interest timeout
   lsdb.processInterestTimedOut(oldInterestName, 0, deadline, interestName, oldSeqNo);
   face->processEvents(ndn::time::milliseconds(1));

   BOOST_REQUIRE(interests.size() > 0);
   it = interests.begin();

   BOOST_CHECK_EQUAL(it->getName(), oldInterestName);
   interests.clear();

   uint64_t newSeqNo = 83;

   ndn::Name newInterestName = interestName;
   newInterestName.appendNumber(newSeqNo);

   lsdb.expressInterest(newInterestName, 0);
   face->processEvents(ndn::time::milliseconds(1));

   BOOST_REQUIRE(interests.size() > 0);
   it = interests.begin();

   BOOST_CHECK_EQUAL(it->getName(), newInterestName);
   interests.clear();

   // Simulate an LSA interest timeout where the sequence number is outdated
   lsdb.processInterestTimedOut(oldInterestName, 0, deadline, interestName, oldSeqNo);
   face->processEvents(ndn::time::milliseconds(1));

   // Interest should not be expressed for outdated sequence number
   BOOST_CHECK_EQUAL(interests.size(), 0);
 }

 BOOST_AUTO_TEST_CASE(LsdbRemoveAndExists)
 {
   ndn::time::system_clock::TimePoint testTimePoint =  ndn::time::system_clock::now();
   NamePrefixList npl1;

   std::string s1 = "name1";
   std::string s2 = "name2";
   std::string router1 = "router1/1";

   npl1.insert(s1);
   npl1.insert(s2);

   //For NameLsa lsType is name.
   //12 is seqNo, randomly generated.
   //1800 is the default life time.
   NameLsa nlsa1(ndn::Name("/router1/1"), NameLsa::TYPE_STRING, 12, testTimePoint, npl1);

   Lsdb lsdb1(nlsr, g_scheduler, nlsr.getSyncLogicHandler());

   lsdb1.installNameLsa(nlsa1);
   lsdb1.writeNameLsdbLog();

   BOOST_CHECK(lsdb1.doesLsaExist(ndn::Name("/router1/1/name"), NameLsa::TYPE_STRING));

   lsdb1.removeNameLsa(router1);

   BOOST_CHECK_EQUAL(lsdb1.doesLsaExist(ndn::Name("/router1/1"), NameLsa::TYPE_STRING), false);
 }

 BOOST_AUTO_TEST_CASE(InstallNameLsa)
 {
   // Install lsa with name1 and name2
   ndn::Name name1("/ndn/name1");
   ndn::Name name2("/ndn/name2");

   NamePrefixList prefixes;
   prefixes.insert(name1);
   prefixes.insert(name2);

   std::string otherRouter("/ndn/site/%C1.router/other-router");
   ndn::time::system_clock::TimePoint MAX_TIME = ndn::time::system_clock::TimePoint::max();

   NameLsa lsa(otherRouter, NameLsa::TYPE_STRING, 1, MAX_TIME, prefixes);
   lsdb.installNameLsa(lsa);

   BOOST_REQUIRE_EQUAL(lsdb.doesLsaExist(otherRouter + "/name", NameLsa::TYPE_STRING), true);
   NamePrefixList& nameList = lsdb.findNameLsa(otherRouter + "/name")->getNpl();

   areNamePrefixListsEqual(nameList, prefixes);

   // Add a prefix: name3
   ndn::Name name3("/ndn/name3");
   prefixes.insert(name3);

   NameLsa addLsa(otherRouter, NameLsa::TYPE_STRING, 2, MAX_TIME, prefixes);
   lsdb.installNameLsa(addLsa);

   // Lsa should include name1, name2, and name3
   areNamePrefixListsEqual(nameList, prefixes);

   // Remove a prefix: name2
   prefixes.remove(name2);

   NameLsa removeLsa(otherRouter, NameLsa::TYPE_STRING, 3, MAX_TIME, prefixes);
   lsdb.installNameLsa(removeLsa);

   // Lsa should include name1 and name3
   areNamePrefixListsEqual(nameList, prefixes);

   // Add and remove a prefix: add name2, remove name3
   prefixes.insert(name2);
   prefixes.remove(name3);

   NameLsa addAndRemoveLsa(otherRouter, NameLsa::TYPE_STRING, 4, MAX_TIME, prefixes);
   lsdb.installNameLsa(addAndRemoveLsa);

   // Lsa should include name1 and name2
   areNamePrefixListsEqual(nameList, prefixes);

   // Install a completely new list of prefixes
   ndn::Name name4("/ndn/name4");
   ndn::Name name5("/ndn/name5");

   NamePrefixList newPrefixes;
   newPrefixes.insert(name4);
   newPrefixes.insert(name5);

   NameLsa newLsa(otherRouter, NameLsa::TYPE_STRING, 5, MAX_TIME, newPrefixes);
   lsdb.installNameLsa(newLsa);

   // Lsa should include name4 and name5
   areNamePrefixListsEqual(nameList, newPrefixes);
 }

 BOOST_AUTO_TEST_SUITE_END()

 } //namespace test
 } //namespace nlsr
	/* -- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -- */
	/**
	* Copyright (c) 2014-2015, The University of Memphis,
	* Regents of the University of California,
	* Arizona Board of Regents.
	*
	* This file is part of NLSR (Named-data Link State Routing).
	* See AUTHORS.md for complete list of NLSR authors and contributors.
	*
	* NLSR 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.
	*
	* NLSR 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
	* NLSR, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
	**/

	#include "test-common.hpp"
	#include "dummy-face.hpp"

	#include "lsdb.hpp"
	#include "nlsr.hpp"
	#include "lsa.hpp"
	#include "name-prefix-list.hpp"
	#include <boost/test/unit_test.hpp>

	namespace nlsr {
	namespace test {

	using ndn::DummyFace;
	using ndn::shared_ptr;

	class LsdbFixture : public BaseFixture
	{
	public:
	LsdbFixture()
	: face(ndn::makeDummyFace())
	, nlsr(g_ioService, g_scheduler, ndn::ref(*face))
	, sync(*face, nlsr.getLsdb(), nlsr.getConfParameter(), nlsr.getSequencingManager())
	, lsdb(nlsr.getLsdb())
	, conf(nlsr.getConfParameter())
	, REGISTER_COMMAND_PREFIX("/localhost/nfd/rib")
	, REGISTER_VERB("register")
	{
	conf.setNetwork("/ndn");
	conf.setSiteName("/site");
	conf.setRouterName("/%C1.router/this-router");

	nlsr.initialize();

	face->processEvents(ndn::time::milliseconds(1));
	face->m_sentInterests.clear();

	INIT_LOGGERS("/tmp", "DEBUG");
	}

	void
	extractParameters(ndn::Interest& interest, ndn::Name::Component& verb,
	ndn::nfd::ControlParameters& extractedParameters)
	{
	const ndn::Name& name = interest.getName();
	verb = name[REGISTER_COMMAND_PREFIX.size()];
	const ndn::Name::Component& parameterComponent = name[REGISTER_COMMAND_PREFIX.size() + 1];

	ndn::Block rawParameters = parameterComponent.blockFromValue();
	extractedParameters.wireDecode(rawParameters);
	}

	void
	areNamePrefixListsEqual(NamePrefixList& lhs, NamePrefixList& rhs)
	{
	typedef std::list<ndn::Name> NameList;

	NameList& lhsList = lhs.getNameList();
	NameList& rhsList = rhs.getNameList();

	BOOST_REQUIRE_EQUAL(lhsList.size(), rhsList.size());

	NameList::iterator i = lhsList.begin();
	NameList::iterator j = rhsList.begin();

	for (; i != lhsList.end(); ++i, ++j) {
	BOOST_CHECK_EQUAL(i, j);
	}
	}

	public:
	shared_ptr<DummyFace> face;
	Nlsr nlsr;
	SyncLogicHandler sync;

	Lsdb& lsdb;
	ConfParameter& conf;

	ndn::Name REGISTER_COMMAND_PREFIX;
	ndn::Name::Component REGISTER_VERB;
	};

	BOOST_FIXTURE_TEST_SUITE(TestLsdb, LsdbFixture)

	BOOST_AUTO_TEST_CASE(LsdbSync)
	{
	ndn::Name interestName("/ndn/NLSR/LSA/cs/%C1.Router/router2/name");
	uint64_t oldSeqNo = 82;

	ndn::Name oldInterestName = interestName;
	oldInterestName.appendNumber(oldSeqNo);

	lsdb.expressInterest(oldInterestName, 0);
	face->processEvents(ndn::time::milliseconds(1));

	std::vector<ndn::Interest>& interests = face->m_sentInterests;

	BOOST_REQUIRE(interests.size() > 0);
	std::vector<ndn::Interest>::iterator it = interests.begin();

	BOOST_CHECK_EQUAL(it->getName(), oldInterestName);
	interests.clear();

	steady_clock::TimePoint deadline = steady_clock::now() +
	ndn::time::seconds(static_cast<int>(LSA_REFRESH_TIME_MAX));

	// Simulate an LSA interest timeout
	lsdb.processInterestTimedOut(oldInterestName, 0, deadline, interestName, oldSeqNo);
	face->processEvents(ndn::time::milliseconds(1));

	BOOST_REQUIRE(interests.size() > 0);
	it = interests.begin();

	BOOST_CHECK_EQUAL(it->getName(), oldInterestName);
	interests.clear();

	uint64_t newSeqNo = 83;

	ndn::Name newInterestName = interestName;
	newInterestName.appendNumber(newSeqNo);

	lsdb.expressInterest(newInterestName, 0);
	face->processEvents(ndn::time::milliseconds(1));

	BOOST_REQUIRE(interests.size() > 0);
	it = interests.begin();

	BOOST_CHECK_EQUAL(it->getName(), newInterestName);
	interests.clear();

	// Simulate an LSA interest timeout where the sequence number is outdated
	lsdb.processInterestTimedOut(oldInterestName, 0, deadline, interestName, oldSeqNo);
	face->processEvents(ndn::time::milliseconds(1));

	// Interest should not be expressed for outdated sequence number
	BOOST_CHECK_EQUAL(interests.size(), 0);
	}

	BOOST_AUTO_TEST_CASE(LsdbRemoveAndExists)
	{
	ndn::time::system_clock::TimePoint testTimePoint = ndn::time::system_clock::now();
	NamePrefixList npl1;

	std::string s1 = "name1";
	std::string s2 = "name2";
	std::string router1 = "router1/1";

	npl1.insert(s1);
	npl1.insert(s2);

	//For NameLsa lsType is name.
	//12 is seqNo, randomly generated.
	//1800 is the default life time.
	NameLsa nlsa1(ndn::Name("/router1/1"), NameLsa::TYPE_STRING, 12, testTimePoint, npl1);

	Lsdb lsdb1(nlsr, g_scheduler, nlsr.getSyncLogicHandler());

	lsdb1.installNameLsa(nlsa1);
	lsdb1.writeNameLsdbLog();

	BOOST_CHECK(lsdb1.doesLsaExist(ndn::Name("/router1/1/name"), NameLsa::TYPE_STRING));

	lsdb1.removeNameLsa(router1);

	BOOST_CHECK_EQUAL(lsdb1.doesLsaExist(ndn::Name("/router1/1"), NameLsa::TYPE_STRING), false);
	}

	BOOST_AUTO_TEST_CASE(InstallNameLsa)
	{
	// Install lsa with name1 and name2
	ndn::Name name1("/ndn/name1");
	ndn::Name name2("/ndn/name2");

	NamePrefixList prefixes;
	prefixes.insert(name1);
	prefixes.insert(name2);

	std::string otherRouter("/ndn/site/%C1.router/other-router");
	ndn::time::system_clock::TimePoint MAX_TIME = ndn::time::system_clock::TimePoint::max();

	NameLsa lsa(otherRouter, NameLsa::TYPE_STRING, 1, MAX_TIME, prefixes);
	lsdb.installNameLsa(lsa);

	BOOST_REQUIRE_EQUAL(lsdb.doesLsaExist(otherRouter + "/name", NameLsa::TYPE_STRING), true);
	NamePrefixList& nameList = lsdb.findNameLsa(otherRouter + "/name")->getNpl();

	areNamePrefixListsEqual(nameList, prefixes);

	// Add a prefix: name3
	ndn::Name name3("/ndn/name3");
	prefixes.insert(name3);

	NameLsa addLsa(otherRouter, NameLsa::TYPE_STRING, 2, MAX_TIME, prefixes);
	lsdb.installNameLsa(addLsa);

	// Lsa should include name1, name2, and name3
	areNamePrefixListsEqual(nameList, prefixes);

	// Remove a prefix: name2
	prefixes.remove(name2);

	NameLsa removeLsa(otherRouter, NameLsa::TYPE_STRING, 3, MAX_TIME, prefixes);
	lsdb.installNameLsa(removeLsa);

	// Lsa should include name1 and name3
	areNamePrefixListsEqual(nameList, prefixes);

	// Add and remove a prefix: add name2, remove name3
	prefixes.insert(name2);
	prefixes.remove(name3);

	NameLsa addAndRemoveLsa(otherRouter, NameLsa::TYPE_STRING, 4, MAX_TIME, prefixes);
	lsdb.installNameLsa(addAndRemoveLsa);

	// Lsa should include name1 and name2
	areNamePrefixListsEqual(nameList, prefixes);

	// Install a completely new list of prefixes
	ndn::Name name4("/ndn/name4");
	ndn::Name name5("/ndn/name5");

	NamePrefixList newPrefixes;
	newPrefixes.insert(name4);
	newPrefixes.insert(name5);

	NameLsa newLsa(otherRouter, NameLsa::TYPE_STRING, 5, MAX_TIME, newPrefixes);
	lsdb.installNameLsa(newLsa);

	// Lsa should include name4 and name5
	areNamePrefixListsEqual(nameList, newPrefixes);
	}

	BOOST_AUTO_TEST_SUITE_END()

	} //namespace test
	} //namespace nlsr