memblock tests: Add memblock_add_node test

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

Signed-off-by: Karolina Drobnik <karolinadrobnik@gmail.com>
Signed-off-by: Mike Rapoport <rppt@kernel.org>
Link: https://lore.kernel.org/r/e2d0e6dd264c8c169242b556f7c5b12153f3dee5.1643796665.git.karolinadrobnik@gmail.com

diff --git a/tools/testing/memblock/tests/basic_api.c b/tools/testing/memblock/tests/basic_api.c
index 8a0d2ffa4e21d47574c4e6691518be584b330c3b..2194a141cfe64eab62d042c8e4d71d2592295b99 100644 (file)
--- a/tools/testing/memblock/tests/basic_api.c
+++ b/tools/testing/memblock/tests/basic_api.c
@@ -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();