vppinfra: pool_free_elts() now supports fixed-size pools

Test added to the unittest plugin / test_vlib.py

Type: improvement
Signed-off-by: Dave Barach <dave@barachs.net>
Change-Id: I73445e57918347c102ff6f5e8c9ddb9bd96f1407
(cherry picked from commit 4de5f9be88857197ddf17e3bff66318f78f4b6bb)

2 files changed, 63 insertions, 0 deletions

diff --git a/src/plugins/unittest/CMakeLists.txt b/src/plugins/unittest/CMakeLists.txt
index 6276a92d749..115ced3393b 100644
--- a/src/plugins/unittest/CMakeLists.txt
+++ b/src/plugins/unittest/CMakeLists.txt
@@ -40,6 +40,7 @@ add_vpp_plugin(unittest
   mfib_test.c
   mpcap_node.c
   policer_test.c
+  pool_test.c
   punt_test.c
   rbtree_test.c
   session_test.c
diff --git a/src/plugins/unittest/pool_test.c b/src/plugins/unittest/pool_test.c
new file mode 100644
index 00000000000..237b6beea09
--- /dev/null
+++ b/src/plugins/unittest/pool_test.c
@@ -0,0 +1,62 @@
+/*
+ * Copyright (c) 2021 Dave Barach
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at:
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <vlib/vlib.h>
+
+static clib_error_t *
+test_pool_command_fn (vlib_main_t *vm, unformat_input_t *input,
+		      vlib_cli_command_t *cmd)
+{
+  int i;
+  u64 *pool;
+
+  pool_init_fixed (pool, 2048);
+
+  i = 0;
+
+  while (pool_free_elts (pool) > 0)
+    {
+      u64 *p __attribute__ ((unused));
+
+      pool_get (pool, p);
+      i++;
+    }
+
+  vlib_cli_output (vm, "allocated %d elts\n", i);
+
+  for (--i; i >= 0; i--)
+    {
+      pool_put_index (pool, i);
+    }
+
+  ALWAYS_ASSERT (pool_free_elts (pool) == 2048);
+
+  vlib_cli_output (vm, "Test succeeded...\n");
+  return 0;
+}
+
+VLIB_CLI_COMMAND (test_pool_command, static) = {
+  .path = "test pool",
+  .short_help = "vppinfra pool.h tests",
+  .function = test_pool_command_fn,
+};
+
+/*
+ * fd.io coding-style-patch-verification: ON
+ *
+ * Local Variables:
+ * eval: (c-set-style "gnu")
+ * End:
+ */