From d314963d0f1d12c45c55c7fd210f93c5cac3a8fc Mon Sep 17 00:00:00 2001 From: Florin Coras Date: Wed, 19 Jun 2019 16:45:09 -0700 Subject: vppinfra: fix rbtree node delete Type:fix Make sure tnil color is black and that the right node colors are updated. Change-Id: Ibd9d7ea9438df4dab977202955957824723a865d Signed-off-by: Florin Coras --- src/plugins/unittest/rbtree_test.c | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) (limited to 'src/plugins/unittest') diff --git a/src/plugins/unittest/rbtree_test.c b/src/plugins/unittest/rbtree_test.c index 490be9c156e..bfab98c3cd7 100644 --- a/src/plugins/unittest/rbtree_test.c +++ b/src/plugins/unittest/rbtree_test.c @@ -160,7 +160,11 @@ rbtree_test_basic (vlib_main_t * vm, unformat_input_t * input) * Delete all keys */ for (i = 0; i < n_keys; i++) - rb_tree_del (rt, i); + { + rb_tree_del (rt, i); + if (rt->nodes[RBTREE_TNIL_INDEX].color != RBTREE_BLACK) + RBTREE_TEST (0, "tnil should be black"); + } RBTREE_TEST (rb_tree_n_nodes (rt) == 1, "number nodes %u is %u", 1, rb_tree_n_nodes (rt)); -- cgit 1.2.3-korg