Merge branch 'collectd-5.7'
[collectd.git] / src / curl_json_test.c
1 /**
2  * collectd - src/curl_json.c
3  * Copyright (C) 2017  Florian octo Forster
4  *
5  * Licensed under the same terms and conditions as src/curl_json.c.
6  *
7  * Authors:
8  *   Florian octo Forster <octo at collectd.org>
9  **/
10
11 #include "curl_json.c"
12
13 #include "testing.h"
14
15 static void test_submit(cj_t *db, cj_key_t *key, value_t *value) {
16   /* hack: we repurpose db->curl to store received values. */
17   c_avl_tree_t *values = (void *)db->curl;
18
19   value_t *value_copy = calloc(1, sizeof(*value_copy));
20   memmove(value_copy, value, sizeof(*value_copy));
21
22   assert(c_avl_insert(values, key->path, value_copy) == 0);
23 }
24
25 static derive_t test_metric(cj_t *db, char const *path) {
26   c_avl_tree_t *values = (void *)db->curl;
27
28   value_t *ret = NULL;
29   if (c_avl_get(values, path, (void *)&ret) == 0) {
30     return ret->derive;
31   }
32
33   return -1;
34 }
35
36 static cj_t *test_setup(char *json, char *key_path) {
37   cj_t *db = calloc(1, sizeof(*db));
38   db->yajl = yajl_alloc(&ycallbacks,
39 #if HAVE_YAJL_V2
40                         /* alloc funcs = */ NULL,
41 #else
42                         /* alloc funcs = */ NULL, NULL,
43 #endif
44                         /* context = */ (void *)db);
45
46   /* hack; see above. */
47   db->curl = (void *)cj_avl_create();
48
49   cj_key_t *key = calloc(1, sizeof(*key));
50   key->path = strdup(key_path);
51   key->type = strdup("MAGIC");
52
53   assert(cj_append_key(db, key) == 0);
54
55   cj_tree_entry_t root = {0};
56   root.type = TREE;
57   root.tree = db->tree;
58   db->state[0].entry = &root;
59
60   cj_curl_callback(json, strlen(json), 1, db);
61 #if HAVE_YAJL_V2
62   yajl_complete_parse(db->yajl);
63 #else
64   yajl_parse_complete(db->yajl);
65 #endif
66
67   db->state[0].entry = NULL;
68
69   return db;
70 }
71
72 static void test_teardown(cj_t *db) {
73   c_avl_tree_t *values = (void *)db->curl;
74   db->curl = NULL;
75
76   void *key;
77   void *value;
78   while (c_avl_pick(values, &key, &value) == 0) {
79     /* key will be freed by cj_free. */
80     free(value);
81   }
82   c_avl_destroy(values);
83
84   yajl_free(db->yajl);
85   db->yajl = NULL;
86
87   cj_free(db);
88 }
89
90 DEF_TEST(parse) {
91   struct {
92     char *json;
93     char *key_path;
94     derive_t want;
95   } cases[] = {
96       /* simple map */
97       {"{\"foo\":42,\"bar\":23}", "foo", 42},
98       {"{\"foo\":42,\"bar\":23}", "bar", 23},
99       /* nested map */
100       {"{\"a\":{\"b\":{\"c\":123}}", "a/b/c", 123},
101       {"{\"x\":{\"y\":{\"z\":789}}", "x/*/z", 789},
102       /* simple array */
103       {"[10,11,12,13]", "0", 10},
104       {"[10,11,12,13]", "1", 11},
105       {"[10,11,12,13]", "2", 12},
106       {"[10,11,12,13]", "3", 13},
107       /* array index after non-numeric entry */
108       {"[true,11]", "1", 11},
109       {"[null,11]", "1", 11},
110       {"[\"s\",11]", "1", 11},
111       {"[{\"k\":\"v\"},11]", "1", 11},
112       {"[[0,1,2],11]", "1", 11},
113       /* nested array */
114       {"[[0,1,2],[3,4,5],[6,7,8]]", "0/0", 0},
115       {"[[0,1,2],[3,4,5],[6,7,8]]", "0/1", 1},
116       {"[[0,1,2],[3,4,5],[6,7,8]]", "0/2", 2},
117       {"[[0,1,2],[3,4,5],[6,7,8]]", "1/0", 3},
118       {"[[0,1,2],[3,4,5],[6,7,8]]", "1/1", 4},
119       {"[[0,1,2],[3,4,5],[6,7,8]]", "1/2", 5},
120       {"[[0,1,2],[3,4,5],[6,7,8]]", "2/0", 6},
121       {"[[0,1,2],[3,4,5],[6,7,8]]", "2/1", 7},
122       {"[[0,1,2],[3,4,5],[6,7,8]]", "2/2", 8},
123       /* testcase from #2266 */
124       {"{\"a\":[[10,11,12,13,14]]}", "a/0/0", 10},
125       {"{\"a\":[[10,11,12,13,14]]}", "a/0/1", 11},
126       {"{\"a\":[[10,11,12,13,14]]}", "a/0/2", 12},
127       {"{\"a\":[[10,11,12,13,14]]}", "a/0/3", 13},
128       {"{\"a\":[[10,11,12,13,14]]}", "a/0/4", 14},
129   };
130
131   for (size_t i = 0; i < STATIC_ARRAY_SIZE(cases); i++) {
132     cj_t *db = test_setup(cases[i].json, cases[i].key_path);
133
134     EXPECT_EQ_INT(cases[i].want, test_metric(db, cases[i].key_path));
135
136     test_teardown(db);
137   }
138
139   return 0;
140 }
141
142 int main(int argc, char **argv) {
143   cj_submit = test_submit;
144
145   RUN_TEST(parse);
146
147   END_TEST;
148 }