blob: 4c030ace24f36d55b7805c583e0806f96102fc95 [file] [log] [blame]
HYuana9b85752014-02-26 02:32:30 -06001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Junxiao Shia6de4292016-07-12 02:08:10 +00003 * Copyright (c) 2014-2016, Regents of the University of California,
Spyridon Mastorakisd0381c02015-02-19 10:29:41 -08004 * 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.
Alexander Afanasyev9bcbc7c2014-04-06 19:37:37 -070010 *
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/>.
Alexander Afanasyev28d586a2014-07-10 20:10:54 -070024 */
HYuana9b85752014-02-26 02:32:30 -060025
26#include "table/name-tree.hpp"
Junxiao Shi60607c72014-11-26 22:40:36 -070027
Junxiao Shid9ee45c2014-02-27 15:38:11 -070028#include "tests/test-common.hpp"
HYuana9b85752014-02-26 02:32:30 -060029
30namespace nfd {
Junxiao Shi2570f3e2016-07-27 02:48:29 +000031namespace name_tree {
Junxiao Shid9ee45c2014-02-27 15:38:11 -070032namespace tests {
HYuana9b85752014-02-26 02:32:30 -060033
Junxiao Shi2570f3e2016-07-27 02:48:29 +000034using namespace nfd::tests;
HYuana9b85752014-02-26 02:32:30 -060035
Junxiao Shi2570f3e2016-07-27 02:48:29 +000036BOOST_AUTO_TEST_SUITE(Table)
37BOOST_FIXTURE_TEST_SUITE(TestNameTree, BaseFixture)
HYuana9b85752014-02-26 02:32:30 -060038
Junxiao Shib660b4c2016-08-06 20:47:44 +000039BOOST_AUTO_TEST_CASE(ComputeHash)
Haowei Yuanf52dac72014-03-24 23:35:03 -050040{
41 Name root("/");
42 root.wireEncode();
Junxiao Shib660b4c2016-08-06 20:47:44 +000043 HashValue hashValue = computeHash(root);
Junxiao Shi2570f3e2016-07-27 02:48:29 +000044 BOOST_CHECK_EQUAL(hashValue, 0);
Haowei Yuanf52dac72014-03-24 23:35:03 -050045
46 Name prefix("/nohello/world/ndn/research");
47 prefix.wireEncode();
Junxiao Shib660b4c2016-08-06 20:47:44 +000048 HashSequence hashes = computeHashes(prefix);
49 BOOST_CHECK_EQUAL(hashes.size(), prefix.size() + 1);
Haowei Yuanf52dac72014-03-24 23:35:03 -050050}
51
Junxiao Shib660b4c2016-08-06 20:47:44 +000052BOOST_AUTO_TEST_SUITE(Hashtable)
53using name_tree::Hashtable;
54
55BOOST_AUTO_TEST_CASE(Modifiers)
56{
57 Hashtable ht(HashtableOptions(16));
58
59 Name name("/A/B/C/D");
60 HashSequence hashes = computeHashes(name);
61
62 BOOST_CHECK_EQUAL(ht.size(), 0);
63 BOOST_CHECK(ht.find(name, 2) == nullptr);
64
65 const Node* node = nullptr;
66 bool isNew = false;
67 std::tie(node, isNew) = ht.insert(name, 2, hashes);
68 BOOST_CHECK_EQUAL(isNew, true);
69 BOOST_CHECK(node != nullptr);
70 BOOST_CHECK_EQUAL(ht.size(), 1);
71 BOOST_CHECK_EQUAL(ht.find(name, 2), node);
72 BOOST_CHECK_EQUAL(ht.find(name, 2, hashes), node);
73
74 BOOST_CHECK(ht.find(name, 0) == nullptr);
75 BOOST_CHECK(ht.find(name, 1) == nullptr);
76 BOOST_CHECK(ht.find(name, 3) == nullptr);
77 BOOST_CHECK(ht.find(name, 4) == nullptr);
78
79 const Node* node2 = nullptr;
80 std::tie(node2, isNew) = ht.insert(name, 2, hashes);
81 BOOST_CHECK_EQUAL(isNew, false);
82 BOOST_CHECK_EQUAL(node2, node);
83 BOOST_CHECK_EQUAL(ht.size(), 1);
84
85 std::tie(node2, isNew) = ht.insert(name, 4, hashes);
86 BOOST_CHECK_EQUAL(isNew, true);
87 BOOST_CHECK(node2 != nullptr);
88 BOOST_CHECK_NE(node2, node);
89 BOOST_CHECK_EQUAL(ht.size(), 2);
90
91 ht.erase(const_cast<Node*>(node2));
92 BOOST_CHECK_EQUAL(ht.size(), 1);
93 BOOST_CHECK(ht.find(name, 4) == nullptr);
94 BOOST_CHECK_EQUAL(ht.find(name, 2), node);
95
96 ht.erase(const_cast<Node*>(node));
97 BOOST_CHECK_EQUAL(ht.size(), 0);
98 BOOST_CHECK(ht.find(name, 2) == nullptr);
99 BOOST_CHECK(ht.find(name, 4) == nullptr);
100}
101
102BOOST_AUTO_TEST_CASE(Resize)
103{
104 HashtableOptions options(9);
105 BOOST_CHECK_EQUAL(options.initialSize, 9);
106 BOOST_CHECK_EQUAL(options.minSize, 9);
107 options.minSize = 6;
108 options.expandLoadFactor = 0.80;
109 options.expandFactor = 5.0;
110 options.shrinkLoadFactor = 0.12;
111 options.shrinkFactor = 0.3;
112
113 Hashtable ht(options);
114
115 auto addNodes = [&ht] (int min, int max) {
116 for (int i = min; i <= max; ++i) {
117 Name name;
118 name.appendNumber(i);
119 HashSequence hashes = computeHashes(name);
120 ht.insert(name, name.size(), hashes);
121 }
122 };
123
124 auto removeNodes = [&ht] (int min, int max) {
125 for (int i = min; i <= max; ++i) {
126 Name name;
127 name.appendNumber(i);
128 const Node* node = ht.find(name, name.size());
129 BOOST_REQUIRE(node != nullptr);
130 ht.erase(const_cast<Node*>(node));
131 }
132 };
133
134 BOOST_CHECK_EQUAL(ht.size(), 0);
135 BOOST_CHECK_EQUAL(ht.getNBuckets(), 9);
136
137 addNodes(1, 1);
138 BOOST_CHECK_EQUAL(ht.size(), 1);
139 BOOST_CHECK_EQUAL(ht.getNBuckets(), 9);
140
141 removeNodes(1, 1);
142 BOOST_CHECK_EQUAL(ht.size(), 0);
143 BOOST_CHECK_EQUAL(ht.getNBuckets(), 6);
144
145 addNodes(1, 4);
146 BOOST_CHECK_EQUAL(ht.size(), 4);
147 BOOST_CHECK_EQUAL(ht.getNBuckets(), 6);
148
149 addNodes(5, 5);
150 BOOST_CHECK_EQUAL(ht.size(), 5);
151 BOOST_CHECK_EQUAL(ht.getNBuckets(), 30);
152
153 addNodes(6, 23);
154 BOOST_CHECK_EQUAL(ht.size(), 23);
155 BOOST_CHECK_EQUAL(ht.getNBuckets(), 30);
156
157 addNodes(24, 25);
158 BOOST_CHECK_EQUAL(ht.size(), 25);
159 BOOST_CHECK_EQUAL(ht.getNBuckets(), 150);
160
161 removeNodes(19, 25);
162 BOOST_CHECK_EQUAL(ht.size(), 18);
163 BOOST_CHECK_EQUAL(ht.getNBuckets(), 150);
164
165 removeNodes(17, 18);
166 BOOST_CHECK_EQUAL(ht.size(), 16);
167 BOOST_CHECK_EQUAL(ht.getNBuckets(), 45);
168
169 removeNodes(7, 16);
170 BOOST_CHECK_EQUAL(ht.size(), 6);
171 BOOST_CHECK_EQUAL(ht.getNBuckets(), 45);
172
173 removeNodes(5, 6);
174 BOOST_CHECK_EQUAL(ht.size(), 4);
175 BOOST_CHECK_EQUAL(ht.getNBuckets(), 13);
176
177 removeNodes(1, 4);
178 BOOST_CHECK_EQUAL(ht.size(), 0);
179 BOOST_CHECK_EQUAL(ht.getNBuckets(), 6);
180}
181
182BOOST_AUTO_TEST_SUITE_END() // Hashtable
183
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000184BOOST_AUTO_TEST_CASE(NameTreeEntry)
HYuana9b85752014-02-26 02:32:30 -0600185{
186 Name prefix("ndn:/named-data/research/abc/def/ghi");
187
Junxiao Shib660b4c2016-08-06 20:47:44 +0000188 auto node = make_unique<Node>(0, prefix);
189 shared_ptr<Entry> npe = node->entry;
Junxiao Shie3cf2852016-08-09 03:50:56 +0000190 BOOST_CHECK_EQUAL(npe->getName(), prefix);
HYuana9b85752014-02-26 02:32:30 -0600191
192 // examine all the get methods
193
Junxiao Shib660b4c2016-08-06 20:47:44 +0000194 Entry* parent = npe->getParent();
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000195 BOOST_CHECK(parent == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600196
Junxiao Shib660b4c2016-08-06 20:47:44 +0000197 const std::vector<Entry*>& children = npe->getChildren();
198 BOOST_CHECK_EQUAL(children.size(), 0);
HYuana9b85752014-02-26 02:32:30 -0600199
Junxiao Shia6de4292016-07-12 02:08:10 +0000200 fib::Entry* fib = npe->getFibEntry();
201 BOOST_CHECK(fib == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600202
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000203 const std::vector<shared_ptr<pit::Entry>>& pitList = npe->getPitEntries();
204 BOOST_CHECK_EQUAL(pitList.size(), 0);
HYuana9b85752014-02-26 02:32:30 -0600205
Haowei Yuane1079fc2014-03-08 14:41:25 -0600206 // examine all the set method
HYuana9b85752014-02-26 02:32:30 -0600207
HYuana9b85752014-02-26 02:32:30 -0600208 Name parentName("ndn:/named-data/research/abc/def");
Junxiao Shib660b4c2016-08-06 20:47:44 +0000209 auto parentNode = make_unique<Node>(1, parentName);
210 npe->setParent(*parentNode->entry);
211 BOOST_CHECK_EQUAL(npe->getParent(), parentNode->entry.get());
HYuana9b85752014-02-26 02:32:30 -0600212
Haowei Yuane1079fc2014-03-08 14:41:25 -0600213 // Insert FIB
HYuana9b85752014-02-26 02:32:30 -0600214
Junxiao Shia6de4292016-07-12 02:08:10 +0000215 npe->setFibEntry(make_unique<fib::Entry>(prefix));
216 BOOST_REQUIRE(npe->getFibEntry() != nullptr);
217 BOOST_CHECK_EQUAL(npe->getFibEntry()->getPrefix(), prefix);
Haowei Yuane1079fc2014-03-08 14:41:25 -0600218
Junxiao Shia6de4292016-07-12 02:08:10 +0000219 npe->setFibEntry(nullptr);
220 BOOST_CHECK(npe->getFibEntry() == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600221
222 // Insert a PIT
223
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000224 auto pitEntry = make_shared<pit::Entry>(*makeInterest(prefix));
225 auto pitEntry2 = make_shared<pit::Entry>(*makeInterest(parentName));
HYuana9b85752014-02-26 02:32:30 -0600226
Alexander Afanasyev28d586a2014-07-10 20:10:54 -0700227 npe->insertPitEntry(pitEntry);
Junxiao Shiefceadc2014-03-09 18:52:57 -0700228 BOOST_CHECK_EQUAL(npe->getPitEntries().size(), 1);
HYuana9b85752014-02-26 02:32:30 -0600229
Alexander Afanasyev28d586a2014-07-10 20:10:54 -0700230 npe->insertPitEntry(pitEntry2);
Junxiao Shiefceadc2014-03-09 18:52:57 -0700231 BOOST_CHECK_EQUAL(npe->getPitEntries().size(), 2);
HYuana9b85752014-02-26 02:32:30 -0600232
Alexander Afanasyev28d586a2014-07-10 20:10:54 -0700233 npe->erasePitEntry(pitEntry);
Junxiao Shiefceadc2014-03-09 18:52:57 -0700234 BOOST_CHECK_EQUAL(npe->getPitEntries().size(), 1);
HYuana9b85752014-02-26 02:32:30 -0600235
Alexander Afanasyev28d586a2014-07-10 20:10:54 -0700236 npe->erasePitEntry(pitEntry2);
Junxiao Shiefceadc2014-03-09 18:52:57 -0700237 BOOST_CHECK_EQUAL(npe->getPitEntries().size(), 0);
HYuana9b85752014-02-26 02:32:30 -0600238}
239
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700240BOOST_AUTO_TEST_CASE(Basic)
HYuana9b85752014-02-26 02:32:30 -0600241{
242 size_t nBuckets = 16;
243 NameTree nt(nBuckets);
244
245 BOOST_CHECK_EQUAL(nt.size(), 0);
Haowei Yuane1079fc2014-03-08 14:41:25 -0600246 BOOST_CHECK_EQUAL(nt.getNBuckets(), nBuckets);
HYuana9b85752014-02-26 02:32:30 -0600247
Haowei Yuane1079fc2014-03-08 14:41:25 -0600248 Name nameABC("ndn:/a/b/c");
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000249 shared_ptr<Entry> npeABC = nt.lookup(nameABC);
HYuana9b85752014-02-26 02:32:30 -0600250 BOOST_CHECK_EQUAL(nt.size(), 4);
Haowei Yuane1079fc2014-03-08 14:41:25 -0600251
252 Name nameABD("/a/b/d");
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000253 shared_ptr<Entry> npeABD = nt.lookup(nameABD);
HYuana9b85752014-02-26 02:32:30 -0600254 BOOST_CHECK_EQUAL(nt.size(), 5);
255
Haowei Yuane1079fc2014-03-08 14:41:25 -0600256 Name nameAE("/a/e/");
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000257 shared_ptr<Entry> npeAE = nt.lookup(nameAE);
HYuana9b85752014-02-26 02:32:30 -0600258 BOOST_CHECK_EQUAL(nt.size(), 6);
259
Haowei Yuane1079fc2014-03-08 14:41:25 -0600260 Name nameF("/f");
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000261 shared_ptr<Entry> npeF = nt.lookup(nameF);
HYuana9b85752014-02-26 02:32:30 -0600262 BOOST_CHECK_EQUAL(nt.size(), 7);
263
Haowei Yuane1079fc2014-03-08 14:41:25 -0600264 // validate lookup() and findExactMatch()
HYuana9b85752014-02-26 02:32:30 -0600265
266 Name nameAB ("/a/b");
Junxiao Shib660b4c2016-08-06 20:47:44 +0000267 BOOST_CHECK_EQUAL(npeABC->getParent(), nt.findExactMatch(nameAB).get());
268 BOOST_CHECK_EQUAL(npeABD->getParent(), nt.findExactMatch(nameAB).get());
HYuana9b85752014-02-26 02:32:30 -0600269
270 Name nameA ("/a");
Junxiao Shib660b4c2016-08-06 20:47:44 +0000271 BOOST_CHECK_EQUAL(npeAE->getParent(), nt.findExactMatch(nameA).get());
HYuana9b85752014-02-26 02:32:30 -0600272
273 Name nameRoot ("/");
Junxiao Shib660b4c2016-08-06 20:47:44 +0000274 BOOST_CHECK_EQUAL(npeF->getParent(), nt.findExactMatch(nameRoot).get());
HYuana9b85752014-02-26 02:32:30 -0600275 BOOST_CHECK_EQUAL(nt.size(), 7);
276
Haowei Yuane1079fc2014-03-08 14:41:25 -0600277 Name name0("/does/not/exist");
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000278 shared_ptr<Entry> npe0 = nt.findExactMatch(name0);
279 BOOST_CHECK(npe0 == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600280
281
282 // Longest Prefix Matching
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000283 shared_ptr<Entry> temp;
HYuana9b85752014-02-26 02:32:30 -0600284 Name nameABCLPM("/a/b/c/def/asdf/nlf");
285 temp = nt.findLongestPrefixMatch(nameABCLPM);
286 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameABC));
287
288 Name nameABDLPM("/a/b/d/def/asdf/nlf");
289 temp = nt.findLongestPrefixMatch(nameABDLPM);
290 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameABD));
291
292 Name nameABLPM("/a/b/hello/world");
293 temp = nt.findLongestPrefixMatch(nameABLPM);
294 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameAB));
295
296 Name nameAELPM("/a/e/hello/world");
297 temp = nt.findLongestPrefixMatch(nameAELPM);
298 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameAE));
299
300 Name nameALPM("/a/hello/world");
301 temp = nt.findLongestPrefixMatch(nameALPM);
302 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameA));
303
304 Name nameFLPM("/f/hello/world");
305 temp = nt.findLongestPrefixMatch(nameFLPM);
306 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameF));
307
308 Name nameRootLPM("/does_not_exist");
309 temp = nt.findLongestPrefixMatch(nameRootLPM);
310 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameRoot));
311
HYuana9b85752014-02-26 02:32:30 -0600312 bool eraseResult = false;
313 temp = nt.findExactMatch(nameABC);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000314 if (temp != nullptr)
Junxiao Shie3cf2852016-08-09 03:50:56 +0000315 eraseResult = nt.eraseIfEmpty(temp.get());
HYuana9b85752014-02-26 02:32:30 -0600316 BOOST_CHECK_EQUAL(nt.size(), 6);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000317 BOOST_CHECK(nt.findExactMatch(nameABC) == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600318 BOOST_CHECK_EQUAL(eraseResult, true);
319
320 eraseResult = false;
321 temp = nt.findExactMatch(nameABCLPM);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000322 if (temp != nullptr)
Junxiao Shie3cf2852016-08-09 03:50:56 +0000323 eraseResult = nt.eraseIfEmpty(temp.get());
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000324 BOOST_CHECK(temp == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600325 BOOST_CHECK_EQUAL(nt.size(), 6);
326 BOOST_CHECK_EQUAL(eraseResult, false);
327
HYuana9b85752014-02-26 02:32:30 -0600328 nt.lookup(nameABC);
329 BOOST_CHECK_EQUAL(nt.size(), 7);
330
331 eraseResult = false;
332 temp = nt.findExactMatch(nameABC);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000333 if (temp != nullptr)
Junxiao Shie3cf2852016-08-09 03:50:56 +0000334 eraseResult = nt.eraseIfEmpty(temp.get());
HYuana9b85752014-02-26 02:32:30 -0600335 BOOST_CHECK_EQUAL(nt.size(), 6);
336 BOOST_CHECK_EQUAL(eraseResult, true);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000337 BOOST_CHECK(nt.findExactMatch(nameABC) == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600338
HYuana9b85752014-02-26 02:32:30 -0600339 BOOST_CHECK_EQUAL(nt.getNBuckets(), 16);
340
Haowei Yuane1079fc2014-03-08 14:41:25 -0600341 // should resize now
HYuana9b85752014-02-26 02:32:30 -0600342 Name nameABCD("a/b/c/d");
343 nt.lookup(nameABCD);
344 Name nameABCDE("a/b/c/d/e");
345 nt.lookup(nameABCDE);
346 BOOST_CHECK_EQUAL(nt.size(), 9);
347 BOOST_CHECK_EQUAL(nt.getNBuckets(), 32);
348
Haowei Yuane1079fc2014-03-08 14:41:25 -0600349 // try to erase /a/b/c, should return false
HYuana9b85752014-02-26 02:32:30 -0600350 temp = nt.findExactMatch(nameABC);
Junxiao Shie3cf2852016-08-09 03:50:56 +0000351 BOOST_CHECK_EQUAL(temp->getName(), nameABC);
352 eraseResult = nt.eraseIfEmpty(temp.get());
HYuana9b85752014-02-26 02:32:30 -0600353 BOOST_CHECK_EQUAL(eraseResult, false);
354 temp = nt.findExactMatch(nameABC);
Junxiao Shie3cf2852016-08-09 03:50:56 +0000355 BOOST_CHECK_EQUAL(temp->getName(), nameABC);
HYuana9b85752014-02-26 02:32:30 -0600356
357 temp = nt.findExactMatch(nameABD);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000358 if (temp != nullptr)
Junxiao Shie3cf2852016-08-09 03:50:56 +0000359 nt.eraseIfEmpty(temp.get());
HYuana9b85752014-02-26 02:32:30 -0600360 BOOST_CHECK_EQUAL(nt.size(), 8);
Haowei Yuane1079fc2014-03-08 14:41:25 -0600361}
HYuana9b85752014-02-26 02:32:30 -0600362
Junxiao Shi60607c72014-11-26 22:40:36 -0700363/** \brief verify a NameTree enumeration contains expected entries
364 *
365 * Example:
Junxiao Shie3cf2852016-08-09 03:50:56 +0000366 * \code
Junxiao Shi60607c72014-11-26 22:40:36 -0700367 * auto& enumerable = ...;
368 * EnumerationVerifier(enumerable)
369 * .expect("/A")
370 * .expect("/B")
371 * .end();
372 * // enumerable must have /A /B and nothing else to pass the test.
373 * \endcode
374 */
375class EnumerationVerifier : noncopyable
Haowei Yuane1079fc2014-03-08 14:41:25 -0600376{
Junxiao Shi60607c72014-11-26 22:40:36 -0700377public:
378 template<typename Enumerable>
379 EnumerationVerifier(Enumerable&& enumerable)
380 {
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000381 for (const Entry& entry : enumerable) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000382 const Name& name = entry.getName();
Junxiao Shi60607c72014-11-26 22:40:36 -0700383 BOOST_CHECK_MESSAGE(m_names.insert(name).second, "duplicate Name " << name);
384 }
385 }
HYuana9b85752014-02-26 02:32:30 -0600386
Junxiao Shi60607c72014-11-26 22:40:36 -0700387 EnumerationVerifier&
388 expect(const Name& name)
389 {
390 BOOST_CHECK_MESSAGE(m_names.erase(name) == 1, "missing Name " << name);
391 return *this;
392 }
Haowei Yuane1079fc2014-03-08 14:41:25 -0600393
Junxiao Shi60607c72014-11-26 22:40:36 -0700394 void
395 end()
396 {
Alexander Afanasyeve84044e2015-02-26 21:52:18 -0800397 BOOST_CHECK(m_names.empty());
Junxiao Shi60607c72014-11-26 22:40:36 -0700398 }
399
400private:
401 std::unordered_set<Name> m_names;
402};
403
404class EnumerationFixture : public BaseFixture
405{
406protected:
407 EnumerationFixture()
408 : nt(N_BUCKETS)
409 {
410 BOOST_CHECK_EQUAL(nt.size(), 0);
411 BOOST_CHECK_EQUAL(nt.getNBuckets(), N_BUCKETS);
412 }
413
414 void
415 insertAbAc()
416 {
417 nt.lookup("/a/b");
418 nt.lookup("/a/c");
419 BOOST_CHECK_EQUAL(nt.size(), 4);
420 // /, /a, /a/b, /a/c
421 }
422
423 void
424 insertAb1Ab2Ac1Ac2()
425 {
426 nt.lookup("/a/b/1");
427 nt.lookup("/a/b/2");
428 nt.lookup("/a/c/1");
429 nt.lookup("/a/c/2");
430 BOOST_CHECK_EQUAL(nt.size(), 8);
431 // /, /a, /a/b, /a/b/1, /a/b/2, /a/c, /a/c/1, /a/c/2
432 }
433
434protected:
435 static const size_t N_BUCKETS = 16;
436 NameTree nt;
437};
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000438
Junxiao Shi60607c72014-11-26 22:40:36 -0700439const size_t EnumerationFixture::N_BUCKETS;
440
441BOOST_FIXTURE_TEST_CASE(IteratorFullEnumerate, EnumerationFixture)
442{
443 nt.lookup("/a/b/c");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600444 BOOST_CHECK_EQUAL(nt.size(), 4);
445
Junxiao Shi60607c72014-11-26 22:40:36 -0700446 nt.lookup("/a/b/d");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600447 BOOST_CHECK_EQUAL(nt.size(), 5);
448
Junxiao Shi60607c72014-11-26 22:40:36 -0700449 nt.lookup("/a/e");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600450 BOOST_CHECK_EQUAL(nt.size(), 6);
451
Junxiao Shi60607c72014-11-26 22:40:36 -0700452 nt.lookup("/f");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600453 BOOST_CHECK_EQUAL(nt.size(), 7);
454
Junxiao Shi60607c72014-11-26 22:40:36 -0700455 nt.lookup("/");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600456 BOOST_CHECK_EQUAL(nt.size(), 7);
457
Junxiao Shi60607c72014-11-26 22:40:36 -0700458 auto&& enumerable = nt.fullEnumerate();
459 EnumerationVerifier(enumerable)
460 .expect("/")
461 .expect("/a")
462 .expect("/a/b")
463 .expect("/a/b/c")
464 .expect("/a/b/d")
465 .expect("/a/e")
466 .expect("/f")
467 .end();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600468}
469
Junxiao Shi60607c72014-11-26 22:40:36 -0700470BOOST_FIXTURE_TEST_SUITE(IteratorPartialEnumerate, EnumerationFixture)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600471
Junxiao Shi60607c72014-11-26 22:40:36 -0700472BOOST_AUTO_TEST_CASE(Empty)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600473{
Junxiao Shi60607c72014-11-26 22:40:36 -0700474 auto&& enumerable = nt.partialEnumerate("/a");
475
476 EnumerationVerifier(enumerable)
477 .end();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600478}
479
Junxiao Shi60607c72014-11-26 22:40:36 -0700480BOOST_AUTO_TEST_CASE(NotIn)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600481{
Junxiao Shi60607c72014-11-26 22:40:36 -0700482 this->insertAbAc();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600483
484 // Enumerate on some name that is not in nameTree
Junxiao Shi60607c72014-11-26 22:40:36 -0700485 Name name0("/0");
486 auto&& enumerable = nt.partialEnumerate("/0");
487
488 EnumerationVerifier(enumerable)
489 .end();
490}
491
492BOOST_AUTO_TEST_CASE(OnlyA)
493{
494 this->insertAbAc();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600495
496 // Accept "root" nameA only
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000497 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000498 return std::make_pair(entry.getName() == "/a", true);
Junxiao Shi60607c72014-11-26 22:40:36 -0700499 });
500
501 EnumerationVerifier(enumerable)
502 .expect("/a")
503 .end();
504}
505
506BOOST_AUTO_TEST_CASE(ExceptA)
507{
508 this->insertAbAc();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600509
510 // Accept anything except "root" nameA
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000511 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000512 return std::make_pair(entry.getName() != "/a", true);
Junxiao Shi60607c72014-11-26 22:40:36 -0700513 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600514
Junxiao Shi60607c72014-11-26 22:40:36 -0700515 EnumerationVerifier(enumerable)
516 .expect("/a/b")
517 .expect("/a/c")
518 .end();
519}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600520
Junxiao Shi60607c72014-11-26 22:40:36 -0700521BOOST_AUTO_TEST_CASE(NoNameANoSubTreeAB)
522{
523 this->insertAb1Ab2Ac1Ac2();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600524
525 // No NameA
526 // No SubTree from NameAB
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000527 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000528 return std::make_pair(entry.getName() != "/a", entry.getName() != "/a/b");
Junxiao Shi60607c72014-11-26 22:40:36 -0700529 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600530
Junxiao Shi60607c72014-11-26 22:40:36 -0700531 EnumerationVerifier(enumerable)
532 .expect("/a/b")
533 .expect("/a/c")
534 .expect("/a/c/1")
535 .expect("/a/c/2")
536 .end();
537}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600538
Junxiao Shi60607c72014-11-26 22:40:36 -0700539BOOST_AUTO_TEST_CASE(NoNameANoSubTreeAC)
540{
541 this->insertAb1Ab2Ac1Ac2();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600542
543 // No NameA
544 // No SubTree from NameAC
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000545 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000546 return std::make_pair(entry.getName() != "/a", entry.getName() != "/a/c");
Junxiao Shi60607c72014-11-26 22:40:36 -0700547 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600548
Junxiao Shi60607c72014-11-26 22:40:36 -0700549 EnumerationVerifier(enumerable)
550 .expect("/a/b")
551 .expect("/a/b/1")
552 .expect("/a/b/2")
553 .expect("/a/c")
554 .end();
555}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600556
Junxiao Shi60607c72014-11-26 22:40:36 -0700557BOOST_AUTO_TEST_CASE(NoSubTreeA)
558{
559 this->insertAb1Ab2Ac1Ac2();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600560
561 // No Subtree from NameA
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000562 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000563 return std::make_pair(true, entry.getName() != "/a");
Junxiao Shi60607c72014-11-26 22:40:36 -0700564 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600565
Junxiao Shi60607c72014-11-26 22:40:36 -0700566 EnumerationVerifier(enumerable)
567 .expect("/a")
568 .end();
569}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600570
Junxiao Shi60607c72014-11-26 22:40:36 -0700571BOOST_AUTO_TEST_CASE(Example)
572{
Haowei Yuane1079fc2014-03-08 14:41:25 -0600573 // Example
574 // /
575 // /A
576 // /A/B x
577 // /A/B/C
578 // /A/D x
579 // /E
580 // /F
581
Junxiao Shi60607c72014-11-26 22:40:36 -0700582 nt.lookup("/A");
583 nt.lookup("/A/B");
584 nt.lookup("/A/B/C");
585 nt.lookup("/A/D");
586 nt.lookup("/E");
587 nt.lookup("/F");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600588
Junxiao Shi60607c72014-11-26 22:40:36 -0700589 auto&& enumerable = nt.partialEnumerate("/A",
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000590 [] (const Entry& entry) -> std::pair<bool, bool> {
Junxiao Shi60607c72014-11-26 22:40:36 -0700591 bool visitEntry = false;
592 bool visitChildren = false;
Haowei Yuane1079fc2014-03-08 14:41:25 -0600593
Junxiao Shie3cf2852016-08-09 03:50:56 +0000594 Name name = entry.getName();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600595
Junxiao Shi60607c72014-11-26 22:40:36 -0700596 if (name == "/" || name == "/A/B" || name == "/A/B/C" || name == "/A/D") {
597 visitEntry = true;
598 }
Haowei Yuane1079fc2014-03-08 14:41:25 -0600599
Junxiao Shi60607c72014-11-26 22:40:36 -0700600 if (name == "/" || name == "/A" || name == "/F") {
601 visitChildren = true;
602 }
Haowei Yuane1079fc2014-03-08 14:41:25 -0600603
Junxiao Shi60607c72014-11-26 22:40:36 -0700604 return {visitEntry, visitChildren};
605 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600606
Junxiao Shi60607c72014-11-26 22:40:36 -0700607 EnumerationVerifier(enumerable)
608 .expect("/A/B")
609 .expect("/A/D")
610 .end();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600611}
612
Junxiao Shi60607c72014-11-26 22:40:36 -0700613BOOST_AUTO_TEST_SUITE_END()
Haowei Yuane1079fc2014-03-08 14:41:25 -0600614
Junxiao Shi60607c72014-11-26 22:40:36 -0700615BOOST_FIXTURE_TEST_CASE(IteratorFindAllMatches, EnumerationFixture)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600616{
Junxiao Shi60607c72014-11-26 22:40:36 -0700617 nt.lookup("/a/b/c/d/e/f");
618 nt.lookup("/a/a/c");
619 nt.lookup("/a/a/d/1");
620 nt.lookup("/a/a/d/2");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600621 BOOST_CHECK_EQUAL(nt.size(), 12);
622
Junxiao Shi60607c72014-11-26 22:40:36 -0700623 auto&& allMatches = nt.findAllMatches("/a/b/c/d/e");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600624
Junxiao Shi60607c72014-11-26 22:40:36 -0700625 EnumerationVerifier(allMatches)
626 .expect("/")
627 .expect("/a")
628 .expect("/a/b")
629 .expect("/a/b/c")
630 .expect("/a/b/c/d")
631 .expect("/a/b/c/d/e")
632 .end();
HYuana9b85752014-02-26 02:32:30 -0600633}
634
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700635BOOST_AUTO_TEST_CASE(HashTableResizeShrink)
Haowei Yuanf52dac72014-03-24 23:35:03 -0500636{
637 size_t nBuckets = 16;
638 NameTree nameTree(nBuckets);
639
640 Name prefix("/a/b/c/d/e/f/g/h"); // requires 9 buckets
641
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000642 shared_ptr<Entry> entry = nameTree.lookup(prefix);
Haowei Yuanf52dac72014-03-24 23:35:03 -0500643 BOOST_CHECK_EQUAL(nameTree.size(), 9);
644 BOOST_CHECK_EQUAL(nameTree.getNBuckets(), 32);
645
Junxiao Shie3cf2852016-08-09 03:50:56 +0000646 nameTree.eraseIfEmpty(entry.get());
Haowei Yuanf52dac72014-03-24 23:35:03 -0500647 BOOST_CHECK_EQUAL(nameTree.size(), 0);
648 BOOST_CHECK_EQUAL(nameTree.getNBuckets(), 16);
649}
650
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700651// .lookup should not invalidate iterator
652BOOST_AUTO_TEST_CASE(SurvivedIteratorAfterLookup)
653{
654 NameTree nt;
655 nt.lookup("/A/B/C");
656 nt.lookup("/E");
657
658 Name nameB("/A/B");
659 std::set<Name> seenNames;
660 for (NameTree::const_iterator it = nt.begin(); it != nt.end(); ++it) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000661 BOOST_CHECK(seenNames.insert(it->getName()).second);
662 if (it->getName() == nameB) {
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700663 nt.lookup("/D");
664 }
665 }
666
667 BOOST_CHECK_EQUAL(seenNames.count("/"), 1);
668 BOOST_CHECK_EQUAL(seenNames.count("/A"), 1);
669 BOOST_CHECK_EQUAL(seenNames.count("/A/B"), 1);
670 BOOST_CHECK_EQUAL(seenNames.count("/A/B/C"), 1);
671 BOOST_CHECK_EQUAL(seenNames.count("/E"), 1);
672
673 seenNames.erase("/D"); // /D may or may not appear
674 BOOST_CHECK(seenNames.size() == 5);
675}
676
Junxiao Shie3cf2852016-08-09 03:50:56 +0000677// .eraseIfEmpty should not invalidate iterator
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700678BOOST_AUTO_TEST_CASE(SurvivedIteratorAfterErase)
679{
680 NameTree nt;
681 nt.lookup("/A/B/C");
682 nt.lookup("/A/D/E");
683 nt.lookup("/A/F/G");
684 nt.lookup("/H");
685
686 Name nameD("/A/D");
687 std::set<Name> seenNames;
688 for (NameTree::const_iterator it = nt.begin(); it != nt.end(); ++it) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000689 BOOST_CHECK(seenNames.insert(it->getName()).second);
690 if (it->getName() == nameD) {
691 nt.eraseIfEmpty(nt.findExactMatch("/A/F/G").get()); // /A/F/G and /A/F are erased
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700692 }
693 }
694
695 BOOST_CHECK_EQUAL(seenNames.count("/"), 1);
696 BOOST_CHECK_EQUAL(seenNames.count("/A"), 1);
697 BOOST_CHECK_EQUAL(seenNames.count("/A/B"), 1);
698 BOOST_CHECK_EQUAL(seenNames.count("/A/B/C"), 1);
699 BOOST_CHECK_EQUAL(seenNames.count("/A/D"), 1);
700 BOOST_CHECK_EQUAL(seenNames.count("/A/D/E"), 1);
701 BOOST_CHECK_EQUAL(seenNames.count("/H"), 1);
702
703 seenNames.erase("/A/F"); // /A/F may or may not appear
704 seenNames.erase("/A/F/G"); // /A/F/G may or may not appear
705 BOOST_CHECK(seenNames.size() == 7);
706}
707
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000708BOOST_AUTO_TEST_SUITE_END() // TestNameTree
709BOOST_AUTO_TEST_SUITE_END() // Table
HYuana9b85752014-02-26 02:32:30 -0600710
Junxiao Shid9ee45c2014-02-27 15:38:11 -0700711} // namespace tests
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000712} // namespace name_tree
HYuana9b85752014-02-26 02:32:30 -0600713} // namespace nfd