tests: kernel: common: Extend bitarray test

Add test for sys_bitarray_alloc and sys_bitarray_free with 32 bit
bitarray which is using an optimized algorithm.

Signed-off-by: Krzysztof Chruściński <[email protected]>

Author: nordic-krch

tests/kernel/common/src/bitarray.c

@@ -282,6 +282,39 @@ ZTEST(bitarray, test_bitarray_set_clear)
 		     "sys_bitarray_test_and_clear_bit() erroneously changed bitarray");
 }
 
+static void test_alloc_free_32(uint32_t mask, uint32_t mask_after, size_t num_bits,
+			       size_t exp_offset, int exp_ret)
+{
+	int ret;
+	size_t offset;
+
+	SYS_BITARRAY_DEFINE(ba_32, 32);
+
+	ba_32.bundles[0] = mask;
+
+	ret = sys_bitarray_alloc(&ba_32, num_bits, &offset);
+	zassert_equal(ret, exp_ret, "sys_bitarray_alloc() failed: %d", ret);
+	if (exp_ret < 0) {
+		return;
+	}
+	zassert_equal(offset, exp_offset,
+		      "sys_bitarray_alloc() offset expected %d, got %d", exp_offset, offset);
+	zassert_equal(ba_32.bundles[0], mask_after, "sys_bitarray_alloc() failed bits comparison");
+
+	ret = sys_bitarray_free(&ba_32, num_bits, offset);
+	zassert_equal(ret, 0, "sys_bitarray_free() failed: %d", ret);
+	zassert_equal(ba_32.bundles[0], mask, "sys_bitarray_alloc() failed bits comparison");
+}
+
+static void alloc_and_free_32_predefined(void)
+{
+	printk("Testing bit array alloc and free with predefined patterns using 32 bit array\n");
+
+	test_alloc_free_32(0x0F0F070F, 0x0F0FFF0F, 5, 11, 0);
+	test_alloc_free_32(0x33333333, 0xF3333333, 3, 0, -ENOSPC);
+	test_alloc_free_32(0x33333333, 0xF3333333, 2, 30, 0);
+}
+
 void alloc_and_free_predefined(void)
 {
 	int ret;
@@ -291,7 +324,7 @@ void alloc_and_free_predefined(void)
 
 	SYS_BITARRAY_DEFINE(ba_128, 128);
 
-	printk("Testing bit array alloc and free with predefined patterns\n");
+	printk("Testing bit array alloc and free with predefined patterns using 128 bit array\n");
 
 	/* Pre-populate the bits */
 	ba_128.bundles[0] = 0x0F0F070F;
@@ -514,6 +547,7 @@ ZTEST(bitarray, test_bitarray_alloc_free)
 	}
 
 	alloc_and_free_predefined();
+	alloc_and_free_32_predefined();
 
 	i = 1;
 	while (i < 65) {