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 */
27 #include "common.h" /* STATIC_ARRAY_SIZE */
28 #include "collectd.h"
30 #include "testing.h"
31 #include "utils_avltree.h"
33 static int compare_total_count = 0;
34 #define RESET_COUNTS() do { compare_total_count = 0; } while (0)
36 static int compare_callback (void const *v0, void const *v1)
37 {
38 assert (v0 != NULL);
39 assert (v1 != NULL);
41 compare_total_count++;
42 return (strcmp (v0, v1));
43 }
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 };
73 c_avl_tree_t *t;
74 size_t i;
76 RESET_COUNTS ();
77 CHECK_NOT_NULL (t = c_avl_create (compare_callback));
79 /* insert */
80 for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++)
81 {
82 char *key;
83 char *value;
85 CHECK_NOT_NULL (key = strdup (cases[i].key));
86 CHECK_NOT_NULL (value = strdup (cases[i].value));
88 CHECK_ZERO (c_avl_insert (t, key, value));
89 EXPECT_EQ_INT ((int) (i + 1), c_avl_size (t));
90 }
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));
96 /* get */
97 for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++)
98 {
99 char *value_ret = NULL;
101 CHECK_ZERO (c_avl_get (t, cases[i].key, (void *) &value_ret));
102 EXPECT_EQ_STR (cases[i].value, value_ret);
103 }
105 /* remove half */
106 for (i = 0; i < STATIC_ARRAY_SIZE (cases) / 2; i++)
107 {
108 char *key = NULL;
109 char *value = NULL;
111 int expected_size = (int) (STATIC_ARRAY_SIZE (cases) - (i + 1));
113 CHECK_ZERO (c_avl_remove (t, cases[i].key, (void *) &key, (void *) &value));
115 EXPECT_EQ_STR (cases[i].key, key);
116 EXPECT_EQ_STR (cases[i].value, value);
118 free (key);
119 free (value);
121 EXPECT_EQ_INT (expected_size, c_avl_size (t));
122 }
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;
130 int expected_size = (int) (STATIC_ARRAY_SIZE (cases) - (i + 1));
132 EXPECT_EQ_INT (expected_size + 1, c_avl_size (t));
133 EXPECT_EQ_INT (0, c_avl_pick (t, (void *) &key, (void *) &value));
135 free (key);
136 free (value);
138 EXPECT_EQ_INT (expected_size, c_avl_size (t));
139 }
141 c_avl_destroy (t);
143 return (0);
144 }
146 int main (void)
147 {
148 RUN_TEST(success);
150 END_TEST;
151 }
153 /* vim: set sw=2 sts=2 et : */