Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| 2 | /** |
Junxiao Shi | b84e674 | 2016-07-19 13:16:22 +0000 | [diff] [blame] | 3 | * Copyright (c) 2014-2016, Regents of the University of California, |
Spyridon Mastorakis | d0381c0 | 2015-02-19 10:29:41 -0800 | [diff] [blame] | 4 | * Arizona Board of Regents, |
| 5 | * Colorado State University, |
| 6 | * University Pierre & Marie Curie, Sorbonne University, |
| 7 | * Washington University in St. Louis, |
| 8 | * Beijing Institute of Technology, |
| 9 | * The University of Memphis. |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 10 | * |
| 11 | * This file is part of NFD (Named Data Networking Forwarding Daemon). |
| 12 | * See AUTHORS.md for complete list of NFD authors and contributors. |
| 13 | * |
| 14 | * NFD is free software: you can redistribute it and/or modify it under the terms |
| 15 | * of the GNU General Public License as published by the Free Software Foundation, |
| 16 | * either version 3 of the License, or (at your option) any later version. |
| 17 | * |
| 18 | * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; |
| 19 | * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR |
| 20 | * PURPOSE. See the GNU General Public License for more details. |
| 21 | * |
| 22 | * You should have received a copy of the GNU General Public License along with |
| 23 | * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>. |
| 24 | */ |
| 25 | |
| 26 | #include "fw/strategy.hpp" |
| 27 | #include "dummy-strategy.hpp" |
| 28 | #include "tests/daemon/face/dummy-face.hpp" |
| 29 | #include <boost/range/adaptor/filtered.hpp> |
Junxiao Shi | b84e674 | 2016-07-19 13:16:22 +0000 | [diff] [blame] | 30 | #include <boost/range/adaptor/transformed.hpp> |
| 31 | #include <boost/range/algorithm/copy.hpp> |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 32 | |
| 33 | #include "tests/test-common.hpp" |
| 34 | |
| 35 | namespace nfd { |
Spyridon Mastorakis | d0381c0 | 2015-02-19 10:29:41 -0800 | [diff] [blame] | 36 | namespace fw { |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 37 | namespace tests { |
| 38 | |
Spyridon Mastorakis | d0381c0 | 2015-02-19 10:29:41 -0800 | [diff] [blame] | 39 | using namespace nfd::tests; |
| 40 | |
Junxiao Shi | cde37ad | 2015-12-24 01:02:05 -0700 | [diff] [blame] | 41 | BOOST_AUTO_TEST_SUITE(Fw) |
| 42 | BOOST_FIXTURE_TEST_SUITE(TestStrategy, BaseFixture) |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 43 | |
| 44 | class FaceTableAccessTestStrategy : public DummyStrategy |
| 45 | { |
| 46 | public: |
| 47 | explicit |
| 48 | FaceTableAccessTestStrategy(Forwarder& forwarder) |
| 49 | : DummyStrategy(forwarder, Name("ndn:/strategy")) |
| 50 | { |
Junxiao Shi | ae04d34 | 2016-07-19 13:20:22 +0000 | [diff] [blame] | 51 | this->afterAddFace.connect([this] (const Face& face) { |
| 52 | this->addedFaces.push_back(face.getId()); |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 53 | }); |
Junxiao Shi | ae04d34 | 2016-07-19 13:20:22 +0000 | [diff] [blame] | 54 | this->beforeRemoveFace.connect([this] (const Face& face) { |
| 55 | this->removedFaces.push_back(face.getId()); |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 56 | }); |
| 57 | } |
| 58 | |
| 59 | std::vector<FaceId> |
| 60 | getLocalFaces() |
| 61 | { |
| 62 | auto enumerable = this->getFaceTable() | |
Junxiao Shi | b84e674 | 2016-07-19 13:16:22 +0000 | [diff] [blame] | 63 | boost::adaptors::filtered([] (Face& face) { |
| 64 | return face.getScope() == ndn::nfd::FACE_SCOPE_LOCAL; |
| 65 | }) | |
| 66 | boost::adaptors::transformed([] (Face& face) { |
| 67 | return face.getId(); |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 68 | }); |
| 69 | |
| 70 | std::vector<FaceId> results; |
Junxiao Shi | b84e674 | 2016-07-19 13:16:22 +0000 | [diff] [blame] | 71 | boost::copy(enumerable, std::back_inserter(results)); |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 72 | return results; |
| 73 | } |
| 74 | |
| 75 | public: |
| 76 | std::vector<FaceId> addedFaces; |
| 77 | std::vector<FaceId> removedFaces; |
| 78 | }; |
| 79 | |
| 80 | BOOST_AUTO_TEST_CASE(FaceTableAccess) |
| 81 | { |
| 82 | Forwarder forwarder; |
| 83 | FaceTableAccessTestStrategy strategy(forwarder); |
| 84 | |
| 85 | auto face1 = make_shared<DummyFace>(); |
Junxiao Shi | cde37ad | 2015-12-24 01:02:05 -0700 | [diff] [blame] | 86 | auto face2 = make_shared<DummyFace>("dummy://", "dummy://", ndn::nfd::FACE_SCOPE_LOCAL); |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 87 | forwarder.addFace(face1); |
| 88 | forwarder.addFace(face2); |
| 89 | FaceId id1 = face1->getId(); |
| 90 | FaceId id2 = face2->getId(); |
| 91 | |
| 92 | BOOST_CHECK(strategy.getLocalFaces() == std::vector<FaceId>{id2}); |
| 93 | |
| 94 | face2->close(); |
| 95 | face1->close(); |
| 96 | |
| 97 | BOOST_CHECK((strategy.addedFaces == std::vector<FaceId>{id1, id2})); |
| 98 | BOOST_CHECK((strategy.removedFaces == std::vector<FaceId>{id2, id1})); |
| 99 | } |
| 100 | |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 101 | class LookupFibFixture : public BaseFixture |
| 102 | { |
| 103 | protected: |
| 104 | class TestStrategy : public DummyStrategy |
| 105 | { |
| 106 | public: |
| 107 | explicit |
| 108 | TestStrategy(Forwarder& forwarder) |
| 109 | : DummyStrategy(forwarder, Name("ndn:/strategy")) |
| 110 | { |
| 111 | } |
| 112 | |
| 113 | const fib::Entry& |
| 114 | lookupFib(const pit::Entry& pitEntry) const |
| 115 | { |
| 116 | return this->Strategy::lookupFib(pitEntry); |
| 117 | } |
| 118 | }; |
| 119 | |
| 120 | LookupFibFixture() |
| 121 | : strategy(forwarder) |
| 122 | , fib(forwarder.getFib()) |
| 123 | , pit(forwarder.getPit()) |
| 124 | , nrt(forwarder.getNetworkRegionTable()) |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 125 | , inFace(make_shared<DummyFace>()) |
| 126 | , outFace(make_shared<DummyFace>()) |
| 127 | , link(makeLink("/net/ndnsim", {{10, "/telia/terabits"}, {20, "/ucla/cs"}})) |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 128 | { |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 129 | forwarder.addFace(inFace); |
| 130 | forwarder.addFace(outFace); |
| 131 | } |
| 132 | |
| 133 | const fib::Entry& |
| 134 | lookupFib(Interest& interest) |
| 135 | { |
| 136 | shared_ptr<pit::Entry> pitEntry = pit.insert(interest).first; |
| 137 | pitEntry->insertOrUpdateInRecord(*inFace, interest); |
| 138 | return strategy.lookupFib(*pitEntry); |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 139 | } |
| 140 | |
| 141 | protected: |
| 142 | Forwarder forwarder; |
| 143 | TestStrategy strategy; |
| 144 | Fib& fib; |
| 145 | Pit& pit; |
| 146 | NetworkRegionTable& nrt; |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 147 | |
| 148 | shared_ptr<Face> inFace; |
| 149 | shared_ptr<Face> outFace; |
| 150 | shared_ptr<Link> link; |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 151 | }; |
| 152 | |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 153 | BOOST_FIXTURE_TEST_SUITE(LookupFib, LookupFibFixture) |
| 154 | |
| 155 | BOOST_AUTO_TEST_CASE(NoLink) |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 156 | { |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 157 | fib.insert("/net/ndnsim").first->addNextHop(*outFace, 10); |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 158 | |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 159 | auto interest = makeInterest("/net/ndnsim/www/index.html"); |
| 160 | const fib::Entry& fibEntry = this->lookupFib(*interest); |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 161 | |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 162 | BOOST_CHECK_EQUAL(fibEntry.getPrefix(), "/net/ndnsim"); |
| 163 | BOOST_CHECK_EQUAL(interest->hasSelectedDelegation(), false); |
Junxiao Shi | cf0f3ce | 2016-09-02 13:01:59 +0000 | [diff] [blame] | 164 | } |
| 165 | |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 166 | BOOST_AUTO_TEST_CASE(ConsumerRegion) |
| 167 | { |
| 168 | nrt.insert("/arizona/cs/avenir"); |
| 169 | fib.insert("/").first->addNextHop(*outFace, 10); |
| 170 | |
| 171 | auto interest = makeInterest("/net/ndnsim/www/index.html"); |
| 172 | interest->setLink(link->wireEncode()); |
| 173 | const fib::Entry& fibEntry = this->lookupFib(*interest); |
| 174 | |
| 175 | BOOST_CHECK_EQUAL(fibEntry.getPrefix(), "/"); |
| 176 | BOOST_CHECK_EQUAL(interest->hasSelectedDelegation(), false); |
| 177 | } |
| 178 | |
Junxiao Shi | 8527f86 | 2016-09-12 21:32:35 +0000 | [diff] [blame^] | 179 | BOOST_AUTO_TEST_CASE(DefaultFreeFirstDelegation) |
Junxiao Shi | 0dba010 | 2016-09-06 01:57:53 +0000 | [diff] [blame] | 180 | { |
| 181 | nrt.insert("/arizona/cs/hobo"); |
| 182 | fib.insert("/telia").first->addNextHop(*outFace, 20); |
| 183 | fib.insert("/ucla").first->addNextHop(*outFace, 10); |
| 184 | |
| 185 | auto interest = makeInterest("/net/ndnsim/www/index.html"); |
| 186 | interest->setLink(link->wireEncode()); |
| 187 | const fib::Entry& fibEntry = this->lookupFib(*interest); |
| 188 | |
| 189 | BOOST_CHECK_EQUAL(fibEntry.getPrefix(), "/telia"); |
| 190 | BOOST_REQUIRE_EQUAL(interest->hasSelectedDelegation(), true); |
| 191 | BOOST_CHECK_EQUAL(interest->getSelectedDelegation(), "/telia/terabits"); |
| 192 | } |
| 193 | |
| 194 | BOOST_AUTO_TEST_CASE(DefaultFreeSecondDelegation) |
| 195 | { |
| 196 | nrt.insert("/arizona/cs/hobo"); |
| 197 | fib.insert("/ucla").first->addNextHop(*outFace, 10); |
| 198 | |
| 199 | auto interest = makeInterest("/net/ndnsim/www/index.html"); |
| 200 | interest->setLink(link->wireEncode()); |
| 201 | const fib::Entry& fibEntry = this->lookupFib(*interest); |
| 202 | |
| 203 | BOOST_CHECK_EQUAL(fibEntry.getPrefix(), "/ucla"); |
| 204 | BOOST_REQUIRE_EQUAL(interest->hasSelectedDelegation(), true); |
| 205 | BOOST_CHECK_EQUAL(interest->getSelectedDelegation(), "/ucla/cs"); |
| 206 | } |
| 207 | |
| 208 | BOOST_AUTO_TEST_CASE(DefaultFreeHasSelectedDelegation) |
| 209 | { |
| 210 | nrt.insert("/ucsd/caida/click"); |
| 211 | fib.insert("/telia").first->addNextHop(*outFace, 10); |
| 212 | fib.insert("/ucla").first->addNextHop(*outFace, 10); |
| 213 | |
| 214 | auto interest = makeInterest("/net/ndnsim/www/index.html"); |
| 215 | interest->setLink(link->wireEncode()); |
| 216 | interest->setSelectedDelegation("/ucla/cs"); |
| 217 | const fib::Entry& fibEntry = this->lookupFib(*interest); |
| 218 | |
| 219 | BOOST_CHECK_EQUAL(fibEntry.getPrefix(), "/ucla"); |
| 220 | BOOST_REQUIRE_EQUAL(interest->hasSelectedDelegation(), true); |
| 221 | BOOST_CHECK_EQUAL(interest->getSelectedDelegation(), "/ucla/cs"); |
| 222 | } |
| 223 | |
| 224 | BOOST_AUTO_TEST_CASE(ProducerRegion) |
| 225 | { |
| 226 | nrt.insert("/ucla/cs/spurs"); |
| 227 | fib.insert("/").first->addNextHop(*outFace, 10); |
| 228 | fib.insert("/ucla").first->addNextHop(*outFace, 10); |
| 229 | fib.insert("/net/ndnsim").first->addNextHop(*outFace, 10); |
| 230 | |
| 231 | auto interest = makeInterest("/net/ndnsim/www/index.html"); |
| 232 | interest->setLink(link->wireEncode()); |
| 233 | interest->setSelectedDelegation("/ucla/cs"); |
| 234 | const fib::Entry& fibEntry = this->lookupFib(*interest); |
| 235 | |
| 236 | BOOST_CHECK_EQUAL(fibEntry.getPrefix(), "/net/ndnsim"); |
| 237 | BOOST_REQUIRE_EQUAL(interest->hasSelectedDelegation(), true); |
| 238 | BOOST_CHECK_EQUAL(interest->getSelectedDelegation(), "/ucla/cs"); |
| 239 | } |
| 240 | |
| 241 | BOOST_AUTO_TEST_SUITE_END() // LookupFib |
| 242 | |
Junxiao Shi | cde37ad | 2015-12-24 01:02:05 -0700 | [diff] [blame] | 243 | BOOST_AUTO_TEST_SUITE_END() // TestStrategy |
| 244 | BOOST_AUTO_TEST_SUITE_END() // Fw |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 245 | |
| 246 | } // namespace tests |
Spyridon Mastorakis | d0381c0 | 2015-02-19 10:29:41 -0800 | [diff] [blame] | 247 | } // namespace fw |
Junxiao Shi | 49e11e7 | 2014-12-14 19:46:05 -0700 | [diff] [blame] | 248 | } // namespace nfd |