core/bitmap: Test bitmap foreach functions

Signed-off-by: Stewart Smith <stewart@linux.vnet.ibm.com>

1 files changed, 11 insertions, 0 deletions

diff --git a/core/test/run-bitmap.c b/core/test/run-bitmap.c
index ceeb7d5..7c68460 100644
--- a/core/test/run-bitmap.c
+++ b/core/test/run-bitmap.c
@@ -22,6 +22,7 @@
 int main(void)
 {
 	bitmap_t *map = malloc(sizeof(bitmap_elem_t));
+	int i;
 	memset(map, 0, sizeof(bitmap_elem_t));
 
 	assert(BITMAP_ELEMS(16) == (BITMAP_ELEMS(8)));
@@ -73,6 +74,16 @@ int main(void)
 	assert(*(unsigned long*)map == 0x00);
 	assert(bitmap_tst_bit(*map, 8) == false);
 
+	bitmap_for_each_zero(*map, 7, i) {
+		bitmap_set_bit(*map, i);
+	}
+
+	for (i = 0; i < 7; i++)
+		assert(bitmap_tst_bit(*map, i) == true);
+
+	assert(bitmap_tst_bit(*map, 8) == false);
+
+
 	free(map);
 
 	return 0;