src/plugins/unittest/pool_test.c

   1 /*
   2  * Copyright (c) 2021 Dave Barach
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include <vlib/vlib.h>
  17
  18 static clib_error_t *
  19 test_pool_command_fn (vlib_main_t *vm, unformat_input_t *input,
  20                       vlib_cli_command_t *cmd)
  21 {
  22   int i;
  23   u64 *pool;
  24
  25   pool_init_fixed (pool, 2048);
  26
  27   i = 0;
  28
  29   while (pool_free_elts (pool) > 0)
  30     {
  31       u64 *p __attribute__ ((unused));
  32
  33       pool_get (pool, p);
  34       i++;
  35     }
  36
  37   vlib_cli_output (vm, "allocated %d elts\n", i);
  38
  39   for (--i; i >= 0; i--)
  40     {
  41       pool_put_index (pool, i);
  42     }
  43
  44   ALWAYS_ASSERT (pool_free_elts (pool) == 2048);
  45
  46   vlib_cli_output (vm, "Test succeeded...\n");
  47   return 0;
  48 }
  49
  50 VLIB_CLI_COMMAND (test_pool_command, static) = {
  51   .path = "test pool",
  52   .short_help = "vppinfra pool.h tests",
  53   .function = test_pool_command_fn,
  54 };
  55
  56 /*
  57  * fd.io coding-style-patch-verification: ON
  58  *
  59  * Local Variables:
  60  * eval: (c-set-style "gnu")
  61  * End:
  62  */