Loader: Remove unneeded load_pri declarations.
[asterisk/asterisk.git] / tests / test_substitution.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2009, Digium, Inc.
5  *
6  * Tilghman Lesher <tlesher AT digium DOT com>
7  *
8  * See http://www.asterisk.org for more information about
9  * the Asterisk project. Please do not directly contact
10  * any of the maintainers of this project for assistance;
11  * the project provides a web site, mailing lists and IRC
12  * channels for your use.
13  *
14  * This program is free software, distributed under the terms of
15  * the GNU General Public License Version 2. See the LICENSE file
16  * at the top of the source tree.
17  */
18
19 /*! \file
20  *
21  * \brief Substitution Test
22  *
23  * \author\verbatim Tilghman Lesher <tlesher AT digium DOT com> \endverbatim
24  * 
25  * \ingroup tests
26  */
27
28 /*** MODULEINFO
29         <depend>TEST_FRAMEWORK</depend>
30         <depend>func_curl</depend>
31         <support_level>core</support_level>
32  ***/
33
34 #include "asterisk.h"
35
36 #include "asterisk/file.h"
37 #include "asterisk/channel.h"
38 #include "asterisk/pbx.h"
39 #include "asterisk/module.h"
40 #include "asterisk/lock.h"
41 #include "asterisk/app.h"
42 #include "asterisk/strings.h"
43 #include "asterisk/stringfields.h"
44 #include "asterisk/threadstorage.h"
45 #include "asterisk/test.h"
46
47 static enum ast_test_result_state test_chan_integer(struct ast_test *test,
48                 struct ast_channel *c, int *ifield, const char *expression)
49 {
50         int i, okay = 1, value1 = -1, value2 = -1;
51         char workspace[4096];
52         struct ast_str *str = ast_str_create(16);
53
54         for (i = 0; i < 256; i++) {
55                 *ifield = i;
56                 ast_str_substitute_variables(&str, 0, c, expression);
57                 pbx_substitute_variables_helper(c, expression, workspace, sizeof(workspace));
58
59                 if (sscanf(workspace, "%d", &value1) != 1 || value1 != i || sscanf(ast_str_buffer(str), "%d", &value2) != 1 || value2 != i) {
60                         ast_test_status_update(test, "%s != %s and/or %d != %d != %d\n", ast_str_buffer(str), workspace, value1, value2, i);
61                         okay = 0;
62                 }
63         }
64         ast_test_status_update(test, "Tested '%s' . . . . . %s\n", expression, okay ? "passed" : "FAILED");
65
66         ast_free(str);
67
68         return okay ? AST_TEST_PASS : AST_TEST_FAIL;
69 }
70
71 static enum ast_test_result_state test_chan_integer_accessor(struct ast_test *test,
72                 struct ast_channel *c, void (*setter)(struct ast_channel *, int),const char *expression)
73 {
74         int i, okay = 1, value1 = -1, value2 = -1;
75         char workspace[4096];
76         struct ast_str *str = ast_str_create(16);
77
78         for (i = 0; i < 256; i++) {
79                 setter(c, i);
80                 ast_str_substitute_variables(&str, 0, c, expression);
81                 pbx_substitute_variables_helper(c, expression, workspace, sizeof(workspace));
82
83                 if (sscanf(workspace, "%d", &value1) != 1 || value1 != i || sscanf(ast_str_buffer(str), "%d", &value2) != 1 || value2 != i) {
84                         ast_test_status_update(test, "%s != %s and/or %d != %d != %d\n", ast_str_buffer(str), workspace, value1, value2, i);
85                         okay = 0;
86                 }
87         }
88         ast_test_status_update(test, "Tested '%s' . . . . . %s\n", expression, okay ? "passed" : "FAILED");
89
90         ast_free(str);
91
92         return okay ? AST_TEST_PASS : AST_TEST_FAIL;
93 }
94
95 static enum ast_test_result_state test_chan_string(struct ast_test *test,
96                 struct ast_channel *c, void (*setter)(struct ast_channel *, const char *),
97                 const char *(*getter)(const struct ast_channel *), const char *expression)
98 {
99         const char *values[] = { "one", "three", "reallylongdinosaursoundingthingwithwordsinit" };
100         int i, okay = 1;
101         char workspace[4096];
102         struct ast_str *str = ast_str_create(16);
103
104         for (i = 0; i < ARRAY_LEN(values); i++) {
105                 setter(c, values[i]);
106                 ast_str_substitute_variables(&str, 0, c, expression);
107                 pbx_substitute_variables_helper(c, expression, workspace, sizeof(workspace));
108
109                 if (strcmp(getter(c), ast_str_buffer(str)) != 0 || strcmp(getter(c), workspace) != 0) {
110                         ast_test_status_update(test, "%s != %s != %s\n", getter(c), ast_str_buffer(str), workspace);
111                         okay = 0;
112                 }
113         }
114         ast_test_status_update(test, "Tested '%s' . . . . . %s\n",
115                 expression, okay ? "passed" : "FAILED");
116
117         ast_free(str);
118
119         return okay ? AST_TEST_PASS : AST_TEST_FAIL;
120 }
121
122 static enum ast_test_result_state test_chan_variable(struct ast_test *test,
123                 struct ast_channel *c, const char *varname)
124 {
125         const char *values[] = { "one", "three", "reallylongdinosaursoundingthingwithwordsinit" };
126         int i, okay = 1;
127         char workspace[4096];
128         struct ast_str *str = ast_str_create(16);
129         struct ast_str *var = ast_str_create(16);
130
131         ast_str_set(&var, 0, "${%s}", varname);
132         for (i = 0; i < ARRAY_LEN(values); i++) {
133                 pbx_builtin_setvar_helper(c, varname, values[i]);
134                 ast_str_substitute_variables(&str, 0, c, ast_str_buffer(var));
135                 pbx_substitute_variables_helper(c, ast_str_buffer(var), workspace, sizeof(workspace));
136
137                 if (strcmp(values[i], ast_str_buffer(str)) != 0 || strcmp(values[i], workspace) != 0) {
138                         ast_test_status_update(test, "%s != %s != %s\n",
139                                         values[i], ast_str_buffer(str), workspace);
140                         okay = 0;
141                 }
142         }
143         ast_test_status_update(test, "Tested '%s' . . . . . %s\n",
144                 ast_str_buffer(var), okay ? "passed" : "FAILED");
145
146         ast_free(str);
147         ast_free(var);
148
149         return okay ? AST_TEST_PASS : AST_TEST_FAIL;
150 }
151
152 static enum ast_test_result_state test_chan_function(struct ast_test *test,
153                 struct ast_channel *c, const char *expression)
154 {
155         int okay = 1;
156         char workspace[4096];
157         struct ast_str *str = ast_str_create(16);
158
159         ast_str_substitute_variables(&str, 0, c, expression);
160         pbx_substitute_variables_helper(c, expression, workspace, sizeof(workspace));
161
162         if (strcmp(workspace, ast_str_buffer(str)) != 0) {
163                 ast_test_status_update(test, "expr: '%s' ... %s != %s\n",
164                                 expression, ast_str_buffer(str), workspace);
165                 okay = 0;
166         }
167         ast_test_status_update(test, "Tested '%s' . . . . . %s\n",
168                 expression, okay ? "passed" : "FAILED");
169
170         ast_free(str);
171
172         return okay ? AST_TEST_PASS : AST_TEST_FAIL;
173 }
174
175 static enum ast_test_result_state test_2way_function(struct ast_test *test,
176                 struct ast_channel *c, const char *encode1, const char *encode2,
177                 const char *decode1, const char *decode2)
178 {
179         struct ast_str *str = ast_str_create(16), *expression = ast_str_alloca(120);
180         int okay;
181
182         ast_str_set(&expression, 0, "%s%s%s", encode1, "foobarbaz", encode2);
183         ast_str_substitute_variables(&str, 0, c, ast_str_buffer(expression));
184         ast_str_set(&expression, 0, "%s%s%s", decode1, ast_str_buffer(str), decode2);
185         ast_str_substitute_variables(&str, 0, c, ast_str_buffer(expression));
186
187         okay = !strcmp(ast_str_buffer(str), "foobarbaz");
188         if (!okay) {
189                 ast_test_status_update(test, "'%s' != 'foobarbaz'\n",
190                                 ast_str_buffer(str));
191         }
192         ast_test_status_update(test, "Tested '%s%s' and '%s%s' . . . . . %s\n",
193                 encode1, encode2, decode1, decode2,
194                 okay ? "passed" : "FAILED");
195
196         ast_free(str);
197
198         return okay ? AST_TEST_PASS : AST_TEST_FAIL;
199 }
200
201 static enum ast_test_result_state test_expected_result(struct ast_test *test,
202                 struct ast_channel *c, const char *expression, const char *result)
203 {
204         struct ast_str *str = ast_str_create(16);
205         int okay;
206
207         ast_str_substitute_variables(&str, 0, c, expression);
208
209         okay = !strcmp(ast_str_buffer(str), result);
210         if (!okay) {
211                 ast_test_status_update(test, "'%s' != '%s'\n",
212                                 ast_str_buffer(str), result);
213         }
214         ast_test_status_update(test, "Tested '%s' ('%s') == '%s' . . . . . %s\n",
215                 ast_str_buffer(str), expression, result,
216                 okay ? "passed" : "FAILED");
217
218         ast_free(str);
219
220         return okay ? AST_TEST_PASS : AST_TEST_FAIL;
221 }
222
223 AST_TEST_DEFINE(test_substitution)
224 {
225         struct ast_channel *c;
226         int i;
227         enum ast_test_result_state res = AST_TEST_PASS;
228
229         switch (cmd) {
230         case TEST_INIT:
231                 info->name = "test_substitution";
232                 info->category = "/main/pbx/";
233                 info->summary = "Test variable and function substitution";
234                 info->description =
235                         "This test executes a variety of variable and function substitutions "
236                         "and ensures that the expected results are received.";
237                 return AST_TEST_NOT_RUN;
238         case TEST_EXECUTE:
239                 break;
240         }
241
242         ast_test_status_update(test, "Testing variable substitution ...\n");
243
244         c = ast_channel_alloc(0, 0, "", "", "", "", "", NULL, NULL, 0, "Test/substitution");
245         ast_channel_unlock(c);
246
247 #define TEST(t) if (t == AST_TEST_FAIL) { res = AST_TEST_FAIL; }
248 #if 0
249         /*
250          * We can no longer test the CALLINGPRES value this way because it is now
251          * a calculated value from the name and number presentation information to
252          * get a combined presentation value.
253          */
254         TEST(test_chan_integer(test, c, &c->caller.id.number.presentation, "${CALLINGPRES}"));
255 #endif
256         TEST(test_chan_integer(test, c, &ast_channel_caller(c)->ani2, "${CALLINGANI2}"));
257         TEST(test_chan_integer(test, c, &ast_channel_caller(c)->id.number.plan, "${CALLINGTON}"));
258         TEST(test_chan_integer(test, c, &ast_channel_dialed(c)->transit_network_select, "${CALLINGTNS}"));
259         TEST(test_chan_integer_accessor(test, c, ast_channel_hangupcause_set, "${HANGUPCAUSE}"));
260         TEST(test_chan_integer_accessor(test, c, ast_channel_priority_set, "${PRIORITY}"));
261         TEST(test_chan_string(test, c, ast_channel_context_set, ast_channel_context, "${CONTEXT}"));
262         TEST(test_chan_string(test, c, ast_channel_exten_set, ast_channel_exten, "${EXTEN}"));
263         TEST(test_chan_variable(test, c, "CHANNEL(language)"));
264         TEST(test_chan_variable(test, c, "CHANNEL(musicclass)"));
265         TEST(test_chan_variable(test, c, "CHANNEL(parkinglot)"));
266         TEST(test_chan_variable(test, c, "CALLERID(name)"));
267         TEST(test_chan_variable(test, c, "CURLOPT(proxyuserpwd)"));
268         TEST(test_chan_variable(test, c, "CDR(foo)"));
269         TEST(test_chan_variable(test, c, "ENV(foo)"));
270         TEST(test_chan_variable(test, c, "GLOBAL(foo)"));
271         TEST(test_chan_variable(test, c, "GROUP()"));
272         TEST(test_2way_function(test, c, "${AES_ENCRYPT(abcdefghijklmnop,", ")}", "${AES_DECRYPT(abcdefghijklmnop,", ")}"));
273         TEST(test_2way_function(test, c, "${BASE64_ENCODE(", ")}", "${BASE64_DECODE(", ")}"));
274         pbx_builtin_setvar_helper(c, "foo", "123");
275         pbx_builtin_setvar_helper(c, "bar", "foo");
276         pbx_builtin_setvar_helper(c, "baz", "fo");
277         TEST(test_expected_result(test, c, "${foo}${foo}", "123123"));
278         TEST(test_expected_result(test, c, "A${foo}A${foo}A", "A123A123A"));
279         TEST(test_expected_result(test, c, "A${${bar}}A", "A123A"));
280         TEST(test_expected_result(test, c, "A${${baz}o}A", "A123A"));
281         TEST(test_expected_result(test, c, "A${${baz}o:1}A", "A23A"));
282         TEST(test_expected_result(test, c, "A${${baz}o:1:1}A", "A2A"));
283         TEST(test_expected_result(test, c, "A${${baz}o:1:-1}A", "A2A"));
284         TEST(test_expected_result(test, c, "A${${baz}o:-1:1}A", "A3A"));
285         TEST(test_expected_result(test, c, "A${${baz}o:-2:1}A", "A2A"));
286         TEST(test_expected_result(test, c, "A${${baz}o:-2:-1}A", "A2A"));
287         pbx_builtin_setvar_helper(c, "list1", "ab&cd&ef");
288         TEST(test_expected_result(test, c, "${LISTFILTER(list1,&,ab)}", "cd&ef"));
289         TEST(test_expected_result(test, c, "${LISTFILTER(list1,&,cd)}", "ab&ef"));
290         TEST(test_expected_result(test, c, "${LISTFILTER(list1,&,ef)}", "ab&cd"));
291         TEST(test_expected_result(test, c, "${LISTFILTER(list1,&,gh)}", "ab&cd&ef"));
292         TEST(test_expected_result(test, c, "${LISTFILTER(list1,&,c)}", "ab&cd&ef"));
293         TEST(test_expected_result(test, c, "${LISTFILTER(list1,&,d)}", "ab&cd&ef"));
294         pbx_builtin_setvar_helper(c, "list2", "ab");
295         TEST(test_expected_result(test, c, "${LISTFILTER(list2,&,ab)}", ""));
296         pbx_builtin_setvar_helper(c, "list_empty", "");
297         TEST(test_expected_result(test, c, "${LISTFILTER(list_empty,&,ab)}", ""));
298         TEST(test_expected_result(test, c, "${SHELL(printf '%d' 123)},${SHELL(printf '%d' 456)}", "123,456"));
299         TEST(test_expected_result(test, c, "${foo},${CDR(answer)},${SHELL(printf '%d' 456)}", "123,,456"));
300         TEST(test_expected_result(test, c, "${foo},${CDR(answer,u)},${SHELL(printf '%d' 456)}", "123,0.000000,456"));
301         TEST(test_expected_result(test, c, "${foo},${this_does_not_exist},${THIS_DOES_NOT_EXIST(either)}", "123,,"));
302 #undef TEST
303
304         /* For testing dialplan functions */
305         for (i = 0; ; i++) {
306                 char *cmd = ast_cli_generator("core show function", "", i);
307                 if (cmd == NULL) {
308                         break;
309                 }
310                 if (strcmp(cmd, "CHANNEL") && strcmp(cmd, "CALLERID") && strncmp(cmd, "CURL", 4) &&
311                                 strncmp(cmd, "AES", 3) && strncmp(cmd, "BASE64", 6) &&
312                                 strcmp(cmd, "CDR") && strcmp(cmd, "ENV") && strcmp(cmd, "GLOBAL") &&
313                                 strcmp(cmd, "GROUP") && strcmp(cmd, "CUT") && strcmp(cmd, "LISTFILTER") &&
314                                 strcmp(cmd, "PP_EACH_EXTENSION") && strcmp(cmd, "SET")) {
315                         struct ast_custom_function *acf = ast_custom_function_find(cmd);
316                         if (acf->read && acf->read2) {
317                                 char expression[80];
318                                 snprintf(expression, sizeof(expression), "${%s(foo)}", cmd);
319                                 if (AST_TEST_FAIL == test_chan_function(test, c, expression)) {
320                                         res = AST_TEST_FAIL;
321                                 }
322                         }
323                 }
324                 ast_free(cmd);
325         }
326
327         ast_hangup(c);
328         return res;
329 }
330
331 static int unload_module(void)
332 {
333         AST_TEST_UNREGISTER(test_substitution);
334         return 0;
335 }
336
337 static int load_module(void)
338 {
339         AST_TEST_REGISTER(test_substitution);
340         return AST_MODULE_LOAD_SUCCESS;
341 }
342
343 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Substitution tests");