michaelh/zephyr
1
0
Fork 0
Code Releases Activity

tests: lib/mem_blocks: add tests for multi mem_blocks allocator

Browse source 
This adds a few tests to make sure the multi memory blocks
allocator functions work correctly.

Signed-off-by: Daniel Leung <daniel.leung@intel.com>
This commit is contained in:
Daniel Leung 2021-11-24 20:26:34 -08:00 • committed by Anas Nashif
commit 7f06fb4129
1 changed files with 121 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

122
tests/lib/mem_blocks/src/main.c
View file

@ -20,6 +20,20 @@ SYS_MEM_BLOCKS_DEFINE_STATIC_WITH_EXT_BUF(mem_block_02,
BLK_SZ, NUM_BLOCKS,
mem_block_02_buf);
static sys_multi_mem_blocks_t alloc_group;
sys_mem_blocks_t *choice_fn(struct sys_multi_mem_blocks *group, void *cfg)
{
/* mem_block_"01" or mem_block_"02" */
uintptr_t num = POINTER_TO_UINT(cfg) - 1;
if (num >= group->num_allocators) {
return NULL;
} else {
return group->allocators[num];
}
}
static bool check_buffer_bound(sys_mem_blocks_t *mem_block, void *ptr)
{
uint8_t *start, *end, *ptr_u8;
@ -97,6 +111,44 @@ static void test_mem_block_multi_alloc_free_alt_buf(void)
alloc_free(&mem_block_02, NUM_BLOCKS, 10);
}
static void test_multi_mem_block_alloc_free(void)
{
int ret;
void *blocks[2][1] = {0};
size_t blk_size;
ret = sys_multi_mem_blocks_alloc(&alloc_group, UINT_TO_POINTER(16),
1, blocks[0], &blk_size);
zassert_equal(ret, -EINVAL,
"sys_multi_mem_blocks_alloc should fail with -EINVAL but not");
ret = sys_multi_mem_blocks_alloc(&alloc_group, UINT_TO_POINTER(1),
1, blocks[0], &blk_size);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_alloc failed (%d)", ret);
zassert_true(check_buffer_bound(&mem_block_01, blocks[0][0]),
"allocated memory is out of bound");
zassert_equal(blk_size, BLK_SZ,
"returned block size is not %d", BLK_SZ);
ret = sys_multi_mem_blocks_alloc(&alloc_group, UINT_TO_POINTER(2),
1, blocks[1], &blk_size);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_alloc failed (%d)", ret);
zassert_true(check_buffer_bound(&mem_block_02, blocks[1][0]),
"allocated memory is out of bound");
zassert_equal(blk_size, BLK_SZ,
"returned block size is not %d", BLK_SZ);
ret = sys_multi_mem_blocks_free(&alloc_group, 1, blocks[0]);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_free failed (%d)", ret);
ret = sys_multi_mem_blocks_free(&alloc_group, 1, blocks[1]);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_free failed (%d)", ret);
}
static void test_mem_block_invalid_params(void)
{
int ret;
@ -154,14 +206,82 @@ static void test_mem_block_invalid_params(void)
"sys_mem_blocks_free should fail with -EFAULT but not");
}
static void test_multi_mem_block_invalid_params(void)
{
int ret;
void *blocks[2] = {0};
ret = sys_multi_mem_blocks_alloc(NULL, UINT_TO_POINTER(16),
1, blocks, NULL);
zassert_equal(ret, -EINVAL,
"sys_multi_mem_blocks_alloc should fail with -EINVAL but not");
ret = sys_multi_mem_blocks_alloc(&alloc_group, UINT_TO_POINTER(16),
1, NULL, NULL);
zassert_equal(ret, -EINVAL,
"sys_multi_mem_blocks_alloc should fail with -EINVAL but not");
ret = sys_multi_mem_blocks_alloc(&alloc_group, UINT_TO_POINTER(16),
0, blocks, NULL);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_alloc failed (%d)", ret);
ret = sys_multi_mem_blocks_alloc(&alloc_group, UINT_TO_POINTER(1),
NUM_BLOCKS + 1, blocks, NULL);
zassert_equal(ret, -ENOMEM,
"sys_multi_mem_blocks_alloc should fail with -ENOMEM but not");
ret = sys_multi_mem_blocks_alloc(&alloc_group, UINT_TO_POINTER(1),
1, blocks, NULL);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_alloc failed (%d)", ret);
ret = sys_multi_mem_blocks_free(NULL, 1, blocks);
zassert_equal(ret, -EINVAL,
"sys_multi_mem_blocks_free should fail with -EINVAL but not");
ret = sys_multi_mem_blocks_free(&alloc_group, 1, NULL);
zassert_equal(ret, -EINVAL,
"sys_multi_mem_blocks_free should fail with -EINVAL but not");
ret = sys_multi_mem_blocks_free(&alloc_group, 0, blocks);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_free failed (%d)", ret);
ret = sys_multi_mem_blocks_free(&alloc_group, NUM_BLOCKS + 1, blocks);
zassert_equal(ret, -EINVAL,
"sys_multi_mem_blocks_free should fail with -EINVAL but not");
ret = sys_multi_mem_blocks_free(&alloc_group, 1, blocks);
zassert_equal(ret, 0,
"sys_multi_mem_blocks_free failed (%d)", ret);
ret = sys_multi_mem_blocks_free(&alloc_group, 1, blocks);
zassert_equal(ret, -EFAULT,
"sys_multi_mem_blocks_free should fail with -EFAULT but not");
/* Fake a pointer */
blocks[0] = mem_block_01.buffer +
(BIT(mem_block_01.blk_sz_shift) * mem_block_01.num_blocks);
ret = sys_multi_mem_blocks_free(&alloc_group, 1, blocks);
zassert_equal(ret, -EINVAL,
"sys_multi_mem_blocks_free should fail with -EINVAL but not");
}
void test_main(void)
{
sys_multi_mem_blocks_init(&alloc_group, choice_fn);
sys_multi_mem_blocks_add_allocator(&alloc_group, &mem_block_01);
sys_multi_mem_blocks_add_allocator(&alloc_group, &mem_block_02);
ztest_test_suite(lib_mem_block_test,
ztest_unit_test(test_mem_block_alloc_free),
ztest_unit_test(test_mem_block_alloc_free_alt_buf),
ztest_unit_test(test_mem_block_multi_alloc_free),
ztest_unit_test(test_mem_block_multi_alloc_free_alt_buf),
ztest_unit_test(test_mem_block_invalid_params)
ztest_unit_test(test_multi_mem_block_alloc_free),
ztest_unit_test(test_mem_block_invalid_params),
ztest_unit_test(test_multi_mem_block_invalid_params)
);
ztest_run_test_suite(lib_mem_block_test);