src/daemon/utils_avltree_test.c

   1 /**
   2  * collectd - src/tests/test_utils_avltree.c
   3  * Copyright (C) 2013       Florian octo Forster
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in
  13  * all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *   Florian octo Forster <octo at collectd.org>
  25  */
  26
  27 #include "common.h" /* STATIC_ARRAY_SIZE */
  28 #include "collectd.h"
  29
  30 #include "testing.h"
  31 #include "utils_avltree.h"
  32
  33 static int compare_total_count = 0;
  34 #define RESET_COUNTS() do { compare_total_count = 0; } while (0)
  35
  36 static int compare_callback (void const *v0, void const *v1)
  37 {
  38   assert (v0 != NULL);
  39   assert (v1 != NULL);
  40
  41   compare_total_count++;
  42   return (strcmp (v0, v1));
  43 }
  44
  45 DEF_TEST(success)
  46 {
  47   struct {
  48     char *key;
  49     char *value;
  50   } cases[] = {
  51     {"Eeph7chu", "vai1reiV"},
  52     {"igh3Paiz", "teegh1Ee"},
  53     {"caip6Uu8", "ooteQu8n"},
  54     {"Aech6vah", "AijeeT0l"},
  55     {"Xah0et2L", "gah8Taep"},
  56     {"BocaeB8n", "oGaig8io"},
  57     {"thai8AhM", "ohjeFo3f"},
  58     {"ohth6ieC", "hoo8ieWo"},
  59     {"aej7Woow", "phahuC2s"},
  60     {"Hai8ier2", "Yie6eimi"},
  61     {"phuXi3Li", "JaiF7ieb"},
  62     {"Shaig5ef", "aihi5Zai"},
  63     {"voh6Aith", "Oozaeto0"},
  64     {"zaiP5kie", "seep5veM"},
  65     {"pae7ba7D", "chie8Ojo"},
  66     {"Gou2ril3", "ouVoo0ha"},
  67     {"lo3Thee3", "ahDu4Zuj"},
  68     {"Rah8kohv", "ieShoc7E"},
  69     {"ieN5engi", "Aevou1ah"},
  70     {"ooTe4OhP", "aingai5Y"},
  71   };
  72
  73   c_avl_tree_t *t;
  74   size_t i;
  75
  76   RESET_COUNTS ();
  77   CHECK_NOT_NULL (t = c_avl_create (compare_callback));
  78
  79   /* insert */
  80   for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++)
  81   {
  82     char *key;
  83     char *value;
  84
  85     CHECK_NOT_NULL (key = strdup (cases[i].key));
  86     CHECK_NOT_NULL (value = strdup (cases[i].value));
  87
  88     CHECK_ZERO (c_avl_insert (t, key, value));
  89     EXPECT_EQ_INT ((int) (i + 1), c_avl_size (t));
  90   }
  91
  92   /* Key already exists. */
  93   for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++)
  94     EXPECT_EQ_INT (1, c_avl_insert (t, cases[i].key, cases[i].value));
  95
  96   /* get */
  97   for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++)
  98   {
  99     char *value_ret = NULL;
 100
 101     CHECK_ZERO (c_avl_get (t, cases[i].key, (void *) &value_ret));
 102     EXPECT_EQ_STR (cases[i].value, value_ret);
 103   }
 104
 105   /* remove half */
 106   for (i = 0; i < STATIC_ARRAY_SIZE (cases) / 2; i++)
 107   {
 108     char *key = NULL;
 109     char *value = NULL;
 110
 111     int expected_size = (int) (STATIC_ARRAY_SIZE (cases) - (i + 1));
 112
 113     CHECK_ZERO (c_avl_remove (t, cases[i].key, (void *) &key, (void *) &value));
 114
 115     EXPECT_EQ_STR (cases[i].key, key);
 116     EXPECT_EQ_STR (cases[i].value, value);
 117
 118     free (key);
 119     free (value);
 120
 121     EXPECT_EQ_INT (expected_size, c_avl_size (t));
 122   }
 123
 124   /* pick the other half */
 125   for (i = STATIC_ARRAY_SIZE (cases) / 2; i < STATIC_ARRAY_SIZE (cases); i++)
 126   {
 127     char *key = NULL;
 128     char *value = NULL;
 129
 130     int expected_size = (int) (STATIC_ARRAY_SIZE (cases) - (i + 1));
 131
 132     EXPECT_EQ_INT (expected_size + 1, c_avl_size (t));
 133     EXPECT_EQ_INT (0, c_avl_pick (t, (void *) &key, (void *) &value));
 134
 135     free (key);
 136     free (value);
 137
 138     EXPECT_EQ_INT (expected_size, c_avl_size (t));
 139   }
 140
 141   c_avl_destroy (t);
 142
 143   return (0);
 144 }
 145
 146 int main (void)
 147 {
 148   RUN_TEST(success);
 149
 150   END_TEST;
 151 }
 152
 153 /* vim: set sw=2 sts=2 et : */