1 /**
2 * collectd - src/tests/test_common.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"
28 #include "testing.h"
30 #if HAVE_LIBKSTAT
31 kstat_ctl_t *kc;
32 #endif /* HAVE_LIBKSTAT */
34 DEF_TEST(sstrncpy)
35 {
36 char buffer[16] = "";
37 char *ptr = &buffer[4];
38 char *ret;
40 buffer[0] = buffer[1] = buffer[2] = buffer[3] = 0xff;
41 buffer[12] = buffer[13] = buffer[14] = buffer[15] = 0xff;
43 ret = sstrncpy (ptr, "foobar", 8);
44 OK(ret == ptr);
45 EXPECT_EQ_STR ("foobar", ptr);
46 OK(buffer[3] == buffer[12]);
48 ret = sstrncpy (ptr, "abc", 8);
49 OK(ret == ptr);
50 EXPECT_EQ_STR ("abc", ptr);
51 OK(buffer[3] == buffer[12]);
53 ret = sstrncpy (ptr, "collectd", 8);
54 OK(ret == ptr);
55 OK(ptr[7] == 0);
56 EXPECT_EQ_STR ("collect", ptr);
57 OK(buffer[3] == buffer[12]);
59 return (0);
60 }
62 DEF_TEST(ssnprintf)
63 {
64 char buffer[16] = "";
65 char *ptr = &buffer[4];
66 int status;
68 buffer[0] = buffer[1] = buffer[2] = buffer[3] = 0xff;
69 buffer[12] = buffer[13] = buffer[14] = buffer[15] = 0xff;
71 status = ssnprintf (ptr, 8, "%i", 1337);
72 OK(status == 4);
73 EXPECT_EQ_STR ("1337", ptr);
75 status = ssnprintf (ptr, 8, "%s", "collectd");
76 OK(status == 8);
77 OK(ptr[7] == 0);
78 EXPECT_EQ_STR ("collect", ptr);
79 OK(buffer[3] == buffer[12]);
81 return (0);
82 }
84 DEF_TEST(sstrdup)
85 {
86 char *ptr;
88 ptr = sstrdup ("collectd");
89 OK(ptr != NULL);
90 EXPECT_EQ_STR ("collectd", ptr);
92 sfree(ptr);
94 ptr = sstrdup (NULL);
95 OK(ptr == NULL);
97 return (0);
98 }
100 DEF_TEST(strsplit)
101 {
102 char buffer[32];
103 char *fields[8];
104 int status;
106 strncpy (buffer, "foo bar", sizeof (buffer));
107 status = strsplit (buffer, fields, 8);
108 OK(status == 2);
109 EXPECT_EQ_STR ("foo", fields[0]);
110 EXPECT_EQ_STR ("bar", fields[1]);
112 strncpy (buffer, "foo \t bar", sizeof (buffer));
113 status = strsplit (buffer, fields, 8);
114 OK(status == 2);
115 EXPECT_EQ_STR ("foo", fields[0]);
116 EXPECT_EQ_STR ("bar", fields[1]);
118 strncpy (buffer, "one two\tthree\rfour\nfive", sizeof (buffer));
119 status = strsplit (buffer, fields, 8);
120 OK(status == 5);
121 EXPECT_EQ_STR ("one", fields[0]);
122 EXPECT_EQ_STR ("two", fields[1]);
123 EXPECT_EQ_STR ("three", fields[2]);
124 EXPECT_EQ_STR ("four", fields[3]);
125 EXPECT_EQ_STR ("five", fields[4]);
127 strncpy (buffer, "\twith trailing\n", sizeof (buffer));
128 status = strsplit (buffer, fields, 8);
129 OK(status == 2);
130 EXPECT_EQ_STR ("with", fields[0]);
131 EXPECT_EQ_STR ("trailing", fields[1]);
133 strncpy (buffer, "1 2 3 4 5 6 7 8 9 10 11 12 13", sizeof (buffer));
134 status = strsplit (buffer, fields, 8);
135 OK(status == 8);
136 EXPECT_EQ_STR ("7", fields[6]);
137 EXPECT_EQ_STR ("8", fields[7]);
139 strncpy (buffer, "single", sizeof (buffer));
140 status = strsplit (buffer, fields, 8);
141 OK(status == 1);
142 EXPECT_EQ_STR ("single", fields[0]);
144 strncpy (buffer, "", sizeof (buffer));
145 status = strsplit (buffer, fields, 8);
146 OK(status == 0);
148 return (0);
149 }
151 DEF_TEST(strjoin)
152 {
153 char buffer[16];
154 char *fields[4];
155 int status;
157 fields[0] = "foo";
158 fields[1] = "bar";
159 fields[2] = "baz";
160 fields[3] = "qux";
162 status = strjoin (buffer, sizeof (buffer), fields, 2, "!");
163 OK(status == 7);
164 EXPECT_EQ_STR ("foo!bar", buffer);
166 status = strjoin (buffer, sizeof (buffer), fields, 1, "!");
167 OK(status == 3);
168 EXPECT_EQ_STR ("foo", buffer);
170 status = strjoin (buffer, sizeof (buffer), fields, 0, "!");
171 OK(status < 0);
173 status = strjoin (buffer, sizeof (buffer), fields, 2, "rcht");
174 OK(status == 10);
175 EXPECT_EQ_STR ("foorchtbar", buffer);
177 status = strjoin (buffer, sizeof (buffer), fields, 4, "");
178 OK(status == 12);
179 EXPECT_EQ_STR ("foobarbazqux", buffer);
181 status = strjoin (buffer, sizeof (buffer), fields, 4, "!");
182 OK(status == 15);
183 EXPECT_EQ_STR ("foo!bar!baz!qux", buffer);
185 fields[0] = "0123";
186 fields[1] = "4567";
187 fields[2] = "8901";
188 fields[3] = "2345";
189 status = strjoin (buffer, sizeof (buffer), fields, 4, "-");
190 OK(status < 0);
192 return (0);
193 }
195 DEF_TEST(escape_slashes)
196 {
197 struct {
198 char *str;
199 char *want;
200 } cases[] = {
201 {"foo/bar/baz", "foo_bar_baz"},
202 {"/like/a/path", "like_a_path"},
203 {"trailing/slash/", "trailing_slash_"},
204 {"foo//bar", "foo__bar"},
205 };
206 size_t i;
208 for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++) {
209 char buffer[32];
211 strncpy (buffer, cases[i].str, sizeof (buffer));
212 OK(escape_slashes (buffer, sizeof (buffer)) == 0);
213 EXPECT_EQ_STR(cases[i].want, buffer);
214 }
216 return 0;
217 }
219 DEF_TEST(escape_string)
220 {
221 struct {
222 char *str;
223 char *want;
224 } cases[] = {
225 {"foobar", "foobar"},
226 {"f00bar", "f00bar"},
227 {"foo bar", "\"foo bar\""},
228 {"foo \"bar\"", "\"foo \\\"bar\\\"\""},
229 {"012345678901234", "012345678901234"},
230 {"012345 78901234", "\"012345 789012\""},
231 {"012345 78901\"34", "\"012345 78901\""},
232 };
233 size_t i;
235 for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++) {
236 char buffer[16];
238 strncpy (buffer, cases[i].str, sizeof (buffer));
239 OK(escape_string (buffer, sizeof (buffer)) == 0);
240 EXPECT_EQ_STR(cases[i].want, buffer);
241 }
243 return 0;
244 }
246 DEF_TEST(strunescape)
247 {
248 char buffer[16];
249 int status;
251 strncpy (buffer, "foo\\tbar", sizeof (buffer));
252 status = strunescape (buffer, sizeof (buffer));
253 OK(status == 0);
254 EXPECT_EQ_STR ("foo\tbar", buffer);
256 strncpy (buffer, "\\tfoo\\r\\n", sizeof (buffer));
257 status = strunescape (buffer, sizeof (buffer));
258 OK(status == 0);
259 EXPECT_EQ_STR ("\tfoo\r\n", buffer);
261 strncpy (buffer, "With \\\"quotes\\\"", sizeof (buffer));
262 status = strunescape (buffer, sizeof (buffer));
263 OK(status == 0);
264 EXPECT_EQ_STR ("With \"quotes\"", buffer);
266 /* Backslash before null byte */
267 strncpy (buffer, "\\tbackslash end\\", sizeof (buffer));
268 status = strunescape (buffer, sizeof (buffer));
269 OK(status != 0);
270 EXPECT_EQ_STR ("\tbackslash end", buffer);
271 return (0);
273 /* Backslash at buffer end */
274 strncpy (buffer, "\\t3\\56", sizeof (buffer));
275 status = strunescape (buffer, 4);
276 OK(status != 0);
277 OK(buffer[0] == '\t');
278 OK(buffer[1] == '3');
279 OK(buffer[2] == 0);
280 OK(buffer[3] == 0);
281 OK(buffer[4] == '5');
282 OK(buffer[5] == '6');
283 OK(buffer[6] == '7');
285 return (0);
286 }
288 DEF_TEST(parse_values)
289 {
290 struct {
291 char buffer[64];
292 int status;
293 gauge_t value;
294 } cases[] = {
295 {"1435044576:42", 0, 42.0},
296 {"1435044576:42:23", -1, NAN},
297 {"1435044576:U", 0, NAN},
298 {"N:12.3", 0, 12.3},
299 {"N:42.0:23", -1, NAN},
300 {"N:U", 0, NAN},
301 {"T:42.0", -1, NAN},
302 };
304 size_t i;
305 for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++)
306 {
307 data_source_t dsrc = {
308 .name = "value",
309 .type = DS_TYPE_GAUGE,
310 .min = 0.0,
311 .max = NAN,
312 };
313 data_set_t ds = {
314 .type = "example",
315 .ds_num = 1,
316 .ds = &dsrc,
317 };
319 value_t v = {
320 .gauge = NAN,
321 };
322 value_list_t vl = {
323 .values = &v,
324 .values_len = 1,
325 .time = 0,
326 .interval = 0,
327 .host = "example.com",
328 .plugin = "common_test",
329 .type = "example",
330 .meta = NULL,
331 };
333 int status = parse_values (cases[i].buffer, &vl, &ds);
334 EXPECT_EQ_INT (cases[i].status, status);
335 if (status != 0)
336 continue;
338 EXPECT_EQ_DOUBLE (cases[i].value, vl.values[0].gauge);
339 }
341 return (0);
342 }
344 DEF_TEST(value_to_rate)
345 {
346 struct {
347 time_t t0;
348 time_t t1;
349 int ds_type;
350 value_t v0;
351 value_t v1;
352 gauge_t want;
353 } cases[] = {
354 { 0, 10, DS_TYPE_DERIVE, {.derive = 0}, {.derive = 1000}, NAN},
355 {10, 20, DS_TYPE_DERIVE, {.derive = 1000}, {.derive = 2000}, 100.0},
356 {20, 30, DS_TYPE_DERIVE, {.derive = 2000}, {.derive = 1800}, -20.0},
357 { 0, 10, DS_TYPE_COUNTER, {.counter = 0}, {.counter = 1000}, NAN},
358 {10, 20, DS_TYPE_COUNTER, {.counter = 1000}, {.counter = 5000}, 400.0},
359 /* 32bit wrap-around. */
360 {20, 30, DS_TYPE_COUNTER, {.counter = 4294967238ULL}, {.counter = 42}, 10.0},
361 /* 64bit wrap-around. */
362 {30, 40, DS_TYPE_COUNTER, {.counter = 18446744073709551558ULL}, {.counter = 42}, 10.0},
363 };
364 size_t i;
366 for (i = 0; i < STATIC_ARRAY_SIZE (cases); i++) {
367 value_to_rate_state_t state = { cases[i].v0, TIME_T_TO_CDTIME_T (cases[i].t0) };
368 gauge_t got;
370 if (cases[i].t0 == 0) {
371 OK(value_to_rate (&got, cases[i].v1, cases[i].ds_type, TIME_T_TO_CDTIME_T(cases[i].t1), &state) == EAGAIN);
372 continue;
373 }
375 OK(value_to_rate (&got, cases[i].v1, cases[i].ds_type, TIME_T_TO_CDTIME_T(cases[i].t1), &state) == 0);
376 EXPECT_EQ_DOUBLE(cases[i].want, got);
377 }
379 return 0;
380 }
382 int main (void)
383 {
384 RUN_TEST(sstrncpy);
385 RUN_TEST(ssnprintf);
386 RUN_TEST(sstrdup);
387 RUN_TEST(strsplit);
388 RUN_TEST(strjoin);
389 RUN_TEST(escape_slashes);
390 RUN_TEST(escape_string);
391 RUN_TEST(strunescape);
392 RUN_TEST(parse_values);
393 RUN_TEST(value_to_rate);
395 END_TEST;
396 }
398 /* vim: set sw=2 sts=2 et : */