| /* -*- 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 "fw/forwarder.hpp" |
| #include "../face/dummy-face.hpp" |
| |
| #include <boost/test/unit_test.hpp> |
| |
| namespace nfd { |
| |
| class ForwarderTestFace : public Face { |
| public: |
| explicit |
| ForwarderTestFace(boost::asio::io_service& ioService) |
| : m_ioService(ioService) |
| { |
| } |
| |
| virtual void |
| sendInterest(const Interest& interest) |
| { |
| m_sentInterests.push_back(interest); |
| m_ioService.stop(); |
| } |
| |
| virtual void |
| sendData(const Data& data) |
| { |
| m_sentDatas.push_back(data); |
| m_ioService.stop(); |
| } |
| |
| virtual void |
| close() |
| { |
| } |
| |
| void |
| receiveInterest(const Interest& interest) |
| { |
| onReceiveInterest(interest); |
| } |
| |
| void |
| receiveData(const Data& data) |
| { |
| onReceiveData(data); |
| } |
| |
| public: |
| std::vector<Interest> m_sentInterests; |
| std::vector<Data> m_sentDatas; |
| |
| private: |
| boost::asio::io_service& m_ioService; |
| }; |
| |
| BOOST_AUTO_TEST_SUITE(FwForwarder) |
| |
| BOOST_AUTO_TEST_CASE(AddRemoveFace) |
| { |
| boost::asio::io_service io; |
| Forwarder forwarder(io); |
| |
| shared_ptr<Face> face1 = make_shared<DummyFace>(); |
| shared_ptr<Face> face2 = make_shared<DummyFace>(); |
| |
| BOOST_CHECK_EQUAL(face1->getId(), INVALID_FACEID); |
| BOOST_CHECK_EQUAL(face2->getId(), INVALID_FACEID); |
| |
| forwarder.addFace(face1); |
| forwarder.addFace(face2); |
| |
| BOOST_CHECK_NE(face1->getId(), INVALID_FACEID); |
| BOOST_CHECK_NE(face2->getId(), INVALID_FACEID); |
| BOOST_CHECK_NE(face1->getId(), face2->getId()); |
| |
| forwarder.removeFace(face1); |
| forwarder.removeFace(face2); |
| |
| BOOST_CHECK_EQUAL(face1->getId(), INVALID_FACEID); |
| BOOST_CHECK_EQUAL(face2->getId(), INVALID_FACEID); |
| } |
| |
| BOOST_AUTO_TEST_CASE(SimpleExchange) |
| { |
| Name nameA ("ndn:/A"); |
| Name nameAB ("ndn:/A/B"); |
| Name nameABC("ndn:/A/B/C"); |
| Interest interestAB(nameAB); |
| interestAB.setInterestLifetime(4000); |
| Data dataABC(nameABC); |
| |
| boost::asio::io_service io; |
| Forwarder forwarder(io); |
| |
| shared_ptr<ForwarderTestFace> face1 = make_shared<ForwarderTestFace>(boost::ref(io)); |
| shared_ptr<ForwarderTestFace> face2 = make_shared<ForwarderTestFace>(boost::ref(io)); |
| forwarder.addFace(face1); |
| forwarder.addFace(face2); |
| |
| Fib& fib = forwarder.getFib(); |
| std::pair<shared_ptr<fib::Entry>, bool> fibInsertResult = |
| fib.insert(Name("ndn:/A")); |
| shared_ptr<fib::Entry> fibEntry = fibInsertResult.first; |
| fibEntry->addNextHop(face2, 0); |
| |
| face1->receiveInterest(interestAB); |
| io.run(); |
| io.reset(); |
| BOOST_REQUIRE_EQUAL(face2->m_sentInterests.size(), 1); |
| BOOST_CHECK(face2->m_sentInterests[0].getName().equals(nameAB)); |
| BOOST_CHECK_EQUAL(face2->m_sentInterests[0].getIncomingFaceId(), face1->getId()); |
| |
| face2->receiveData(dataABC); |
| io.run(); |
| io.reset(); |
| BOOST_REQUIRE_EQUAL(face1->m_sentDatas.size(), 1); |
| BOOST_CHECK(face1->m_sentDatas[0].getName().equals(nameABC)); |
| BOOST_CHECK_EQUAL(face1->m_sentDatas[0].getIncomingFaceId(), face2->getId()); |
| } |
| |
| class ScopeLocalhostTestForwarder : public Forwarder |
| { |
| public: |
| explicit |
| ScopeLocalhostTestForwarder(boost::asio::io_service& ioService) |
| : Forwarder(ioService) |
| { |
| } |
| |
| virtual void |
| onDataUnsolicited(Face& inFace, const Data& data) |
| { |
| ++m_onDataUnsolicited_count; |
| } |
| |
| protected: |
| virtual void |
| dispatchToStrategy(const Face& inFace, |
| const Interest& interest, |
| shared_ptr<fib::Entry> fibEntry, |
| shared_ptr<pit::Entry> pitEntry) |
| { |
| ++m_dispatchToStrategy_count; |
| } |
| |
| public: |
| int m_dispatchToStrategy_count; |
| int m_onDataUnsolicited_count; |
| }; |
| |
| BOOST_AUTO_TEST_CASE(ScopeLocalhost) |
| { |
| boost::asio::io_service io; |
| ScopeLocalhostTestForwarder forwarder(io); |
| shared_ptr<DummyLocalFace> face1 = make_shared<DummyLocalFace>(); |
| shared_ptr<DummyFace> face2 = make_shared<DummyFace>(); |
| forwarder.addFace(face1); |
| forwarder.addFace(face2); |
| |
| // local face, /localhost: OK |
| forwarder.m_dispatchToStrategy_count = 0; |
| forwarder.onIncomingInterest(*face1, Interest(Name("/localhost/A1"))); |
| BOOST_CHECK_EQUAL(forwarder.m_dispatchToStrategy_count, 1); |
| |
| // non-local face, /localhost: violate |
| forwarder.m_dispatchToStrategy_count = 0; |
| forwarder.onIncomingInterest(*face2, Interest(Name("/localhost/A2"))); |
| BOOST_CHECK_EQUAL(forwarder.m_dispatchToStrategy_count, 0); |
| |
| // local face, non-/localhost: OK |
| forwarder.m_dispatchToStrategy_count = 0; |
| forwarder.onIncomingInterest(*face1, Interest(Name("/A3"))); |
| BOOST_CHECK_EQUAL(forwarder.m_dispatchToStrategy_count, 1); |
| |
| // non-local face, non-/localhost: OK |
| forwarder.m_dispatchToStrategy_count = 0; |
| forwarder.onIncomingInterest(*face2, Interest(Name("/A4"))); |
| BOOST_CHECK_EQUAL(forwarder.m_dispatchToStrategy_count, 1); |
| |
| // local face, /localhost: OK |
| forwarder.m_onDataUnsolicited_count = 0; |
| forwarder.onIncomingData(*face1, Data(Name("/localhost/B1"))); |
| BOOST_CHECK_EQUAL(forwarder.m_onDataUnsolicited_count, 1); |
| |
| // non-local face, /localhost: OK |
| forwarder.m_onDataUnsolicited_count = 0; |
| forwarder.onIncomingData(*face2, Data(Name("/localhost/B2"))); |
| BOOST_CHECK_EQUAL(forwarder.m_onDataUnsolicited_count, 0); |
| |
| // local face, non-/localhost: OK |
| forwarder.m_onDataUnsolicited_count = 0; |
| forwarder.onIncomingData(*face1, Data(Name("/B3"))); |
| BOOST_CHECK_EQUAL(forwarder.m_onDataUnsolicited_count, 1); |
| |
| // non-local face, non-/localhost: OK |
| forwarder.m_onDataUnsolicited_count = 0; |
| forwarder.onIncomingData(*face2, Data(Name("/B4"))); |
| BOOST_CHECK_EQUAL(forwarder.m_onDataUnsolicited_count, 1); |
| } |
| |
| BOOST_AUTO_TEST_SUITE_END() |
| |
| } // namespace nfd |