Commit e393c093 authored by Karolina Drobnik's avatar Karolina Drobnik Committed by Mike Rapoport
Browse files

memblock tests: Add memblock_add_node test 



Add a simple test for NUMA-aware variant of memblock_add function.

Signed-off-by: default avatarKarolina Drobnik <karolinadrobnik@gmail.com>
Signed-off-by: default avatarMike Rapoport <rppt@kernel.org>
Link: https://lore.kernel.org/r/e2d0e6dd264c8c169242b556f7c5b12153f3dee5.1643796665.git.karolinadrobnik@gmail.com
parent b4d96893

tools/testing/memblock/tests/basic_api.c

+34 −0
Original line number Diff line number Diff line
@@ -55,6 +55,39 @@ static int memblock_add_simple_check(void) 
	return 0;

}



/*

 * A simple test that adds a memory block of a specified base address, size

 * NUMA node and memory flags to the collection of available memory regions.

 * It checks if the new entry, region counter and total memory size have

 * expected values.

 */

static int memblock_add_node_simple_check(void)

{

	struct memblock_region *rgn;



	rgn = &memblock.memory.regions[0];



	struct region r = {

		.base = SZ_1M,

		.size = SZ_16M

	};



	reset_memblock();

	memblock_add_node(r.base, r.size, 1, MEMBLOCK_HOTPLUG);



	assert(rgn->base == r.base);

	assert(rgn->size == r.size);

#ifdef CONFIG_NUMA

	assert(rgn->nid == 1);

#endif

	assert(rgn->flags == MEMBLOCK_HOTPLUG);



	assert(memblock.memory.cnt == 1);

	assert(memblock.memory.total_size == r.size);



	return 0;

}



/*

 * A test that tries to add two memory blocks that don't overlap with one

 * another. It checks if two correctly initialized entries were added to the
@@ -230,6 +263,7 @@ static int memblock_add_twice_check(void) 
static int memblock_add_checks(void)

{

	memblock_add_simple_check();

	memblock_add_node_simple_check();

	memblock_add_disjoint_check();

	memblock_add_overlap_top_check();

	memblock_add_overlap_bottom_check();