Commit | Line | Data |
---|---|---|
73e9abbe JR |
1 | /* |
2 | * Copyright (c) - 2019 Jonathan Rajotte-Julien | |
3 | * <jonathan.rajotte-julien@efficios.com> | |
4 | * | |
5 | * This program is free software; you can redistribute it and/or modify it | |
6 | * under the terms of the GNU General Public License as published by as | |
7 | * published by the Free Software Foundation; only version 2 of the License. | |
8 | * | |
9 | * This program is distributed in the hope that it will be useful, but WITHOUT | |
10 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
11 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for | |
12 | * more details. | |
13 | * | |
14 | * You should have received a copy of the GNU General Public License along with | |
15 | * this program; if not, write to the Free Software Foundation, Inc., 51 | |
16 | * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. | |
17 | */ | |
18 | ||
19 | #include <assert.h> | |
20 | #include <stdbool.h> | |
21 | #include <stdio.h> | |
22 | #include <stdlib.h> | |
23 | #include <string.h> | |
24 | #include <tap/tap.h> | |
25 | ||
26 | #include "backward-compatibility-group-by.h" | |
27 | ||
28 | /* Number of TAP tests in this file */ | |
29 | #define NUM_TESTS_PER_TEST 1 | |
30 | ||
31 | struct test { | |
32 | char *stream_path; | |
33 | char *session_name; | |
34 | char *hostname; | |
35 | char *creation_time; | |
36 | char *extra_path; | |
37 | char *leftover; | |
38 | bool is_valid; | |
39 | }; | |
40 | ||
41 | int lttng_opt_quiet; | |
42 | int lttng_opt_mi; | |
43 | int lttng_opt_verbose; | |
44 | ||
45 | struct test tests[] = { | |
46 | /* Default name session streaming. */ | |
47 | {"hostname/auto-20190918-164429/ust/uid/1000/64-bit", | |
48 | "auto-20190918-164429", "hostname", | |
49 | "20190918-164429", "", "ust/uid/1000/64-bit", | |
50 | true}, | |
51 | /* Custom default name session */ | |
52 | {"hostname/custom_auto-20190319-120000/ust/uid/1000/64-bit", | |
53 | "custom_auto-20190319-120000", "hostname", | |
54 | "20190319-120000", "", "ust/uid/1000/64-bit", | |
55 | true}, | |
56 | /* Named session streaming */ | |
57 | {"hostname/test-20190918-164709/ust/uid/1000/64-bit", "test", | |
58 | "hostname", "20190918-164709", "", | |
59 | "ust/uid/1000/64-bit", true}, | |
60 | /* Default session snapshot streaming */ | |
61 | {"hostname//snapshot-1-20190918-164829-0/ust//uid/1000/64-bit", | |
62 | "my_session", "hostname", "", "", | |
63 | "snapshot-1-20190918-164829-0/ust//uid/1000/64-bit", | |
64 | true}, | |
65 | /* Named session snapshot streaming */ | |
66 | {"hostname//snapshot-1-20190918-175919-0/ust//uid/1000/64-bit", | |
67 | "my_session", "hostname", "", "", | |
68 | "snapshot-1-20190918-175919-0/ust//uid/1000/64-bit", | |
69 | true}, | |
70 | /* Default name session, live */ | |
71 | {"hostname//auto-20190918-171641/ust/uid/1000/64-bit", | |
72 | "auto-20190918-171641", "hostname", | |
73 | "20190918-171641", "", "ust/uid/1000/64-bit", | |
74 | true}, | |
75 | /* Named session, live */ | |
76 | {"hostname//test-20190918-180333//ust/uid/1000/64-bit", | |
77 | "test-20190918-180333", "hostname", | |
78 | "20190918-180333", "", "/ust/uid/1000/64-bit", | |
79 | true}, | |
80 | /* Default name session, streaming & live , extra path */ | |
81 | {"hostname/extra/path/ust/uid/1000/64-bit", | |
82 | "auto-20190919-122110", "hostname", | |
83 | "20190919-122110", "extra", | |
84 | "path/ust/uid/1000/64-bit", true}, | |
85 | /* Named session, live, extra path */ | |
86 | {"hostname/extra/path/ust/uid/1000/64-bit", "test", "hostname", | |
87 | "", "extra", "path/ust/uid/1000/64-bit", true}, | |
88 | /* Named session, snapshot, extra path */ | |
89 | {"hostname/extra/path/snapshot-1-20190919-140702-0/ust//uid/1000/64-bit", | |
90 | "test", "hostname", "", "extra", | |
91 | "path/snapshot-1-20190919-140702-0/ust//uid/1000/64-bit", | |
92 | true}, | |
93 | /* Corner cases*/ | |
94 | /* Named session with valid datetime in it */ | |
95 | /* Default name session, extra path with session name in it*/ | |
96 | {"hostname/test-20190319-120000-20190918-180921/ust/uid/1000/64-bit", | |
97 | "test-20190319-120000", "hostname", | |
98 | "20190918-180921", "", "ust/uid/1000/64-bit", | |
99 | true}, | |
100 | /* Empty path */ | |
101 | {"", "test", "", "", "", "", false}, | |
102 | /* Path without second token */ | |
103 | {"hostname", "test", "hostname", "", "", "", false}, | |
104 | /* No leftover */ | |
105 | {"hostname/test", "test", "hostname", "", "", "", true}, | |
106 | /* Path with ession name but no datetime */ | |
107 | {"hostname/test/ust/uid/1000/64-bit", "test", "hostname", "", | |
108 | "", "ust/uid/1000/64-bit", true}, | |
109 | }; | |
110 | ||
111 | static char *craft_expected(struct test *test) | |
112 | { | |
113 | int ret; | |
114 | char *result = NULL; | |
115 | ||
116 | ret = asprintf(&result, "%s/%s%s%s/%s%s%s", test->session_name, | |
117 | test->hostname, | |
118 | test->creation_time[0] != '\0' ? "-" : "", | |
119 | test->creation_time, test->extra_path, | |
120 | test->extra_path[0] != '\0' ? "/" : "", test->leftover); | |
121 | if (ret < 0) { | |
122 | result = NULL; | |
123 | } | |
124 | return result; | |
125 | } | |
126 | ||
127 | int main(int argc, char **argv) | |
128 | { | |
129 | int num_test = sizeof(tests) / sizeof(struct test); | |
130 | ||
131 | plan_tests(NUM_TESTS_PER_TEST * num_test); | |
132 | diag("Backward compatibility utils for lttng-relayd --group-by-session"); | |
133 | for (int i = 0; i < num_test; i++) { | |
134 | char *expected = NULL; | |
135 | char *result = NULL; | |
136 | ||
137 | expected = craft_expected(&tests[i]); | |
138 | if (!expected) { | |
139 | fprintf(stderr, "Failed to craft expected output\n"); | |
140 | goto loop; | |
141 | } | |
142 | ||
143 | result = backward_compat_group_by_session( | |
144 | tests[i].stream_path, tests[i].session_name); | |
145 | if (!result && tests[i].is_valid) { | |
146 | fprintf(stderr, "Failed to get result\n"); | |
147 | goto loop; | |
148 | } else if (!result && tests[i].is_valid == false) { | |
149 | pass("Returned null as expected"); | |
150 | goto loop; | |
151 | } | |
152 | ||
153 | ok(strncmp(expected, result, strlen(expected)) == 0, | |
154 | "In: %s, out: %s, expected: %s", | |
155 | tests[i].stream_path, result, expected); | |
156 | loop: | |
157 | free(expected); | |
158 | free(result); | |
159 | } | |
160 | return exit_status(); | |
161 | } |