Annotation of embedaddon/strongswan/src/libstrongswan/tests/suites/test_hashtable.c, revision 1.1
1.1 ! misho 1: /*
! 2: * Copyright (C) 2010-2013 Tobias Brunner
! 3: * HSR Hochschule fuer Technik Rapperswil
! 4: *
! 5: * This program is free software; you can redistribute it and/or modify it
! 6: * under the terms of the GNU General Public License as published by the
! 7: * Free Software Foundation; either version 2 of the License, or (at your
! 8: * option) any later version. See <http://www.fsf.org/copyleft/gpl.txt>.
! 9: *
! 10: * This program is distributed in the hope that it will be useful, but
! 11: * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
! 12: * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
! 13: * for more details.
! 14: */
! 15:
! 16: #include "test_suite.h"
! 17:
! 18: #include <collections/hashtable.h>
! 19: #include <utils/chunk.h>
! 20:
! 21: /*******************************************************************************
! 22: * string hash table functions
! 23: */
! 24:
! 25: static u_int hash(char *key)
! 26: {
! 27: return chunk_hash(chunk_from_str(key));
! 28: }
! 29:
! 30: static bool equals(char *key1, char *key2)
! 31: {
! 32: return streq(key1, key2);
! 33: }
! 34:
! 35: /*******************************************************************************
! 36: * test fixture
! 37: */
! 38:
! 39: static hashtable_t *ht;
! 40:
! 41: START_SETUP(setup_ht)
! 42: {
! 43: ht = hashtable_create((hashtable_hash_t)hash,
! 44: (hashtable_equals_t)equals, 0);
! 45: ck_assert_int_eq(ht->get_count(ht), 0);
! 46: }
! 47: END_SETUP
! 48:
! 49: START_TEARDOWN(teardown_ht)
! 50: {
! 51: ht->destroy(ht);
! 52: }
! 53: END_TEARDOWN
! 54:
! 55: /*******************************************************************************
! 56: * put/get
! 57: */
! 58:
! 59: START_TEST(test_put_get)
! 60: {
! 61: char *k1 = "key1", *k2 = "key2", *k3 = "key3";
! 62: char *v1 = "val1", *v2 = "val2", *v3 = "val3", *value;
! 63:
! 64: value = ht->put(ht, k1, v1);
! 65: ck_assert_int_eq(ht->get_count(ht), 1);
! 66: ck_assert(streq(ht->get(ht, k1), v1));
! 67: ck_assert(ht->get(ht, k2) == NULL);
! 68: ck_assert(ht->get(ht, k3) == NULL);
! 69: ck_assert(value == NULL);
! 70:
! 71: ht->put(ht, k2, v2);
! 72: ht->put(ht, k3, v3);
! 73: ck_assert_int_eq(ht->get_count(ht), 3);
! 74: ck_assert(streq(ht->get(ht, k1), v1));
! 75: ck_assert(streq(ht->get(ht, k2), v2));
! 76: ck_assert(streq(ht->get(ht, k3), v3));
! 77:
! 78: value = ht->put(ht, k2, v1);
! 79: ck_assert_int_eq(ht->get_count(ht), 3);
! 80: ck_assert(streq(value, v2));
! 81: ck_assert(streq(ht->get(ht, k2), v1));
! 82: }
! 83: END_TEST
! 84:
! 85: /*******************************************************************************
! 86: * get_match
! 87: */
! 88:
! 89: static u_int hash_match(char *key)
! 90: {
! 91: return chunk_hash(chunk_create(key, 4));
! 92: }
! 93:
! 94: static bool equal_match(char *key1, char *key2)
! 95: {
! 96: if (!strneq(key1, key2, 4))
! 97: {
! 98: return FALSE;
! 99: }
! 100: /* look for an item with a key < than what we look for */
! 101: return strcmp(key1, key2) >= 0;
! 102: }
! 103:
! 104: START_TEST(test_get_match)
! 105: {
! 106: char *k1 = "key1_a", *k2 = "key2", *k3 = "key1_b", *k4 = "key1_c";
! 107: char *v1 = "val1", *v2 = "val2", *v3 = "val3", *value;
! 108:
! 109: ht = hashtable_create((hashtable_hash_t)hash_match,
! 110: (hashtable_equals_t)equals, 0);
! 111:
! 112: ht->put(ht, k1, v1);
! 113: ht->put(ht, k2, v2);
! 114: value = ht->put(ht, k3, v3);
! 115: ck_assert_int_eq(ht->get_count(ht), 3);
! 116: ck_assert(streq(ht->get(ht, k1), v1));
! 117: ck_assert(streq(ht->get(ht, k2), v2));
! 118: ck_assert(streq(ht->get(ht, k3), v3));
! 119: ck_assert(value == NULL);
! 120:
! 121: value = ht->get_match(ht, k1, (hashtable_equals_t)equal_match);
! 122: ck_assert(value != NULL);
! 123: ck_assert(streq(value, v1));
! 124: value = ht->get_match(ht, k2, (hashtable_equals_t)equal_match);
! 125: ck_assert(value != NULL);
! 126: ck_assert(streq(value, v2));
! 127: value = ht->get_match(ht, k3, (hashtable_equals_t)equal_match);
! 128: ck_assert(value != NULL);
! 129: ck_assert(streq(value, v1));
! 130: value = ht->get_match(ht, k4, (hashtable_equals_t)equal_match);
! 131: ck_assert(value != NULL);
! 132: ck_assert(streq(value, v1));
! 133:
! 134: ht->destroy(ht);
! 135: }
! 136: END_TEST
! 137:
! 138: /*******************************************************************************
! 139: * remove
! 140: */
! 141:
! 142: static void do_remove(char *k1, char *k2, char *k3)
! 143: {
! 144: char *v1 = "val1", *v2 = "val2", *v3 = "val3", *value;
! 145:
! 146: ht->put(ht, k1, v1);
! 147: ht->put(ht, k2, v2);
! 148: ht->put(ht, k3, v3);
! 149:
! 150: value = ht->remove(ht, k2);
! 151: ck_assert_int_eq(ht->get_count(ht), 2);
! 152: ck_assert(streq(ht->get(ht, k1), v1));
! 153: ck_assert(streq(ht->get(ht, k3), v3));
! 154: ck_assert(streq(value, v2));
! 155: ck_assert(ht->get(ht, k2) == NULL);
! 156:
! 157: value = ht->remove(ht, k2);
! 158: ck_assert_int_eq(ht->get_count(ht), 2);
! 159: ck_assert(value == NULL);
! 160:
! 161: value = ht->remove(ht, k1);
! 162: value = ht->remove(ht, k3);
! 163: ck_assert_int_eq(ht->get_count(ht), 0);
! 164: ck_assert(ht->get(ht, k1) == NULL);
! 165: ck_assert(ht->get(ht, k2) == NULL);
! 166: ck_assert(ht->get(ht, k3) == NULL);
! 167: }
! 168:
! 169: START_TEST(test_remove)
! 170: {
! 171: char *k1 = "key1", *k2 = "key2", *k3 = "key3";
! 172:
! 173: do_remove(k1, k2, k3);
! 174: }
! 175: END_TEST
! 176:
! 177: START_TEST(test_remove_one_bucket)
! 178: {
! 179: char *k1 = "key1_a", *k2 = "key1_b", *k3 = "key1_c";
! 180:
! 181: ht->destroy(ht);
! 182: /* set a capacity to avoid rehashing, which would change the items' order */
! 183: ht = hashtable_create((hashtable_hash_t)hash_match,
! 184: (hashtable_equals_t)equals, 8);
! 185:
! 186: do_remove(k1, k2, k3);
! 187: }
! 188: END_TEST
! 189:
! 190: /*******************************************************************************
! 191: * enumerator
! 192: */
! 193:
! 194: START_TEST(test_enumerator)
! 195: {
! 196: char *k1 = "key1", *k2 = "key2", *k3 = "key3", *key;
! 197: char *v1 = "val1", *v2 = "val2", *v3 = "val3", *value;
! 198: enumerator_t *enumerator;
! 199: int count;
! 200:
! 201: ht->put(ht, k1, v1);
! 202: ht->put(ht, k2, v2);
! 203: ht->put(ht, k3, v3);
! 204:
! 205: count = 0;
! 206: enumerator = ht->create_enumerator(ht);
! 207: while (enumerator->enumerate(enumerator, &key, &value))
! 208: {
! 209: ck_assert(streq(key, k1) || streq(key, k2) || streq(key, k3));
! 210: ck_assert(streq(value, v1) || streq(value, v2) || streq(value, v3));
! 211: ck_assert(!streq(key, k1) || streq(value, v1));
! 212: ck_assert(!streq(key, k2) || streq(value, v2));
! 213: ck_assert(!streq(key, k3) || streq(value, v3));
! 214: count++;
! 215: }
! 216: enumerator->destroy(enumerator);
! 217: ck_assert_int_eq(count, 3);
! 218:
! 219: count = 0;
! 220: enumerator = ht->create_enumerator(ht);
! 221: while (enumerator->enumerate(enumerator, NULL, NULL))
! 222: {
! 223: count++;
! 224: }
! 225: enumerator->destroy(enumerator);
! 226: ck_assert_int_eq(count, 3);
! 227:
! 228: value = ht->remove(ht, k1);
! 229: value = ht->remove(ht, k2);
! 230: value = ht->remove(ht, k3);
! 231:
! 232: count = 0;
! 233: enumerator = ht->create_enumerator(ht);
! 234: while (enumerator->enumerate(enumerator, &key, &value))
! 235: {
! 236: count++;
! 237: }
! 238: enumerator->destroy(enumerator);
! 239: ck_assert_int_eq(count, 0);
! 240: }
! 241: END_TEST
! 242:
! 243: /*******************************************************************************
! 244: * remove_at
! 245: */
! 246:
! 247: static void do_remove_at(char *k1, char *k2, char *k3)
! 248: {
! 249: char *v1 = "val1", *v2 = "val2", *v3 = "val3", *value, *key;
! 250: enumerator_t *enumerator;
! 251:
! 252: ht->put(ht, k1, v1);
! 253: ht->put(ht, k2, v2);
! 254: ht->put(ht, k3, v3);
! 255:
! 256: enumerator = ht->create_enumerator(ht);
! 257: ht->remove_at(ht, enumerator);
! 258: while (enumerator->enumerate(enumerator, &key, &value))
! 259: {
! 260: if (streq(key, k2))
! 261: {
! 262: ht->remove_at(ht, enumerator);
! 263: }
! 264: }
! 265: enumerator->destroy(enumerator);
! 266:
! 267: ck_assert_int_eq(ht->get_count(ht), 2);
! 268: ck_assert(ht->get(ht, k1) != NULL);
! 269: ck_assert(ht->get(ht, k3) != NULL);
! 270: ck_assert(ht->get(ht, k2) == NULL);
! 271:
! 272: ht->put(ht, k2, v2);
! 273:
! 274: ck_assert_int_eq(ht->get_count(ht), 3);
! 275: ck_assert(ht->get(ht, k1) != NULL);
! 276: ck_assert(ht->get(ht, k2) != NULL);
! 277: ck_assert(ht->get(ht, k3) != NULL);
! 278:
! 279: enumerator = ht->create_enumerator(ht);
! 280: while (enumerator->enumerate(enumerator, &key, &value))
! 281: {
! 282: ht->remove_at(ht, enumerator);
! 283: }
! 284: enumerator->destroy(enumerator);
! 285:
! 286: ck_assert_int_eq(ht->get_count(ht), 0);
! 287: ck_assert(ht->get(ht, k1) == NULL);
! 288: ck_assert(ht->get(ht, k2) == NULL);
! 289: ck_assert(ht->get(ht, k3) == NULL);
! 290: }
! 291:
! 292: START_TEST(test_remove_at)
! 293: {
! 294: char *k1 = "key1", *k2 = "key2", *k3 = "key3";
! 295:
! 296: do_remove_at(k1, k2, k3);
! 297: }
! 298: END_TEST
! 299:
! 300: START_TEST(test_remove_at_one_bucket)
! 301: {
! 302: char *k1 = "key1_a", *k2 = "key1_b", *k3 = "key1_c";
! 303:
! 304: ht->destroy(ht);
! 305: /* set a capacity to avoid rehashing, which would change the items' order */
! 306: ht = hashtable_create((hashtable_hash_t)hash_match,
! 307: (hashtable_equals_t)equals, 8);
! 308: do_remove_at(k1, k2, k3);
! 309: }
! 310: END_TEST
! 311:
! 312: Suite *hashtable_suite_create()
! 313: {
! 314: Suite *s;
! 315: TCase *tc;
! 316:
! 317: s = suite_create("hashtable");
! 318:
! 319: tc = tcase_create("put/get");
! 320: tcase_add_checked_fixture(tc, setup_ht, teardown_ht);
! 321: tcase_add_test(tc, test_put_get);
! 322: suite_add_tcase(s, tc);
! 323:
! 324: tc = tcase_create("get_match");
! 325: tcase_add_test(tc, test_get_match);
! 326: suite_add_tcase(s, tc);
! 327:
! 328: tc = tcase_create("remove");
! 329: tcase_add_checked_fixture(tc, setup_ht, teardown_ht);
! 330: tcase_add_test(tc, test_remove);
! 331: tcase_add_test(tc, test_remove_one_bucket);
! 332: suite_add_tcase(s, tc);
! 333:
! 334: tc = tcase_create("enumerator");
! 335: tcase_add_checked_fixture(tc, setup_ht, teardown_ht);
! 336: tcase_add_test(tc, test_enumerator);
! 337: suite_add_tcase(s, tc);
! 338:
! 339: tc = tcase_create("remove_at");
! 340: tcase_add_checked_fixture(tc, setup_ht, teardown_ht);
! 341: tcase_add_test(tc, test_remove_at);
! 342: tcase_add_test(tc, test_remove_at_one_bucket);
! 343: suite_add_tcase(s, tc);
! 344:
! 345: return s;
! 346: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>