Mon, 15 Jul 2013 14:25:50 +0200
new map foreach macro
1 /*
2 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
3 *
4 * Copyright 2013 Olaf Wintermann. All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions are met:
8 *
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 *
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
29 #include "map_tests.h"
31 UCX_TEST_IMPLEMENT(test_ucx_map_new) {
32 UcxMap *map = ucx_map_new(16);
33 UCX_TEST_BEGIN
34 UCX_TEST_ASSERT(map->size == 16, "wrong size");
35 UCX_TEST_ASSERT(map->map != NULL, "failed");
37 UCX_TEST_END
38 ucx_map_free(map);
39 }
41 UCX_TEST_IMPLEMENT(test_ucx_key) {
42 UcxKey key = ucx_key((void*)"This is a text.", 15);
43 UCX_TEST_BEGIN
44 UCX_TEST_ASSERT(strncmp((const char*)key.data, "This is a text.", 15) == 0,
45 "failed");
46 UCX_TEST_ASSERT(key.len == 15, "failed");
47 UCX_TEST_ASSERT(key.hash == 1261186027, "hash failed");
49 UCX_TEST_END
50 }
52 UCX_TEST_IMPLEMENT(test_ucx_map_put) {
54 UcxMap *map = ucx_map_new(4);
56 int td[5];
57 td[0] = 10; td[1] = 42; td[2] = 70; td[3] = 11200; td[4] = 80000;
59 UCX_TEST_BEGIN
60 ucx_map_cstr_put(map, "Key2", &td[2]); /* 3.2 */
61 ucx_map_cstr_put(map, "Key0", &td[0]); /* 0.0 */
62 ucx_map_cstr_put(map, "Key1", &td[1]); /* 3.0 */
63 ucx_map_cstr_put(map, "KeY3", &td[3]); /* 3.1 */
64 ucx_map_cstr_put(map, "KEY4", &td[4]); /* 1.0 */
66 UCX_TEST_ASSERT(*((int*)map->map[0]->data) == td[0], "failed Key0");
67 UCX_TEST_ASSERT(*((int*)map->map[1]->data) == td[4], "failed KEY4");
68 UCX_TEST_ASSERT(*((int*)map->map[3]->data) == td[1], "failed Key1");
70 UCX_TEST_ASSERT(map->map[3]->next != NULL, "no list at slot 3");
71 UCX_TEST_ASSERT(map->map[3]->next->next != NULL, "list corrupt at slot 3");
72 UCX_TEST_ASSERT(*((int*)map->map[3]->next->data) == td[3],
73 "failed KeY3")
74 UCX_TEST_ASSERT(*((int*)map->map[3]->next->next->data) == td[2],
75 "failed KeY2");
77 UCX_TEST_ASSERT(map->map[0]->next == NULL, "slot 0 not terminated");
78 UCX_TEST_ASSERT(map->map[1]->next == NULL, "slot 1 not terminated");
79 UCX_TEST_ASSERT(map->map[2] == NULL, "slot 2 not empty");
80 UCX_TEST_ASSERT(map->map[3]->next->next->next == NULL,
81 "slot 3 not terminated")
83 ucx_map_cstr_put(map, "KeY3", &td[4]); /* replace 3.1 */
85 UCX_TEST_ASSERT(*((int*)map->map[3]->data) == td[1],
86 "overwrite failed")
87 UCX_TEST_ASSERT(*((int*)map->map[3]->next->data) == td[4],
88 "overwrite failed");
89 UCX_TEST_ASSERT(*((int*)map->map[3]->next->next->data) == td[2],
90 "overwrite failed")
91 UCX_TEST_ASSERT(map->map[3]->next->next->next == NULL, "overwrite failed");
93 UCX_TEST_END
94 ucx_map_free(map);
95 }
97 UCX_TEST_IMPLEMENT(test_ucx_map_get) {
98 UcxMap *map = ucx_map_new(4);
100 int td[5];
101 td[0] = 10; td[1] = 42; td[2] = 70; td[3] = 11200; td[4] = 80000;
103 ucx_map_cstr_put(map, "Key2", &td[2]);
104 ucx_map_cstr_put(map, "Key0", &td[0]);
105 ucx_map_cstr_put(map, "Key1", &td[1]);
106 ucx_map_cstr_put(map, "KeY3", &td[3]);
107 ucx_map_cstr_put(map, "KEY4", &td[4]);
108 UCX_TEST_BEGIN
110 td[0] = *((int*)ucx_map_cstr_get(map, "Key0"));
111 td[1] = *((int*)ucx_map_cstr_get(map, "Key1"));
112 td[2] = *((int*)ucx_map_cstr_get(map, "Key2"));
113 td[3] = *((int*)ucx_map_cstr_get(map, "KeY3"));
114 td[4] = *((int*)ucx_map_cstr_get(map, "KEY4"));
115 UCX_TEST_ASSERT(td[0] == 10, "failed key 0");
116 UCX_TEST_ASSERT(td[1] == 42, "failed key 1");
117 UCX_TEST_ASSERT(td[2] == 70, "failed key 2");
118 UCX_TEST_ASSERT(td[3] == 11200, "failed key 3");
119 UCX_TEST_ASSERT(td[4] == 80000, "failed key 4");
121 UCX_TEST_ASSERT(map->count == 5, "expected 5 remaining values");
122 UCX_TEST_ASSERT(ucx_map_cstr_get(map, "Key0") != NULL, "element removed");
124 UCX_TEST_END
125 ucx_map_free(map);
126 }
128 UCX_TEST_IMPLEMENT(test_ucx_map_remove) {
129 UcxMap *map = ucx_map_new(4);
131 int td[5];
132 td[0] = 10; td[1] = 42; td[2] = 70; td[3] = 11200; td[4] = 80000;
134 ucx_map_cstr_put(map, "Key2", &td[2]); /* 0 */
135 ucx_map_cstr_put(map, "Key0", &td[0]); /* 0 */
136 ucx_map_cstr_put(map, "Key1", &td[1]); /* 3 */
137 ucx_map_cstr_put(map, "KeY3", &td[3]); /* 2 */
138 ucx_map_cstr_put(map, "KEY4", &td[4]); /* 0 */
139 UCX_TEST_BEGIN
141 td[0] = *((int*)ucx_map_cstr_remove(map, "Key0"));
142 td[1] = *((int*)ucx_map_cstr_get(map, "Key1"));
143 td[2] = *((int*)ucx_map_cstr_remove(map, "Key2"));
144 td[3] = *((int*)ucx_map_cstr_get(map, "KeY3"));
145 td[4] = *((int*)ucx_map_cstr_get(map, "KEY4"));
146 UCX_TEST_ASSERT(td[0] == 10, "failed key 0");
147 UCX_TEST_ASSERT(td[1] == 42, "failed key 1");
148 UCX_TEST_ASSERT(td[2] == 70, "failed key 2");
149 UCX_TEST_ASSERT(td[3] == 11200, "failed key 3");
150 UCX_TEST_ASSERT(td[4] == 80000, "failed key 4");
152 UCX_TEST_ASSERT(map->count == 3, "expected 3 remaining values");
153 UCX_TEST_ASSERT(ucx_map_cstr_get(map, "Key0")==NULL, "element not removed");
154 UCX_TEST_ASSERT(ucx_map_cstr_get(map, "Key1")!=NULL, "element removed");
155 UCX_TEST_ASSERT(ucx_map_cstr_get(map, "Key2")==NULL, "element not removed");
156 UCX_TEST_ASSERT(ucx_map_cstr_get(map, "KeY3")!=NULL, "element removed");
157 UCX_TEST_ASSERT(ucx_map_cstr_get(map, "KEY4")!=NULL, "element removed");
159 UCX_TEST_ASSERT(ucx_map_cstr_remove(map, "Key2") == NULL,
160 "subsequent remove call shall return NULL");
162 UCX_TEST_END
163 ucx_map_free(map);
164 }
166 UCX_TEST_SUBROUTINE(test_ucx_map_itersrt, UcxMap *map) {
167 int v1 = 10;
168 int v2 = 15;
169 int v3 = 7;
170 int v4 = 9;
172 ucx_map_cstr_put(map, "v1", &v1);
173 ucx_map_cstr_put(map, "v2", &v2);
174 ucx_map_cstr_put(map, "v3", &v3);
175 ucx_map_cstr_put(map, "v4", &v4);
177 UcxMapIterator i = ucx_map_iterator(map);
178 int check = 0;
179 int hit = 0;
181 int* v;
182 UCX_MAP_FOREACH(key, v, i) {
183 check += *v;
184 hit++;
185 }
187 UCX_TEST_ASSERT(hit == 4, "test1: wrong number of hits");
188 UCX_TEST_ASSERT(check == v1+v2+v3+v4, "test1: wrong result");
189 }
191 UCX_TEST_IMPLEMENT(test_ucx_map_iterator) {
192 UcxMap *map = ucx_map_new(16);
193 UCX_TEST_BEGIN
194 UCX_TEST_CALL_SUBROUTINE(test_ucx_map_itersrt, map)
195 UCX_TEST_END
196 ucx_map_free(map);
197 }
199 UCX_TEST_IMPLEMENT(test_ucx_map_iterator_chain) {
200 UcxMap *map = ucx_map_new(1);
201 UCX_TEST_BEGIN
202 UCX_TEST_CALL_SUBROUTINE(test_ucx_map_itersrt, map)
203 UCX_TEST_END
204 ucx_map_free(map);
205 }
207 void* test_ucx_map_store_load_encdec(void *value, void *data, size_t *size) {
208 const char *string = (const char*) value;
209 size_t n = strlen(string);
210 char *encoded = (char*) malloc(n+1);
211 for (size_t i = 0 ; i < n ; i++) {
212 encoded[i] = string[n-1-i];
213 }
214 encoded[n] = 0;
215 *size = n+1;
216 return encoded;
217 }
219 UCX_TEST_IMPLEMENT(test_ucx_map_store_load) {
220 UcxMap *map = ucx_map_new(4);
222 ucx_map_cstr_put(map, "test", (void*)"test");
223 ucx_map_cstr_put(map, "key", (void*)"value");
224 ucx_map_cstr_put(map, "other.very.long.key", (void*)"value");
225 ucx_map_cstr_put(map, "testkey", (void*)"testvalue");
226 ucx_map_cstr_put(map, "simple", (void*)"not a key but an extremely long "
227 "value to test if the buffer extension works as designed");
229 UCX_TEST_BEGIN
230 FILE *f = tmpfile();
231 UCX_TEST_ASSERT(f, "test file cannot be opened, test aborted")
232 int r;
234 fwrite(" # comment test\n", 1, 16, f);
235 r = ucx_map_store_enc(map, f, test_ucx_map_store_load_encdec, NULL);
236 fwrite("!discard this", 1, 13, f);
237 fflush(f);
239 ucx_map_free(map);
240 map = ucx_map_new(1);
241 fseek(f, 0, SEEK_SET);
242 UcxAllocator allocator = UCX_ALLOCATOR_DEFAULT;
243 r += ucx_map_load_enc(map, f, allocator,
244 test_ucx_map_store_load_encdec, NULL);
245 fclose(f);
247 const char *value;
248 UCX_TEST_ASSERT(r == 0, "IO errors, test cannot be performed");
250 value = (const char *) ucx_map_cstr_get(map, "test");
251 UCX_TEST_ASSERT(value != NULL, "value not found for key: test");
252 UCX_TEST_ASSERT(strcmp(value, "test") == 0, "value error for key: test");
254 value = (const char *) ucx_map_cstr_get(map, "key");
255 UCX_TEST_ASSERT(value != NULL, "value not found for key: key");
256 UCX_TEST_ASSERT(strcmp(value, "value") == 0, "value error for key: key");
258 value = (const char *) ucx_map_cstr_get(map, "other.very.long.key");
259 UCX_TEST_ASSERT(value != NULL,
260 "value not found for key: other.very.long.key");
261 UCX_TEST_ASSERT(strcmp(value, "value") == 0,
262 "value error for key: other.very.long.key");
264 value = (const char *) ucx_map_cstr_get(map, "testkey");
265 UCX_TEST_ASSERT(value != NULL, "value not found for key: testkey");
266 UCX_TEST_ASSERT(strcmp(value, "testvalue") == 0,
267 "value error for key: testkey");
269 value = (const char *) ucx_map_cstr_get(map, "simple");
270 UCX_TEST_ASSERT(value != NULL, "value not found for key: simple");
271 UCX_TEST_ASSERT(strcmp(value, "not a key but an extremely long value "
272 "to test if the buffer extension works as designed") == 0,
273 "value error for key: simple");
275 void *d;
276 UcxMapIterator iter = ucx_map_iterator(map);
277 UCX_MAP_FOREACH(key, d, iter) {
278 free(d);
279 }
280 ucx_map_free(map);
281 UCX_TEST_END
282 }
284 UCX_TEST_IMPLEMENT(test_ucx_map_store_load_with_mempool) {
285 UcxMap *map = ucx_map_new(4);
287 ucx_map_cstr_put(map, "test", (void*)"test");
288 ucx_map_cstr_put(map, "key", (void*)"value");
289 ucx_map_cstr_put(map, "testkey", (void*)"testvalue");
290 ucx_map_cstr_put(map, "simple", (void*)"a simple value");
292 UCX_TEST_BEGIN
293 FILE *f = tmpfile();
294 UCX_TEST_ASSERT(f, "test file cannot be opened, test aborted");
295 int r;
296 r = ucx_map_store_enc(map, f, NULL, NULL);
297 ucx_map_free(map);
298 fflush(f);
300 UcxMempool *pool = ucx_mempool_new(4);
301 map = ucx_map_new(4);
302 fseek(f, 0, SEEK_SET);
303 UcxAllocator allocator = UCX_ALLOCATOR_MEMPOOL(pool);
304 r += ucx_map_load_enc(map, f, allocator,
305 test_ucx_map_store_load_encdec, NULL);
306 fclose(f);
308 UCX_TEST_ASSERT(r == 0, "IO errors, test cannot be performed");
309 UcxMapIterator iter = ucx_map_iterator(map);
310 const char *value; size_t n;
311 UCX_MAP_FOREACH(key, value, iter) {
312 n = strlen(value);
313 UCX_TEST_ASSERT(strncmp((const char*) pool->data[iter.index], value, n),
314 "values of map does not match pooled values");
315 }
317 ucx_mempool_free(pool);
318 ucx_map_free(map);
319 UCX_TEST_END
320 }
322 UCX_TEST_IMPLEMENT(test_ucx_map_clone) {
323 UcxMap *map = ucx_map_new(4);
325 ucx_map_cstr_put(map, "key1", (void*)"value1");
326 ucx_map_cstr_put(map, "key2", (void*)"value2");
327 ucx_map_cstr_put(map, "key3", (void*)"value3");
329 UcxMap *clone = ucx_map_clone(map, NULL, NULL);
331 const char *v1 = (const char *) ucx_map_cstr_get(map, "key1");
332 const char *v2 = (const char *) ucx_map_cstr_get(map, "key2");
333 const char *v3 = (const char *) ucx_map_cstr_get(map, "key3");
335 UCX_TEST_BEGIN
337 UCX_TEST_ASSERT(v1 != NULL, "failed key 1");
338 UCX_TEST_ASSERT(v2 != NULL, "failed key 2");
339 UCX_TEST_ASSERT(v3 != NULL, "failed key 3");
341 const char *c1 = (const char *) ucx_map_cstr_get(clone, "key1");
342 const char *c2 = (const char *) ucx_map_cstr_get(clone, "key2");
343 const char *c3 = (const char *) ucx_map_cstr_get(clone, "key3");
345 UCX_TEST_ASSERT(c1 != NULL, "failed key 1 (clone)");
346 UCX_TEST_ASSERT(c2 != NULL, "failed key 2 (clone)");
347 UCX_TEST_ASSERT(c3 != NULL, "failed key 3 (clone)");
349 UCX_TEST_ASSERT(strcmp(c1, v1) == 0, "value error for key1");
350 UCX_TEST_ASSERT(strcmp(c2, v2) == 0, "value error for key2");
351 UCX_TEST_ASSERT(strcmp(c3, v3) == 0, "value error for key3");
353 UCX_TEST_END
355 ucx_map_free(map);
356 ucx_map_free(clone);
357 }
359 UCX_TEST_IMPLEMENT(test_ucx_map_rehash) {
360 UcxMap *map = ucx_map_new(4);
362 char keys[10][5];
363 char values[10][7];
364 for (int i = 0 ; i < 10 ; i++) {
365 strcpy(keys[i], "key");
366 keys[i][3] = 48+i; keys[i][4] = 0;
367 strcpy(values[i], "value");
368 values[i][5] = 48+i; values[i][6] = 0;
370 ucx_map_cstr_put(map, keys[i], values[i]);
371 }
373 ucx_map_rehash(map);
375 UCX_TEST_BEGIN
376 UCX_TEST_ASSERT(map->size == 25, "new capacity shall be 2.5 * count");
377 UCX_TEST_ASSERT(map->count == 10, "new map element count incorrect");
378 for (int i = 0 ; i < 10 ; i++) {
379 const char *value = (const char *) ucx_map_cstr_get(map, keys[i]);
380 UCX_TEST_ASSERT(value != NULL, "new map is missing old keys");
381 UCX_TEST_ASSERT(strncmp(value, values[i], 6) == 0,
382 "new map contains incorrect values");
383 }
384 ucx_map_rehash(map);
385 UCX_TEST_ASSERT(map->size == 25,
386 "subsequent rehashing call shall not change size");
387 UCX_TEST_END
389 ucx_map_free(map);
390 }