2 * collectd - src/tests/test_utils_vl_lookup.c
3 * Copyright (C) 2012 Florian Forster
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:
12 * The above copyright notice and this permission notice shall be included in
13 * all copies or substantial portions of the Software.
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.
24 * Florian Forster <octo at collectd.org>
30 #include "utils_vl_lookup.h"
32 static _Bool expect_new_obj = 0;
33 static _Bool have_new_obj = 0;
35 static lookup_identifier_t last_class_ident;
36 static lookup_identifier_t last_obj_ident;
38 static data_source_t dsrc_test = {"value", DS_TYPE_DERIVE, 0.0, NAN};
39 static data_set_t const ds_test = {"test", 1, &dsrc_test};
41 static data_source_t dsrc_unknown = {"value", DS_TYPE_DERIVE, 0.0, NAN};
42 static data_set_t const ds_unknown = {"unknown", 1, &dsrc_unknown};
44 static int lookup_obj_callback(data_set_t const *ds, value_list_t const *vl,
45 void *user_class, void *user_obj) {
46 lookup_identifier_t *class = user_class;
47 lookup_identifier_t *obj = user_obj;
49 OK1(expect_new_obj == have_new_obj,
50 (expect_new_obj ? "New obj is created." : "Updating existing obj."));
52 memcpy(&last_class_ident, class, sizeof(last_class_ident));
53 memcpy(&last_obj_ident, obj, sizeof(last_obj_ident));
55 if (strcmp(obj->plugin_instance, "failure") == 0)
61 static void *lookup_class_callback(data_set_t const *ds, value_list_t const *vl,
63 lookup_identifier_t *class = user_class;
64 lookup_identifier_t *obj;
66 assert(expect_new_obj);
68 memcpy(&last_class_ident, class, sizeof(last_class_ident));
70 obj = malloc(sizeof(*obj));
71 strncpy(obj->host, vl->host, sizeof(obj->host));
72 strncpy(obj->plugin, vl->plugin, sizeof(obj->plugin));
73 strncpy(obj->plugin_instance, vl->plugin_instance,
74 sizeof(obj->plugin_instance));
75 strncpy(obj->type, vl->type, sizeof(obj->type));
76 strncpy(obj->type_instance, vl->type_instance, sizeof(obj->type_instance));
83 static int checked_lookup_add(lookup_t *obj, /* {{{ */
84 char const *host, char const *plugin,
85 char const *plugin_instance, char const *type,
86 char const *type_instance,
87 unsigned int group_by) {
88 lookup_identifier_t ident;
91 strncpy(ident.host, host, sizeof(ident.host));
92 strncpy(ident.plugin, plugin, sizeof(ident.plugin));
93 strncpy(ident.plugin_instance, plugin_instance,
94 sizeof(ident.plugin_instance));
95 strncpy(ident.type, type, sizeof(ident.type));
96 strncpy(ident.type_instance, type_instance, sizeof(ident.type_instance));
98 user_class = malloc(sizeof(ident));
99 memmove(user_class, &ident, sizeof(ident));
101 OK(lookup_add(obj, &ident, group_by, user_class) == 0);
103 } /* }}} int checked_lookup_add */
105 static int checked_lookup_search(lookup_t *obj, char const *host,
107 char const *plugin_instance, char const *type,
108 char const *type_instance, _Bool expect_new) {
110 value_list_t vl = VALUE_LIST_INIT;
111 data_set_t const *ds = &ds_unknown;
113 strncpy(vl.host, host, sizeof(vl.host));
114 strncpy(vl.plugin, plugin, sizeof(vl.plugin));
115 strncpy(vl.plugin_instance, plugin_instance, sizeof(vl.plugin_instance));
116 strncpy(vl.type, type, sizeof(vl.type));
117 strncpy(vl.type_instance, type_instance, sizeof(vl.type_instance));
119 if (strcmp(vl.type, "test") == 0)
122 expect_new_obj = expect_new;
125 status = lookup_search(obj, ds, &vl);
129 DEF_TEST(group_by_specific_host) {
131 CHECK_NOT_NULL(obj = lookup_create(lookup_class_callback, lookup_obj_callback,
132 (void *)free, (void *)free));
134 checked_lookup_add(obj, "/.*/", "test", "", "test", "/.*/", LU_GROUP_BY_HOST);
135 checked_lookup_search(obj, "host0", "test", "", "test", "0",
136 /* expect new = */ 1);
137 checked_lookup_search(obj, "host0", "test", "", "test", "1",
138 /* expect new = */ 0);
139 checked_lookup_search(obj, "host1", "test", "", "test", "0",
140 /* expect new = */ 1);
141 checked_lookup_search(obj, "host1", "test", "", "test", "1",
142 /* expect new = */ 0);
148 DEF_TEST(group_by_any_host) {
150 CHECK_NOT_NULL(obj = lookup_create(lookup_class_callback, lookup_obj_callback,
151 (void *)free, (void *)free));
153 checked_lookup_add(obj, "/.*/", "/.*/", "/.*/", "test", "/.*/",
155 checked_lookup_search(obj, "host0", "plugin0", "", "test", "0",
156 /* expect new = */ 1);
157 checked_lookup_search(obj, "host0", "plugin0", "", "test", "1",
158 /* expect new = */ 0);
159 checked_lookup_search(obj, "host0", "plugin1", "", "test", "0",
160 /* expect new = */ 0);
161 checked_lookup_search(obj, "host0", "plugin1", "", "test", "1",
162 /* expect new = */ 0);
163 checked_lookup_search(obj, "host1", "plugin0", "", "test", "0",
164 /* expect new = */ 1);
165 checked_lookup_search(obj, "host1", "plugin0", "", "test", "1",
166 /* expect new = */ 0);
167 checked_lookup_search(obj, "host1", "plugin1", "", "test", "0",
168 /* expect new = */ 0);
169 checked_lookup_search(obj, "host1", "plugin1", "", "test", "1",
170 /* expect new = */ 0);
176 DEF_TEST(multiple_lookups) {
180 CHECK_NOT_NULL(obj = lookup_create(lookup_class_callback, lookup_obj_callback,
181 (void *)free, (void *)free));
183 checked_lookup_add(obj, "/.*/", "plugin0", "", "test", "/.*/",
185 checked_lookup_add(obj, "/.*/", "/.*/", "", "test", "ti0", LU_GROUP_BY_HOST);
187 status = checked_lookup_search(obj, "host0", "plugin1", "", "test", "",
188 /* expect new = */ 0);
190 status = checked_lookup_search(obj, "host0", "plugin0", "", "test", "",
191 /* expect new = */ 1);
193 status = checked_lookup_search(obj, "host0", "plugin1", "", "test", "ti0",
194 /* expect new = */ 1);
196 status = checked_lookup_search(obj, "host0", "plugin0", "", "test", "ti0",
197 /* expect new = */ 0);
206 CHECK_NOT_NULL(obj = lookup_create(lookup_class_callback, lookup_obj_callback,
207 (void *)free, (void *)free));
209 checked_lookup_add(obj, "/^db[0-9]\\./", "cpu", "/.*/", "cpu", "/.*/",
210 LU_GROUP_BY_TYPE_INSTANCE);
211 checked_lookup_search(obj, "db0.example.com", "cpu", "0", "cpu", "user",
212 /* expect new = */ 1);
213 checked_lookup_search(obj, "db0.example.com", "cpu", "0", "cpu", "idle",
214 /* expect new = */ 1);
215 checked_lookup_search(obj, "db0.example.com", "cpu", "1", "cpu", "user",
216 /* expect new = */ 0);
217 checked_lookup_search(obj, "db0.example.com", "cpu", "1", "cpu", "idle",
218 /* expect new = */ 0);
219 checked_lookup_search(obj, "app0.example.com", "cpu", "0", "cpu", "user",
220 /* expect new = */ 0);
221 checked_lookup_search(obj, "app0.example.com", "cpu", "0", "cpu", "idle",
222 /* expect new = */ 0);
223 checked_lookup_search(obj, "db1.example.com", "cpu", "0", "cpu", "user",
224 /* expect new = */ 0);
225 checked_lookup_search(obj, "db1.example.com", "cpu", "0", "cpu", "idle",
226 /* expect new = */ 0);
227 checked_lookup_search(obj, "db1.example.com", "cpu", "0", "cpu", "system",
228 /* expect new = */ 1);
234 int main(int argc, char **argv) /* {{{ */
236 RUN_TEST(group_by_specific_host);
237 RUN_TEST(group_by_any_host);
238 RUN_TEST(multiple_lookups);