Ashlesh Gawande | 32ec3fd | 2018-07-18 13:42:32 -0500 | [diff] [blame^] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| 2 | /** |
| 3 | * Copyright (c) 2014-2018, The University of Memphis, |
| 4 | * Regents of the University of California, |
| 5 | * Arizona Board of Regents. |
| 6 | * |
| 7 | * This file is part of NLSR (Named-data Link State Routing). |
| 8 | * See AUTHORS.md for complete list of NLSR authors and contributors. |
| 9 | * |
| 10 | * NLSR is free software: you can redistribute it and/or modify it under the terms |
| 11 | * of the GNU General Public License as published by the Free Software Foundation, |
| 12 | * either version 3 of the License, or (at your option) any later version. |
| 13 | * |
| 14 | * NLSR is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; |
| 15 | * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR |
| 16 | * PURPOSE. See the GNU General Public License for more details. |
| 17 | * |
| 18 | * You should have received a copy of the GNU General Public License along with |
| 19 | * NLSR, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>. |
| 20 | **/ |
| 21 | |
| 22 | #include "communication/sync-protocol-adapter.hpp" |
| 23 | #include "test-common.hpp" |
| 24 | #include "boost-test.hpp" |
| 25 | |
| 26 | #include <ndn-cxx/util/dummy-client-face.hpp> |
| 27 | |
| 28 | #include <boost/mpl/int.hpp> |
| 29 | #include <boost/mpl/vector.hpp> |
| 30 | |
| 31 | namespace nlsr { |
| 32 | namespace test { |
| 33 | |
| 34 | using namespace ndn; |
| 35 | |
| 36 | class SyncProtocolAdapterFixture : public UnitTestTimeFixture |
| 37 | { |
| 38 | public: |
| 39 | SyncProtocolAdapterFixture() |
| 40 | : syncPrefix("/localhop/ndn/nlsr/sync/") |
| 41 | , nameLsaUserPrefix("/localhop/ndn/nlsr/LSA/NAME") |
| 42 | , syncInterestLifetime(time::seconds(60)) |
| 43 | { |
| 44 | syncPrefix.appendVersion(4); |
| 45 | } |
| 46 | |
| 47 | template <int32_t T> |
| 48 | void |
| 49 | addNodes() |
| 50 | { |
| 51 | for (int i = 0; i < 2; i++) { |
| 52 | faces[i] = std::make_shared<ndn::util::DummyClientFace>(m_ioService, |
| 53 | util::DummyClientFace::Options{true, true}); |
| 54 | userPrefixes[i] = Name(nameLsaUserPrefix).appendNumber(i); |
| 55 | nodes[i] = std::make_shared<SyncProtocolAdapter>(*faces[i], T, syncPrefix, |
| 56 | userPrefixes[i], |
| 57 | syncInterestLifetime, |
| 58 | [i, this] (const ndn::Name& updateName, |
| 59 | uint64_t highSeq) { |
| 60 | prefixToSeq[i].emplace(updateName, highSeq); |
| 61 | }); |
| 62 | } |
| 63 | |
| 64 | faces[0]->linkTo(*faces[1]); |
| 65 | advanceClocks(ndn::time::milliseconds(10), 10); |
| 66 | } |
| 67 | |
| 68 | public: |
| 69 | Name syncPrefix, nameLsaUserPrefix; |
| 70 | Name userPrefixes[2]; |
| 71 | time::milliseconds syncInterestLifetime; |
| 72 | std::shared_ptr<ndn::util::DummyClientFace> faces[2]; |
| 73 | std::shared_ptr<SyncProtocolAdapter> nodes[2]; |
| 74 | std::map<ndn::Name, uint64_t> prefixToSeq[2]; |
| 75 | }; |
| 76 | |
| 77 | using boost::mpl::int_; |
| 78 | using Protocols = boost::mpl::vector<int_<SYNC_PROTOCOL_CHRONOSYNC>, int_<SYNC_PROTOCOL_PSYNC>>; |
| 79 | |
| 80 | BOOST_FIXTURE_TEST_SUITE(TestSyncProtocolAdapter, SyncProtocolAdapterFixture) |
| 81 | |
| 82 | BOOST_AUTO_TEST_CASE_TEMPLATE(Sync, SyncProtocol, Protocols) |
| 83 | { |
| 84 | addNodes<SyncProtocol::value>(); |
| 85 | |
| 86 | nodes[0]->publishUpdate(userPrefixes[0], 10); |
| 87 | advanceClocks(ndn::time::milliseconds(1000), 100); |
| 88 | |
| 89 | auto it = prefixToSeq[1].find(userPrefixes[0]); |
| 90 | BOOST_CHECK(it != prefixToSeq[1].end()); |
| 91 | BOOST_CHECK_EQUAL(it->first, userPrefixes[0]); |
| 92 | BOOST_CHECK_EQUAL(it->second, 10); |
| 93 | |
| 94 | nodes[1]->publishUpdate(userPrefixes[1], 100); |
| 95 | advanceClocks(ndn::time::milliseconds(1000), 100); |
| 96 | |
| 97 | it = prefixToSeq[0].find(userPrefixes[1]); |
| 98 | BOOST_CHECK(it != prefixToSeq[0].end()); |
| 99 | BOOST_CHECK_EQUAL(it->first, userPrefixes[1]); |
| 100 | BOOST_CHECK_EQUAL(it->second, 100); |
| 101 | |
| 102 | Name adjLsaUserPrefix("/localhop/ndn/nlsr/LSA/ADJACENCY"); |
| 103 | nodes[0]->addUserNode(adjLsaUserPrefix); |
| 104 | advanceClocks(ndn::time::milliseconds(1000), 100); |
| 105 | nodes[0]->publishUpdate(adjLsaUserPrefix, 10); |
| 106 | advanceClocks(ndn::time::milliseconds(1000), 100); |
| 107 | |
| 108 | it = prefixToSeq[1].find(adjLsaUserPrefix); |
| 109 | BOOST_CHECK(it != prefixToSeq[1].end()); |
| 110 | BOOST_CHECK_EQUAL(it->first, adjLsaUserPrefix); |
| 111 | BOOST_CHECK_EQUAL(it->second, 10); |
| 112 | } |
| 113 | |
| 114 | BOOST_AUTO_TEST_SUITE_END() |
| 115 | |
| 116 | } // namespace test |
| 117 | } // namespace nlsr |