maple_tree: add some alloc node test case

Add some maple_tree alloc node tese case.

Link: https://lkml.kernel.org/r/20240626160631.3636515-2-Liam.Howlett@oracle.com
Signed-off-by: Jiazi Li <jqqlijiazi@gmail.com>
Signed-off-by: Liam R. Howlett <Liam.Howlett@oracle.com>
Suggested-by: Liam R. Howlett <Liam.Howlett@oracle.com>
Cc: Wei Yang <richard.weiyang@gmail.com>
Signed-off-by: Andrew Morton <akpm@linux-foundation.org>

diff --git a/tools/testing/radix-tree/maple.c b/tools/testing/radix-tree/maple.c
index 551ae6898c1d2dd7d05c446e5956f7e168ba1c9c..bc30050227fdaf251506ad5df31dada3c0488468 100644 (file)
--- a/tools/testing/radix-tree/maple.c
+++ b/tools/testing/radix-tree/maple.c
@@ -462,6 +462,28 @@ static noinline void __init check_new_node(struct maple_tree *mt)
        MT_BUG_ON(mt, mas_allocated(&mas) != 10 + MAPLE_ALLOC_SLOTS - 1);
        mas_destroy(&mas);
 
+       mas.node = MA_ERROR(-ENOMEM);
+       mas_node_count(&mas, MAPLE_ALLOC_SLOTS + 1); /* Request */
+       mas_nomem(&mas, GFP_KERNEL); /* Fill request */
+       MT_BUG_ON(mt, mas_allocated(&mas) != MAPLE_ALLOC_SLOTS + 1);
+       mas.node = MA_ERROR(-ENOMEM);
+       mas_node_count(&mas, MAPLE_ALLOC_SLOTS * 2 + 2); /* Request */
+       mas_nomem(&mas, GFP_KERNEL); /* Fill request */
+       mas.status = ma_start;
+       MT_BUG_ON(mt, mas_allocated(&mas) != MAPLE_ALLOC_SLOTS * 2 + 2);
+       mas_destroy(&mas);
+
+       mas.node = MA_ERROR(-ENOMEM);
+       mas_node_count(&mas, MAPLE_ALLOC_SLOTS * 2 + 1); /* Request */
+       mas_nomem(&mas, GFP_KERNEL); /* Fill request */
+       MT_BUG_ON(mt, mas_allocated(&mas) != MAPLE_ALLOC_SLOTS * 2 + 1);
+       mas.node = MA_ERROR(-ENOMEM);
+       mas_node_count(&mas, MAPLE_ALLOC_SLOTS * 3 + 2); /* Request */
+       mas_nomem(&mas, GFP_KERNEL); /* Fill request */
+       mas.status = ma_start;
+       MT_BUG_ON(mt, mas_allocated(&mas) != MAPLE_ALLOC_SLOTS * 3 + 2);
+       mas_destroy(&mas);
+
        mtree_unlock(mt);
 }