OLD | NEW |
| (Empty) |
1 // Copyright (c) 2014, the Dart project authors. Please see the AUTHORS file | |
2 // for details. All rights reserved. Use of this source code is governed by a | |
3 // BSD-style license that can be found in the LICENSE file. | |
4 | |
5 #include <algorithm> | |
6 #include <cstring> | |
7 #include <map> | |
8 #include <set> | |
9 #include <string> | |
10 #include <utility> | |
11 #include <vector> | |
12 | |
13 #include "platform/assert.h" | |
14 #include "vm/unit_test.h" | |
15 #include "vm/hash_table.h" | |
16 | |
17 namespace dart { | |
18 | |
19 // Various ways to look up strings. Uses length as the hash code to make it | |
20 // easy to engineer collisions. | |
21 class TestTraits { | |
22 public: | |
23 static bool IsMatch(const char* key, const Object& obj) { | |
24 return String::Cast(obj).Equals(key); | |
25 } | |
26 static uword Hash(const char* key) { | |
27 return static_cast<uword>(strlen(key)); | |
28 } | |
29 static bool IsMatch(const Object& a, const Object& b) { | |
30 return a.IsString() && b.IsString() && | |
31 String::Cast(a).Equals(String::Cast(b)); | |
32 } | |
33 static uword Hash(const Object& obj) { | |
34 return String::Cast(obj).Length(); | |
35 } | |
36 }; | |
37 | |
38 | |
39 template<typename Table> | |
40 void Validate(const Table& table) { | |
41 // Verify consistency of entry state tracking. | |
42 intptr_t num_entries = table.NumEntries(); | |
43 intptr_t num_unused = table.NumUnused(); | |
44 intptr_t num_occupied = table.NumOccupied(); | |
45 intptr_t num_deleted = table.NumDeleted(); | |
46 for (intptr_t i = 0; i < num_entries; ++i) { | |
47 EXPECT_EQ(1, table.IsUnused(i) + table.IsOccupied(i) + table.IsDeleted(i)); | |
48 num_unused -= table.IsUnused(i); | |
49 num_occupied -= table.IsOccupied(i); | |
50 num_deleted -= table.IsDeleted(i); | |
51 } | |
52 EXPECT_EQ(0, num_unused); | |
53 EXPECT_EQ(0, num_occupied); | |
54 EXPECT_EQ(0, num_deleted); | |
55 } | |
56 | |
57 | |
58 TEST_CASE(HashTable) { | |
59 typedef HashTable<TestTraits, 2, 1> Table; | |
60 Table table(Array::Handle(HashTables::New<Table>(5))); | |
61 // Ensure that we did get at least 5 entries. | |
62 EXPECT_LE(5, table.NumEntries()); | |
63 EXPECT_EQ(0, table.NumOccupied()); | |
64 Validate(table); | |
65 EXPECT_EQ(-1, table.FindKey("a")); | |
66 | |
67 // Insertion and lookup. | |
68 intptr_t a_entry = -1; | |
69 EXPECT(!table.FindKeyOrDeletedOrUnused("a", &a_entry)); | |
70 EXPECT_NE(-1, a_entry); | |
71 String& a = String::Handle(String::New("a")); | |
72 table.InsertKey(a_entry, a); | |
73 EXPECT_EQ(1, table.NumOccupied()); | |
74 Validate(table); | |
75 EXPECT_EQ(a_entry, table.FindKey("a")); | |
76 EXPECT_EQ(-1, table.FindKey("b")); | |
77 intptr_t a_entry_again = -1; | |
78 EXPECT(table.FindKeyOrDeletedOrUnused("a", &a_entry_again)); | |
79 EXPECT_EQ(a_entry, a_entry_again); | |
80 intptr_t b_entry = -1; | |
81 EXPECT(!table.FindKeyOrDeletedOrUnused("b", &b_entry)); | |
82 String& b = String::Handle(String::New("b")); | |
83 table.InsertKey(b_entry, b); | |
84 EXPECT_EQ(2, table.NumOccupied()); | |
85 Validate(table); | |
86 | |
87 // Deletion. | |
88 table.DeleteEntry(a_entry); | |
89 EXPECT_EQ(1, table.NumOccupied()); | |
90 Validate(table); | |
91 EXPECT_EQ(-1, table.FindKey("a")); | |
92 EXPECT_EQ(b_entry, table.FindKey("b")); | |
93 intptr_t c_entry = -1; | |
94 EXPECT(!table.FindKeyOrDeletedOrUnused("c", &c_entry)); | |
95 String& c = String::Handle(String::New("c")); | |
96 table.InsertKey(c_entry, c); | |
97 EXPECT_EQ(2, table.NumOccupied()); | |
98 Validate(table); | |
99 EXPECT_EQ(c_entry, table.FindKey("c")); | |
100 | |
101 // Ensure we can actually reach 5 occupied entries (without expansion). | |
102 { | |
103 intptr_t entry = -1; | |
104 EXPECT(!table.FindKeyOrDeletedOrUnused("d", &entry)); | |
105 String& k = String::Handle(String::New("d")); | |
106 table.InsertKey(entry, k); | |
107 EXPECT(!table.FindKeyOrDeletedOrUnused("e", &entry)); | |
108 k = String::New("e"); | |
109 table.InsertKey(entry, k); | |
110 EXPECT(!table.FindKeyOrDeletedOrUnused("f", &entry)); | |
111 k = String::New("f"); | |
112 table.InsertKey(entry, k); | |
113 EXPECT_EQ(5, table.NumOccupied()); | |
114 } | |
115 table.Release(); | |
116 } | |
117 | |
118 | |
119 TEST_CASE(EnumIndexHashMap) { | |
120 typedef EnumIndexHashMap<TestTraits> Table; | |
121 Table table(Array::Handle(HashTables::New<Table>(5))); | |
122 table.UpdateOrInsert(String::Handle(String::New("a")), | |
123 String::Handle(String::New("A"))); | |
124 EXPECT(table.ContainsKey("a")); | |
125 table.UpdateValue("a", String::Handle(String::New("AAA"))); | |
126 String& a_value = String::Handle(); | |
127 a_value ^= table.GetOrNull("a"); | |
128 EXPECT(a_value.Equals("AAA")); | |
129 Object& null_value = Object::Handle(table.GetOrNull("0")); | |
130 EXPECT(null_value.IsNull()); | |
131 table.Release(); | |
132 } | |
133 | |
134 | |
135 std::string ToStdString(const String& str) { | |
136 EXPECT(str.IsOneByteString()); | |
137 std::string result; | |
138 for (intptr_t i = 0; i < str.Length(); ++i) { | |
139 result += static_cast<char>(str.CharAt(i)); | |
140 } | |
141 return result; | |
142 } | |
143 | |
144 | |
145 // Checks that 'expected' and 'actual' are equal sets. If 'ordered' is true, | |
146 // it also verifies that their iteration orders match, i.e., that actual's | |
147 // insertion order coincides with lexicographic order. | |
148 template<typename Set> | |
149 void VerifyStringSetsEqual(const std::set<std::string>& expected, | |
150 const Set& actual, | |
151 bool ordered) { | |
152 // Get actual keys in iteration order. | |
153 Array& keys = Array::Handle(HashTables::ToArray(actual, true)); | |
154 // Cardinality must match. | |
155 EXPECT_EQ(static_cast<intptr_t>(expected.size()), keys.Length()); | |
156 std::vector<std::string> expected_vec(expected.begin(), expected.end()); | |
157 // Check containment. | |
158 for (uintptr_t i = 0; i < expected_vec.size(); ++i) { | |
159 EXPECT(actual.ContainsKey(expected_vec[i].c_str())); | |
160 } | |
161 // Equality, including order, if requested. | |
162 std::vector<std::string> actual_vec; | |
163 String& key = String::Handle(); | |
164 for (int i = 0; i < keys.Length(); ++i) { | |
165 key ^= keys.At(i); | |
166 actual_vec.push_back(ToStdString(key)); | |
167 } | |
168 if (!ordered) { | |
169 std::sort(actual_vec.begin(), actual_vec.end()); | |
170 } | |
171 EXPECT(std::equal(actual_vec.begin(), actual_vec.end(), | |
172 expected_vec.begin())); | |
173 } | |
174 | |
175 | |
176 // Checks that 'expected' and 'actual' are equal maps. If 'ordered' is true, | |
177 // it also verifies that their iteration orders match, i.e., that actual's | |
178 // insertion order coincides with lexicographic order. | |
179 template<typename Map> | |
180 void VerifyStringMapsEqual(const std::map<std::string, int>& expected, | |
181 const Map& actual, | |
182 bool ordered) { | |
183 intptr_t expected_size = expected.size(); | |
184 // Get actual concatenated (key, value) pairs in iteration order. | |
185 Array& entries = Array::Handle(HashTables::ToArray(actual, true)); | |
186 // Cardinality must match. | |
187 EXPECT_EQ(expected_size * 2, entries.Length()); | |
188 std::vector<std::pair<std::string, int> > expected_vec(expected.begin(), | |
189 expected.end()); | |
190 // Check containment. | |
191 Smi& value = Smi::Handle(); | |
192 for (uintptr_t i = 0; i < expected_vec.size(); ++i) { | |
193 std::string key = expected_vec[i].first; | |
194 EXPECT(actual.ContainsKey(key.c_str())); | |
195 value ^= actual.GetOrNull(key.c_str()); | |
196 EXPECT_EQ(expected_vec[i].second, value.Value()); | |
197 } | |
198 if (!ordered) { | |
199 return; | |
200 } | |
201 // Equality including order. | |
202 std::vector<std::string> actual_vec; | |
203 String& key = String::Handle(); | |
204 for (int i = 0; i < expected_size; ++i) { | |
205 key ^= entries.At(2 * i); | |
206 value ^= entries.At(2 * i + 1); | |
207 EXPECT(expected_vec[i].first == ToStdString(key)); | |
208 EXPECT_EQ(expected_vec[i].second, value.Value()); | |
209 } | |
210 } | |
211 | |
212 | |
213 template<typename Set> | |
214 void TestSet(intptr_t initial_capacity, bool ordered) { | |
215 std::set<std::string> expected; | |
216 Set actual(Array::Handle(HashTables::New<Set>(initial_capacity))); | |
217 // Insert the following strings twice: | |
218 // aaa...aaa (length 26) | |
219 // bbb..bbb | |
220 // ... | |
221 // yy | |
222 // z | |
223 for (int i = 0; i < 2; ++i) { | |
224 for (char ch = 'a'; ch <= 'z'; ++ch) { | |
225 std::string key('z' - ch + 1, ch); | |
226 expected.insert(key); | |
227 bool present = actual.Insert(String::Handle(String::New(key.c_str()))); | |
228 EXPECT_EQ((i != 0), present); | |
229 Validate(actual); | |
230 VerifyStringSetsEqual(expected, actual, ordered); | |
231 } | |
232 } | |
233 // TODO(koda): Delete all entries. | |
234 actual.Release(); | |
235 } | |
236 | |
237 | |
238 template<typename Map> | |
239 void TestMap(intptr_t initial_capacity, bool ordered) { | |
240 std::map<std::string, int> expected; | |
241 Map actual(Array::Handle(HashTables::New<Map>(initial_capacity))); | |
242 // Insert the following (strings, int) mapping: | |
243 // aaa...aaa -> 26 | |
244 // bbb..bbb -> 25 | |
245 // ... | |
246 // yy -> 2 | |
247 // z -> 1 | |
248 for (int i = 0; i < 2; ++i) { | |
249 for (char ch = 'a'; ch <= 'z'; ++ch) { | |
250 int length = 'z' - ch + 1; | |
251 std::string key(length, ch); | |
252 // Map everything to zero initially, then update to their final values. | |
253 int value = length * i; | |
254 expected[key] = value; | |
255 bool present = | |
256 actual.UpdateOrInsert(String::Handle(String::New(key.c_str())), | |
257 Smi::Handle(Smi::New(value))); | |
258 EXPECT_EQ((i != 0), present); | |
259 Validate(actual); | |
260 VerifyStringMapsEqual(expected, actual, ordered); | |
261 } | |
262 } | |
263 // TODO(koda): Delete all entries. | |
264 actual.Release(); | |
265 } | |
266 | |
267 | |
268 TEST_CASE(Sets) { | |
269 for (intptr_t initial_capacity = 0; | |
270 initial_capacity < 32; | |
271 ++initial_capacity) { | |
272 TestSet<UnorderedHashSet<TestTraits> >(initial_capacity, false); | |
273 TestSet<EnumIndexHashSet<TestTraits> >(initial_capacity, true); | |
274 } | |
275 } | |
276 | |
277 | |
278 TEST_CASE(Maps) { | |
279 for (intptr_t initial_capacity = 0; | |
280 initial_capacity < 32; | |
281 ++initial_capacity) { | |
282 TestMap<UnorderedHashMap<TestTraits> >(initial_capacity, false); | |
283 TestMap<EnumIndexHashMap<TestTraits> >(initial_capacity, true); | |
284 } | |
285 } | |
286 | |
287 } // namespace dart | |
OLD | NEW |