[PATCH 7/9] unit: hashmap: Re-entrancy tests added

[Jukka Rissanen <jukka.rissanen@linux.intel.com>]

[-- Attachment #1: Type: text/plain, Size: 2464 bytes --]

Check that we can add and remove items from hash while inside
foreach callback.
---
 unit/test-hashmap.c | 81 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)

diff --git a/unit/test-hashmap.c b/unit/test-hashmap.c
index 75c249c..51d03ba 100644
--- a/unit/test-hashmap.c
+++ b/unit/test-hashmap.c
@@ -184,11 +184,92 @@ static void test_str(const void *test_data)
 	l_hashmap_destroy(hashmap, NULL);
 }
 
+static void foreach_cb(const void *key, void *value, void *user_data)
+{
+	struct l_hashmap *hashmap = user_data;
+	static bool removed, insert, conflict;
+	unsigned int entries;
+	char *str, *mykey = "b";
+	bool ret;
+
+	if (!removed) {
+		printf("Removing entry in foreach\n");
+
+		entries = l_hashmap_size(hashmap);
+
+		str = l_hashmap_remove(hashmap, "a");
+		assert(str);
+
+		assert(!l_hashmap_lookup(hashmap, "a"));
+
+		assert(entries == (l_hashmap_size(hashmap) + 1));
+
+		removed = true;
+	}
+
+	if (!insert) {
+		printf("Adding entry in foreach\n");
+
+		entries = l_hashmap_size(hashmap);
+
+		ret = l_hashmap_insert(hashmap, mykey, mykey);
+		assert(ret);
+
+		assert(entries == (l_hashmap_size(hashmap) - 1));
+
+		insert = true;
+	}
+
+	if (!conflict) {
+		printf("Adding conflicting entry in foreach\n");
+
+		entries = l_hashmap_size(hashmap);
+
+		assert(l_hashmap_lookup(hashmap, mykey));
+
+		ret = l_hashmap_insert(hashmap, mykey, mykey);
+		assert(ret);
+
+		assert(entries == l_hashmap_size(hashmap));
+
+		conflict = true;
+	}
+}
+
+static void test_reentrant(const void *test_data)
+{
+	struct l_hashmap *hashmap;
+	const void *ptr;
+	const char **itr, *strings[] = {
+		"hello",
+		"world",
+		"a",
+		"a longer key here",
+		NULL
+	};
+
+	hashmap = l_hashmap_string_new();
+	assert(hashmap);
+	assert(l_hashmap_size(hashmap) == 0);
+	assert(l_hashmap_isempty(hashmap));
+
+	for (itr = strings; *itr != NULL; itr++) {
+		assert(l_hashmap_insert(hashmap, *itr, itr));
+		ptr = l_hashmap_lookup(hashmap, *itr);
+		assert(ptr == itr);
+	}
+
+	l_hashmap_foreach(hashmap, foreach_cb, hashmap);
+
+	l_hashmap_destroy(hashmap, NULL);
+}
+
 int main(int argc, char *argv[])
 {
 	l_test_init(&argc, &argv);
 
 	l_test_add("Pointer Test", test_ptr, NULL);
 	l_test_add("String Test", test_str, NULL);
+	l_test_add("Re-entrant Test", test_reentrant, NULL);
 	return l_test_run();
 }
-- 
1.8.3.1