blob: 2fce7f1ea59d602014b1f406dc1e9cb3d137b71a [file] [log] [blame]
/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
/**
* Copyright (C) 2014 Named Data Networking Project
* See COPYING for copyright and distribution information.
*/
#include "mgmt/internal-face.hpp"
#include "mgmt/fib-manager.hpp"
#include "table/fib.hpp"
#include "../face/dummy-face.hpp"
#include <ndn-cpp-dev/management/fib-management-options.hpp>
#include <ndn-cpp-dev/management/control-response.hpp>
#include <ndn-cpp-dev/encoding/block.hpp>
#include <boost/test/unit_test.hpp>
namespace nfd {
NFD_LOG_INIT("InternalFaceTest");
class InternalFaceFixture
{
public:
shared_ptr<Face>
getFace(FaceId id)
{
if (m_faces.size() < id)
{
BOOST_FAIL("Attempted to access invalid FaceId: " << id);
}
return m_faces[id-1];
}
void
addFace(shared_ptr<Face> face)
{
m_faces.push_back(face);
}
private:
std::vector<shared_ptr<Face> > m_faces;
};
BOOST_AUTO_TEST_SUITE(MgmtInternalFace)
void
validateControlResponse(const Data& response,
uint32_t expectedCode,
const std::string& expectedText)
{
Block controlRaw = response.getContent().blockFromValue();
ndn::ControlResponse control;
control.wireDecode(controlRaw);
NFD_LOG_DEBUG("received control response"
<< " Name: " << response.getName()
<< " code: " << control.getCode()
<< " text: " << control.getText());
BOOST_REQUIRE(control.getCode() == expectedCode);
BOOST_REQUIRE(control.getText() == expectedText);
}
BOOST_AUTO_TEST_CASE(ValidAddNextHop)
{
InternalFaceFixture fixture;
fixture.addFace(make_shared<DummyFace>());
shared_ptr<InternalFace> face(new InternalFace);
Fib fib;
FibManager manager(fib,
bind(&InternalFaceFixture::getFace,
&fixture, _1),
face);
face->setInterestFilter("/localhost/nfd/fib",
bind(&FibManager::onFibRequest,
&manager, _2));
face->onReceiveData +=
bind(&validateControlResponse, _1, 200, "OK");
ndn::FibManagementOptions options;
options.setName("/hello");
options.setFaceId(1);
options.setCost(1);
Block encodedOptions(options.wireEncode());
Name commandName("/localhost/nfd/fib");
commandName.append("add-nexthop");
commandName.append(encodedOptions);
Interest command(commandName);
face->sendInterest(command);
shared_ptr<fib::Entry> entry = fib.findLongestPrefixMatch("/hello");
if (entry)
{
const fib::NextHopList& hops = entry->getNextHops();
BOOST_REQUIRE(hops.size() == 1);
BOOST_CHECK(hops[0].getCost() == 1);
}
else
{
BOOST_FAIL("Failed to find expected fib entry");
}
}
BOOST_AUTO_TEST_CASE(InvalidPrefixRegistration)
{
InternalFaceFixture fixture;
fixture.addFace(make_shared<DummyFace>());
shared_ptr<InternalFace> face(new InternalFace);
Fib fib;
FibManager manager(fib,
bind(&InternalFaceFixture::getFace,
&fixture, _1),
face);
face->setInterestFilter("/localhost/nfd/fib",
bind(&FibManager::onFibRequest,
&manager, _2));
face->onReceiveData +=
bind(&validateControlResponse, _1, 404, "MALFORMED");
Interest nonRegInterest("/hello");
face->sendInterest(nonRegInterest);
shared_ptr<fib::Entry> entry = fib.findLongestPrefixMatch("/hello");
if (entry)
{
const fib::NextHopList& hops = entry->getNextHops();
BOOST_REQUIRE(hops.size() == 0);
}
}
void
validateOnInterestCallback(const Name& name, const Interest& interest)
{
NFD_LOG_DEBUG("Reached correct callback");
}
void
validateNoOnInterestCallback(const Name& name, const Interest& interest)
{
BOOST_FAIL("Reached wrong callback");
}
BOOST_AUTO_TEST_CASE(SendInterestHitEnd)
{
InternalFaceFixture fixture;
fixture.addFace(make_shared<DummyFace>());
shared_ptr<InternalFace> face(new InternalFace);
Fib fib;
FibManager manager(fib,
bind(&InternalFaceFixture::getFace,
&fixture, _1),
face);
face->setInterestFilter("/localhost/nfd/fib",
&validateOnInterestCallback);
// generate command whose name is canonically
// ordered after /localhost/nfd/fib so that
// we hit the end of the std::map
Name commandName("/localhost/nfd/fib/end");
Interest command(commandName);
face->sendInterest(command);
}
BOOST_AUTO_TEST_CASE(SendInterestHitBegin)
{
InternalFaceFixture fixture;
fixture.addFace(make_shared<DummyFace>());
shared_ptr<InternalFace> face(new InternalFace);
Fib fib;
FibManager manager(fib,
bind(&InternalFaceFixture::getFace,
&fixture, _1),
face);
face->setInterestFilter("/localhost/nfd/fib",
&validateNoOnInterestCallback);
// generate command whose name is canonically
// ordered before /localhost/nfd/fib so that
// we hit the beginning of the std::map
Name commandName("/localhost/nfd");
Interest command(commandName);
face->sendInterest(command);
}
BOOST_AUTO_TEST_CASE(SendInterestHitExact)
{
InternalFaceFixture fixture;
fixture.addFace(make_shared<DummyFace>());
shared_ptr<InternalFace> face(new InternalFace);
Fib fib;
FibManager manager(fib,
bind(&InternalFaceFixture::getFace,
&fixture, _1),
face);
face->setInterestFilter("/localhost/nfd/eib",
&validateNoOnInterestCallback);
face->setInterestFilter("/localhost/nfd/fib",
&validateOnInterestCallback);
face->setInterestFilter("/localhost/nfd/gib",
&validateNoOnInterestCallback);
// generate command whose name exactly matches
// /localhost/nfd/fib
Name commandName("/localhost/nfd/fib");
Interest command(commandName);
face->sendInterest(command);
}
BOOST_AUTO_TEST_CASE(SendInterestHitPrevious)
{
InternalFaceFixture fixture;
fixture.addFace(make_shared<DummyFace>());
shared_ptr<InternalFace> face(new InternalFace);
Fib fib;
FibManager manager(fib,
bind(&InternalFaceFixture::getFace,
&fixture, _1),
face);
face->setInterestFilter("/localhost/nfd/fib",
&validateOnInterestCallback);
face->setInterestFilter("/localhost/nfd/fib/zzzzzzzzzzzzz/",
&validateNoOnInterestCallback);
// generate command whose name exactly matches
// an Interest filter
Name commandName("/localhost/nfd/fib/previous");
Interest command(commandName);
face->sendInterest(command);
}
BOOST_AUTO_TEST_SUITE_END()
} // namespace nfd