blob: bf4602200a749221c389d484b844fe8007bb7bd7 [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
Junxiao Shi7f358432016-08-11 17:06:33 +0000248 // lookup
249
Haowei Yuane1079fc2014-03-08 14:41:25 -0600250 Name nameABC("ndn:/a/b/c");
Junxiao Shi7f358432016-08-11 17:06:33 +0000251 Entry& npeABC = nt.lookup(nameABC);
HYuana9b85752014-02-26 02:32:30 -0600252 BOOST_CHECK_EQUAL(nt.size(), 4);
Haowei Yuane1079fc2014-03-08 14:41:25 -0600253
254 Name nameABD("/a/b/d");
Junxiao Shi7f358432016-08-11 17:06:33 +0000255 Entry& npeABD = nt.lookup(nameABD);
HYuana9b85752014-02-26 02:32:30 -0600256 BOOST_CHECK_EQUAL(nt.size(), 5);
257
Haowei Yuane1079fc2014-03-08 14:41:25 -0600258 Name nameAE("/a/e/");
Junxiao Shi7f358432016-08-11 17:06:33 +0000259 Entry& npeAE = nt.lookup(nameAE);
HYuana9b85752014-02-26 02:32:30 -0600260 BOOST_CHECK_EQUAL(nt.size(), 6);
261
Haowei Yuane1079fc2014-03-08 14:41:25 -0600262 Name nameF("/f");
Junxiao Shi7f358432016-08-11 17:06:33 +0000263 Entry& npeF = nt.lookup(nameF);
HYuana9b85752014-02-26 02:32:30 -0600264 BOOST_CHECK_EQUAL(nt.size(), 7);
265
Junxiao Shi7f358432016-08-11 17:06:33 +0000266 // getParent and findExactMatch
HYuana9b85752014-02-26 02:32:30 -0600267
Junxiao Shi811c0102016-08-10 04:12:45 +0000268 Name nameAB("/a/b");
Junxiao Shi7f358432016-08-11 17:06:33 +0000269 BOOST_CHECK_EQUAL(npeABC.getParent(), nt.findExactMatch(nameAB));
270 BOOST_CHECK_EQUAL(npeABD.getParent(), nt.findExactMatch(nameAB));
HYuana9b85752014-02-26 02:32:30 -0600271
Junxiao Shi7f358432016-08-11 17:06:33 +0000272 Name nameA("/a");
273 BOOST_CHECK_EQUAL(npeAE.getParent(), nt.findExactMatch(nameA));
HYuana9b85752014-02-26 02:32:30 -0600274
Junxiao Shi7f358432016-08-11 17:06:33 +0000275 Name nameRoot("/");
276 BOOST_CHECK_EQUAL(npeF.getParent(), nt.findExactMatch(nameRoot));
HYuana9b85752014-02-26 02:32:30 -0600277 BOOST_CHECK_EQUAL(nt.size(), 7);
278
Haowei Yuane1079fc2014-03-08 14:41:25 -0600279 Name name0("/does/not/exist");
Junxiao Shi811c0102016-08-10 04:12:45 +0000280 Entry* npe0 = nt.findExactMatch(name0);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000281 BOOST_CHECK(npe0 == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600282
283
Junxiao Shi7f358432016-08-11 17:06:33 +0000284 // findLongestPrefixMatch
285
Junxiao Shi811c0102016-08-10 04:12:45 +0000286 Entry* temp = nullptr;
HYuana9b85752014-02-26 02:32:30 -0600287 Name nameABCLPM("/a/b/c/def/asdf/nlf");
288 temp = nt.findLongestPrefixMatch(nameABCLPM);
289 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameABC));
290
291 Name nameABDLPM("/a/b/d/def/asdf/nlf");
292 temp = nt.findLongestPrefixMatch(nameABDLPM);
293 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameABD));
294
295 Name nameABLPM("/a/b/hello/world");
296 temp = nt.findLongestPrefixMatch(nameABLPM);
297 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameAB));
298
299 Name nameAELPM("/a/e/hello/world");
300 temp = nt.findLongestPrefixMatch(nameAELPM);
301 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameAE));
302
303 Name nameALPM("/a/hello/world");
304 temp = nt.findLongestPrefixMatch(nameALPM);
305 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameA));
306
307 Name nameFLPM("/f/hello/world");
308 temp = nt.findLongestPrefixMatch(nameFLPM);
309 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameF));
310
311 Name nameRootLPM("/does_not_exist");
312 temp = nt.findLongestPrefixMatch(nameRootLPM);
313 BOOST_CHECK_EQUAL(temp, nt.findExactMatch(nameRoot));
314
HYuana9b85752014-02-26 02:32:30 -0600315 bool eraseResult = false;
316 temp = nt.findExactMatch(nameABC);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000317 if (temp != nullptr)
Junxiao Shi811c0102016-08-10 04:12:45 +0000318 eraseResult = nt.eraseIfEmpty(temp);
HYuana9b85752014-02-26 02:32:30 -0600319 BOOST_CHECK_EQUAL(nt.size(), 6);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000320 BOOST_CHECK(nt.findExactMatch(nameABC) == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600321 BOOST_CHECK_EQUAL(eraseResult, true);
322
323 eraseResult = false;
324 temp = nt.findExactMatch(nameABCLPM);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000325 if (temp != nullptr)
Junxiao Shi811c0102016-08-10 04:12:45 +0000326 eraseResult = nt.eraseIfEmpty(temp);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000327 BOOST_CHECK(temp == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600328 BOOST_CHECK_EQUAL(nt.size(), 6);
329 BOOST_CHECK_EQUAL(eraseResult, false);
330
HYuana9b85752014-02-26 02:32:30 -0600331 nt.lookup(nameABC);
332 BOOST_CHECK_EQUAL(nt.size(), 7);
333
334 eraseResult = false;
335 temp = nt.findExactMatch(nameABC);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000336 if (temp != nullptr)
Junxiao Shi811c0102016-08-10 04:12:45 +0000337 eraseResult = nt.eraseIfEmpty(temp);
HYuana9b85752014-02-26 02:32:30 -0600338 BOOST_CHECK_EQUAL(nt.size(), 6);
339 BOOST_CHECK_EQUAL(eraseResult, true);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000340 BOOST_CHECK(nt.findExactMatch(nameABC) == nullptr);
HYuana9b85752014-02-26 02:32:30 -0600341
HYuana9b85752014-02-26 02:32:30 -0600342 BOOST_CHECK_EQUAL(nt.getNBuckets(), 16);
343
Haowei Yuane1079fc2014-03-08 14:41:25 -0600344 // should resize now
HYuana9b85752014-02-26 02:32:30 -0600345 Name nameABCD("a/b/c/d");
346 nt.lookup(nameABCD);
347 Name nameABCDE("a/b/c/d/e");
348 nt.lookup(nameABCDE);
349 BOOST_CHECK_EQUAL(nt.size(), 9);
350 BOOST_CHECK_EQUAL(nt.getNBuckets(), 32);
351
Haowei Yuane1079fc2014-03-08 14:41:25 -0600352 // try to erase /a/b/c, should return false
HYuana9b85752014-02-26 02:32:30 -0600353 temp = nt.findExactMatch(nameABC);
Junxiao Shie3cf2852016-08-09 03:50:56 +0000354 BOOST_CHECK_EQUAL(temp->getName(), nameABC);
Junxiao Shi811c0102016-08-10 04:12:45 +0000355 eraseResult = nt.eraseIfEmpty(temp);
HYuana9b85752014-02-26 02:32:30 -0600356 BOOST_CHECK_EQUAL(eraseResult, false);
357 temp = nt.findExactMatch(nameABC);
Junxiao Shie3cf2852016-08-09 03:50:56 +0000358 BOOST_CHECK_EQUAL(temp->getName(), nameABC);
HYuana9b85752014-02-26 02:32:30 -0600359
360 temp = nt.findExactMatch(nameABD);
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000361 if (temp != nullptr)
Junxiao Shi811c0102016-08-10 04:12:45 +0000362 nt.eraseIfEmpty(temp);
HYuana9b85752014-02-26 02:32:30 -0600363 BOOST_CHECK_EQUAL(nt.size(), 8);
Haowei Yuane1079fc2014-03-08 14:41:25 -0600364}
HYuana9b85752014-02-26 02:32:30 -0600365
Junxiao Shi60607c72014-11-26 22:40:36 -0700366/** \brief verify a NameTree enumeration contains expected entries
367 *
368 * Example:
Junxiao Shie3cf2852016-08-09 03:50:56 +0000369 * \code
Junxiao Shi60607c72014-11-26 22:40:36 -0700370 * auto& enumerable = ...;
371 * EnumerationVerifier(enumerable)
372 * .expect("/A")
373 * .expect("/B")
374 * .end();
375 * // enumerable must have /A /B and nothing else to pass the test.
376 * \endcode
377 */
378class EnumerationVerifier : noncopyable
Haowei Yuane1079fc2014-03-08 14:41:25 -0600379{
Junxiao Shi60607c72014-11-26 22:40:36 -0700380public:
381 template<typename Enumerable>
382 EnumerationVerifier(Enumerable&& enumerable)
383 {
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000384 for (const Entry& entry : enumerable) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000385 const Name& name = entry.getName();
Junxiao Shi60607c72014-11-26 22:40:36 -0700386 BOOST_CHECK_MESSAGE(m_names.insert(name).second, "duplicate Name " << name);
387 }
388 }
HYuana9b85752014-02-26 02:32:30 -0600389
Junxiao Shi60607c72014-11-26 22:40:36 -0700390 EnumerationVerifier&
391 expect(const Name& name)
392 {
393 BOOST_CHECK_MESSAGE(m_names.erase(name) == 1, "missing Name " << name);
394 return *this;
395 }
Haowei Yuane1079fc2014-03-08 14:41:25 -0600396
Junxiao Shi60607c72014-11-26 22:40:36 -0700397 void
398 end()
399 {
Alexander Afanasyeve84044e2015-02-26 21:52:18 -0800400 BOOST_CHECK(m_names.empty());
Junxiao Shi60607c72014-11-26 22:40:36 -0700401 }
402
403private:
404 std::unordered_set<Name> m_names;
405};
406
407class EnumerationFixture : public BaseFixture
408{
409protected:
410 EnumerationFixture()
411 : nt(N_BUCKETS)
412 {
413 BOOST_CHECK_EQUAL(nt.size(), 0);
414 BOOST_CHECK_EQUAL(nt.getNBuckets(), N_BUCKETS);
415 }
416
417 void
418 insertAbAc()
419 {
420 nt.lookup("/a/b");
421 nt.lookup("/a/c");
422 BOOST_CHECK_EQUAL(nt.size(), 4);
423 // /, /a, /a/b, /a/c
424 }
425
426 void
427 insertAb1Ab2Ac1Ac2()
428 {
429 nt.lookup("/a/b/1");
430 nt.lookup("/a/b/2");
431 nt.lookup("/a/c/1");
432 nt.lookup("/a/c/2");
433 BOOST_CHECK_EQUAL(nt.size(), 8);
434 // /, /a, /a/b, /a/b/1, /a/b/2, /a/c, /a/c/1, /a/c/2
435 }
436
437protected:
438 static const size_t N_BUCKETS = 16;
439 NameTree nt;
440};
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000441
Junxiao Shi60607c72014-11-26 22:40:36 -0700442const size_t EnumerationFixture::N_BUCKETS;
443
444BOOST_FIXTURE_TEST_CASE(IteratorFullEnumerate, EnumerationFixture)
445{
446 nt.lookup("/a/b/c");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600447 BOOST_CHECK_EQUAL(nt.size(), 4);
448
Junxiao Shi60607c72014-11-26 22:40:36 -0700449 nt.lookup("/a/b/d");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600450 BOOST_CHECK_EQUAL(nt.size(), 5);
451
Junxiao Shi60607c72014-11-26 22:40:36 -0700452 nt.lookup("/a/e");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600453 BOOST_CHECK_EQUAL(nt.size(), 6);
454
Junxiao Shi60607c72014-11-26 22:40:36 -0700455 nt.lookup("/f");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600456 BOOST_CHECK_EQUAL(nt.size(), 7);
457
Junxiao Shi60607c72014-11-26 22:40:36 -0700458 nt.lookup("/");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600459 BOOST_CHECK_EQUAL(nt.size(), 7);
460
Junxiao Shi60607c72014-11-26 22:40:36 -0700461 auto&& enumerable = nt.fullEnumerate();
462 EnumerationVerifier(enumerable)
463 .expect("/")
464 .expect("/a")
465 .expect("/a/b")
466 .expect("/a/b/c")
467 .expect("/a/b/d")
468 .expect("/a/e")
469 .expect("/f")
470 .end();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600471}
472
Junxiao Shi60607c72014-11-26 22:40:36 -0700473BOOST_FIXTURE_TEST_SUITE(IteratorPartialEnumerate, EnumerationFixture)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600474
Junxiao Shi60607c72014-11-26 22:40:36 -0700475BOOST_AUTO_TEST_CASE(Empty)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600476{
Junxiao Shi60607c72014-11-26 22:40:36 -0700477 auto&& enumerable = nt.partialEnumerate("/a");
478
479 EnumerationVerifier(enumerable)
480 .end();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600481}
482
Junxiao Shi60607c72014-11-26 22:40:36 -0700483BOOST_AUTO_TEST_CASE(NotIn)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600484{
Junxiao Shi60607c72014-11-26 22:40:36 -0700485 this->insertAbAc();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600486
487 // Enumerate on some name that is not in nameTree
Junxiao Shi60607c72014-11-26 22:40:36 -0700488 Name name0("/0");
489 auto&& enumerable = nt.partialEnumerate("/0");
490
491 EnumerationVerifier(enumerable)
492 .end();
493}
494
495BOOST_AUTO_TEST_CASE(OnlyA)
496{
497 this->insertAbAc();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600498
499 // Accept "root" nameA only
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000500 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000501 return std::make_pair(entry.getName() == "/a", true);
Junxiao Shi60607c72014-11-26 22:40:36 -0700502 });
503
504 EnumerationVerifier(enumerable)
505 .expect("/a")
506 .end();
507}
508
509BOOST_AUTO_TEST_CASE(ExceptA)
510{
511 this->insertAbAc();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600512
513 // Accept anything except "root" nameA
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000514 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000515 return std::make_pair(entry.getName() != "/a", true);
Junxiao Shi60607c72014-11-26 22:40:36 -0700516 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600517
Junxiao Shi60607c72014-11-26 22:40:36 -0700518 EnumerationVerifier(enumerable)
519 .expect("/a/b")
520 .expect("/a/c")
521 .end();
522}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600523
Junxiao Shi60607c72014-11-26 22:40:36 -0700524BOOST_AUTO_TEST_CASE(NoNameANoSubTreeAB)
525{
526 this->insertAb1Ab2Ac1Ac2();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600527
528 // No NameA
529 // No SubTree from NameAB
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000530 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000531 return std::make_pair(entry.getName() != "/a", entry.getName() != "/a/b");
Junxiao Shi60607c72014-11-26 22:40:36 -0700532 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600533
Junxiao Shi60607c72014-11-26 22:40:36 -0700534 EnumerationVerifier(enumerable)
535 .expect("/a/b")
536 .expect("/a/c")
537 .expect("/a/c/1")
538 .expect("/a/c/2")
539 .end();
540}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600541
Junxiao Shi60607c72014-11-26 22:40:36 -0700542BOOST_AUTO_TEST_CASE(NoNameANoSubTreeAC)
543{
544 this->insertAb1Ab2Ac1Ac2();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600545
546 // No NameA
547 // No SubTree from NameAC
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000548 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000549 return std::make_pair(entry.getName() != "/a", entry.getName() != "/a/c");
Junxiao Shi60607c72014-11-26 22:40:36 -0700550 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600551
Junxiao Shi60607c72014-11-26 22:40:36 -0700552 EnumerationVerifier(enumerable)
553 .expect("/a/b")
554 .expect("/a/b/1")
555 .expect("/a/b/2")
556 .expect("/a/c")
557 .end();
558}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600559
Junxiao Shi60607c72014-11-26 22:40:36 -0700560BOOST_AUTO_TEST_CASE(NoSubTreeA)
561{
562 this->insertAb1Ab2Ac1Ac2();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600563
564 // No Subtree from NameA
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000565 auto&& enumerable = nt.partialEnumerate("/a", [] (const Entry& entry) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000566 return std::make_pair(true, entry.getName() != "/a");
Junxiao Shi60607c72014-11-26 22:40:36 -0700567 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600568
Junxiao Shi60607c72014-11-26 22:40:36 -0700569 EnumerationVerifier(enumerable)
570 .expect("/a")
571 .end();
572}
Haowei Yuane1079fc2014-03-08 14:41:25 -0600573
Junxiao Shi60607c72014-11-26 22:40:36 -0700574BOOST_AUTO_TEST_CASE(Example)
575{
Haowei Yuane1079fc2014-03-08 14:41:25 -0600576 // Example
577 // /
578 // /A
579 // /A/B x
580 // /A/B/C
581 // /A/D x
582 // /E
583 // /F
584
Junxiao Shi60607c72014-11-26 22:40:36 -0700585 nt.lookup("/A");
586 nt.lookup("/A/B");
587 nt.lookup("/A/B/C");
588 nt.lookup("/A/D");
589 nt.lookup("/E");
590 nt.lookup("/F");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600591
Junxiao Shi60607c72014-11-26 22:40:36 -0700592 auto&& enumerable = nt.partialEnumerate("/A",
Junxiao Shi811c0102016-08-10 04:12:45 +0000593 [] (const Entry& entry) {
Junxiao Shi60607c72014-11-26 22:40:36 -0700594 bool visitEntry = false;
595 bool visitChildren = false;
Haowei Yuane1079fc2014-03-08 14:41:25 -0600596
Junxiao Shie3cf2852016-08-09 03:50:56 +0000597 Name name = entry.getName();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600598
Junxiao Shi60607c72014-11-26 22:40:36 -0700599 if (name == "/" || name == "/A/B" || name == "/A/B/C" || name == "/A/D") {
600 visitEntry = true;
601 }
Haowei Yuane1079fc2014-03-08 14:41:25 -0600602
Junxiao Shi60607c72014-11-26 22:40:36 -0700603 if (name == "/" || name == "/A" || name == "/F") {
604 visitChildren = true;
605 }
Haowei Yuane1079fc2014-03-08 14:41:25 -0600606
Junxiao Shi811c0102016-08-10 04:12:45 +0000607 return std::make_pair(visitEntry, visitChildren);
Junxiao Shi60607c72014-11-26 22:40:36 -0700608 });
Haowei Yuane1079fc2014-03-08 14:41:25 -0600609
Junxiao Shi60607c72014-11-26 22:40:36 -0700610 EnumerationVerifier(enumerable)
611 .expect("/A/B")
612 .expect("/A/D")
613 .end();
Haowei Yuane1079fc2014-03-08 14:41:25 -0600614}
615
Junxiao Shi60607c72014-11-26 22:40:36 -0700616BOOST_AUTO_TEST_SUITE_END()
Haowei Yuane1079fc2014-03-08 14:41:25 -0600617
Junxiao Shi60607c72014-11-26 22:40:36 -0700618BOOST_FIXTURE_TEST_CASE(IteratorFindAllMatches, EnumerationFixture)
Haowei Yuane1079fc2014-03-08 14:41:25 -0600619{
Junxiao Shi60607c72014-11-26 22:40:36 -0700620 nt.lookup("/a/b/c/d/e/f");
621 nt.lookup("/a/a/c");
622 nt.lookup("/a/a/d/1");
623 nt.lookup("/a/a/d/2");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600624 BOOST_CHECK_EQUAL(nt.size(), 12);
625
Junxiao Shi60607c72014-11-26 22:40:36 -0700626 auto&& allMatches = nt.findAllMatches("/a/b/c/d/e");
Haowei Yuane1079fc2014-03-08 14:41:25 -0600627
Junxiao Shi60607c72014-11-26 22:40:36 -0700628 EnumerationVerifier(allMatches)
629 .expect("/")
630 .expect("/a")
631 .expect("/a/b")
632 .expect("/a/b/c")
633 .expect("/a/b/c/d")
634 .expect("/a/b/c/d/e")
635 .end();
HYuana9b85752014-02-26 02:32:30 -0600636}
637
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700638BOOST_AUTO_TEST_CASE(HashTableResizeShrink)
Haowei Yuanf52dac72014-03-24 23:35:03 -0500639{
640 size_t nBuckets = 16;
641 NameTree nameTree(nBuckets);
642
643 Name prefix("/a/b/c/d/e/f/g/h"); // requires 9 buckets
644
Junxiao Shi7f358432016-08-11 17:06:33 +0000645 Entry& entry = nameTree.lookup(prefix);
Haowei Yuanf52dac72014-03-24 23:35:03 -0500646 BOOST_CHECK_EQUAL(nameTree.size(), 9);
647 BOOST_CHECK_EQUAL(nameTree.getNBuckets(), 32);
648
Junxiao Shi7f358432016-08-11 17:06:33 +0000649 nameTree.eraseIfEmpty(&entry);
Haowei Yuanf52dac72014-03-24 23:35:03 -0500650 BOOST_CHECK_EQUAL(nameTree.size(), 0);
651 BOOST_CHECK_EQUAL(nameTree.getNBuckets(), 16);
652}
653
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700654// .lookup should not invalidate iterator
655BOOST_AUTO_TEST_CASE(SurvivedIteratorAfterLookup)
656{
657 NameTree nt;
658 nt.lookup("/A/B/C");
659 nt.lookup("/E");
660
661 Name nameB("/A/B");
662 std::set<Name> seenNames;
663 for (NameTree::const_iterator it = nt.begin(); it != nt.end(); ++it) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000664 BOOST_CHECK(seenNames.insert(it->getName()).second);
665 if (it->getName() == nameB) {
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700666 nt.lookup("/D");
667 }
668 }
669
670 BOOST_CHECK_EQUAL(seenNames.count("/"), 1);
671 BOOST_CHECK_EQUAL(seenNames.count("/A"), 1);
672 BOOST_CHECK_EQUAL(seenNames.count("/A/B"), 1);
673 BOOST_CHECK_EQUAL(seenNames.count("/A/B/C"), 1);
674 BOOST_CHECK_EQUAL(seenNames.count("/E"), 1);
675
676 seenNames.erase("/D"); // /D may or may not appear
677 BOOST_CHECK(seenNames.size() == 5);
678}
679
Junxiao Shie3cf2852016-08-09 03:50:56 +0000680// .eraseIfEmpty should not invalidate iterator
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700681BOOST_AUTO_TEST_CASE(SurvivedIteratorAfterErase)
682{
683 NameTree nt;
684 nt.lookup("/A/B/C");
685 nt.lookup("/A/D/E");
686 nt.lookup("/A/F/G");
687 nt.lookup("/H");
688
689 Name nameD("/A/D");
690 std::set<Name> seenNames;
691 for (NameTree::const_iterator it = nt.begin(); it != nt.end(); ++it) {
Junxiao Shie3cf2852016-08-09 03:50:56 +0000692 BOOST_CHECK(seenNames.insert(it->getName()).second);
693 if (it->getName() == nameD) {
Junxiao Shi811c0102016-08-10 04:12:45 +0000694 nt.eraseIfEmpty(nt.findExactMatch("/A/F/G")); // /A/F/G and /A/F are erased
Alexander Afanasyevb3033242014-08-04 11:09:05 -0700695 }
696 }
697
698 BOOST_CHECK_EQUAL(seenNames.count("/"), 1);
699 BOOST_CHECK_EQUAL(seenNames.count("/A"), 1);
700 BOOST_CHECK_EQUAL(seenNames.count("/A/B"), 1);
701 BOOST_CHECK_EQUAL(seenNames.count("/A/B/C"), 1);
702 BOOST_CHECK_EQUAL(seenNames.count("/A/D"), 1);
703 BOOST_CHECK_EQUAL(seenNames.count("/A/D/E"), 1);
704 BOOST_CHECK_EQUAL(seenNames.count("/H"), 1);
705
706 seenNames.erase("/A/F"); // /A/F may or may not appear
707 seenNames.erase("/A/F/G"); // /A/F/G may or may not appear
708 BOOST_CHECK(seenNames.size() == 7);
709}
710
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000711BOOST_AUTO_TEST_SUITE_END() // TestNameTree
712BOOST_AUTO_TEST_SUITE_END() // Table
HYuana9b85752014-02-26 02:32:30 -0600713
Junxiao Shid9ee45c2014-02-27 15:38:11 -0700714} // namespace tests
Junxiao Shi2570f3e2016-07-27 02:48:29 +0000715} // namespace name_tree
HYuana9b85752014-02-26 02:32:30 -0600716} // namespace nfd