diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_substitution.c | 241 |
1 files changed, 241 insertions, 0 deletions
diff --git a/tests/test_substitution.c b/tests/test_substitution.c new file mode 100644 index 000000000..c995e8ae6 --- /dev/null +++ b/tests/test_substitution.c @@ -0,0 +1,241 @@ +/* + * Asterisk -- An open source telephony toolkit. + * + * Copyright (C) 2009, Digium, Inc. + * + * Tilghman Lesher <tlesher AT digium DOT com> + * + * See http://www.asterisk.org for more information about + * the Asterisk project. Please do not directly contact + * any of the maintainers of this project for assistance; + * the project provides a web site, mailing lists and IRC + * channels for your use. + * + * This program is free software, distributed under the terms of + * the GNU General Public License Version 2. See the LICENSE file + * at the top of the source tree. + */ + +/*! \file + * + * \brief Substitution Test + * + * \author\verbatim Tilghman Lesher <tlesher AT digium DOT com> \endverbatim + * + * \ingroup tests + */ + +/*** MODULEINFO + <defaultenabled>no</defaultenabled> + ***/ + +#include "asterisk.h" + +ASTERISK_FILE_VERSION(__FILE__, "$Revision$") + +#include "asterisk/file.h" +#include "asterisk/channel.h" +#include "asterisk/pbx.h" +#include "asterisk/module.h" +#include "asterisk/lock.h" +#include "asterisk/app.h" +#include "asterisk/strings.h" +#include "asterisk/stringfields.h" +#include "asterisk/threadstorage.h" +#include "asterisk/cli.h" + +AST_THREADSTORAGE(buf_buf); +AST_THREADSTORAGE(var_buf); + +static void test_chan_integer(int fd, struct ast_channel *c, int *ifield, const char *expression) +{ + int i, okay = 1, value1 = -1, value2 = -1; + char workspace[4096]; + struct ast_str *str = ast_str_thread_get(&buf_buf, 16); + + for (i = 0; i < 256; i++) { + *ifield = i; + ast_str_substitute_variables(&str, 0, c, expression); + pbx_substitute_variables_helper(c, expression, workspace, sizeof(workspace)); + if (sscanf(workspace, "%d", &value1) != 1 || value1 != i || sscanf(ast_str_buffer(str), "%d", &value2) != 1 || value2 != i) { + ast_cli(fd, "%s != %s and/or %d != %d != %d\n", ast_str_buffer(str), workspace, value1, value2, i); + okay = 0; + break; + } + } + ast_cli(fd, "Testing '%s' . . . . . %s\n", expression, okay ? "passed" : "FAILED"); +} + +static void test_chan_string(int fd, struct ast_channel *c, char *cfield, size_t cfieldsize, const char *expression) +{ + const char *values[] = { "one", "three", "reallylongdinosaursoundingthingwithwordsinit" }; + int i, okay = 1; + char workspace[4096]; + struct ast_str *str = ast_str_thread_get(&buf_buf, 16); + + for (i = 0; i < ARRAY_LEN(values); i++) { + ast_copy_string(cfield, values[i], cfieldsize); + ast_str_substitute_variables(&str, 0, c, expression); + pbx_substitute_variables_helper(c, expression, workspace, sizeof(workspace)); + if (strcmp(cfield, ast_str_buffer(str)) != 0 || strcmp(cfield, workspace) != 0) { + ast_cli(fd, "%s != %s != %s\n", cfield, ast_str_buffer(str), workspace); + okay = 0; + break; + } + } + ast_cli(fd, "Testing '%s' . . . . . %s\n", expression, okay ? "passed" : "FAILED"); +} + +static void test_chan_variable(int fd, struct ast_channel *c, const char *varname) +{ + const char *values[] = { "one", "three", "reallylongdinosaursoundingthingwithwordsinit" }; + int i, okay = 1; + char workspace[4096]; + struct ast_str *str = ast_str_thread_get(&buf_buf, 16); + struct ast_str *var = ast_str_thread_get(&var_buf, 16); + + ast_str_set(&var, 0, "${%s}", varname); + for (i = 0; i < ARRAY_LEN(values); i++) { + pbx_builtin_setvar_helper(c, varname, values[i]); + ast_str_substitute_variables(&str, 0, c, ast_str_buffer(var)); + pbx_substitute_variables_helper(c, ast_str_buffer(var), workspace, sizeof(workspace)); + if (strcmp(values[i], ast_str_buffer(str)) != 0 || strcmp(values[i], workspace) != 0) { + ast_cli(fd, "%s != %s != %s\n", values[i], ast_str_buffer(str), workspace); + okay = 0; + break; + } + } + ast_cli(fd, "Testing '%s' . . . . . %s\n", ast_str_buffer(var), okay ? "passed" : "FAILED"); +} + +static void test_chan_function(int fd, struct ast_channel *c, const char *expression) +{ + int okay = 1; + char workspace[4096]; + struct ast_str *str = ast_str_thread_get(&buf_buf, 16); + + ast_str_substitute_variables(&str, 0, c, expression); + pbx_substitute_variables_helper(c, expression, workspace, sizeof(workspace)); + if (strcmp(workspace, ast_str_buffer(str)) != 0) { + ast_cli(fd, "%s != %s\n", ast_str_buffer(str), workspace); + okay = 0; + } + ast_cli(fd, "Testing '%s' . . . . . %s\n", expression, okay ? "passed" : "FAILED"); +} + +static void test_2way_function(int fd, struct ast_channel *c, const char *encode1, const char *encode2, const char *decode1, const char *decode2) +{ + struct ast_str *str = ast_str_thread_get(&buf_buf, 16), *expression = ast_str_alloca(120); + + ast_str_set(&expression, 0, "%s%s%s", encode1, "foobarbaz", encode2); + ast_str_substitute_variables(&str, 0, c, ast_str_buffer(expression)); + ast_str_set(&expression, 0, "%s%s%s", decode1, ast_str_buffer(str), decode2); + ast_str_substitute_variables(&str, 0, c, ast_str_buffer(expression)); + ast_cli(fd, "Testing '%s%s' and '%s%s' . . . . . %s\n", encode1, encode2, decode1, decode2, !strcmp(ast_str_buffer(str), "foobarbaz") ? "passed" : "FAILED"); + if (strcmp(ast_str_buffer(str), "foobarbaz")) { + ast_cli(fd, " '%s' != 'foobarbaz'\n", ast_str_buffer(str)); + } +} + +static void test_expected_result(int fd, struct ast_channel *c, const char *expression, const char *result) +{ + struct ast_str *str = ast_str_thread_get(&buf_buf, 16); + ast_str_substitute_variables(&str, 0, c, expression); + ast_cli(fd, "Testing '%s' ('%s') == '%s' . . . . . %s\n", ast_str_buffer(str), expression, result, !strcmp(ast_str_buffer(str), result) ? "passed" : "FAILED"); +} + +static char *handle_cli_test_substitution(struct ast_cli_entry *e, int cmd, struct ast_cli_args *a) +{ + struct ast_channel *c; + int i; + + switch (cmd) { + case CLI_INIT: + e->command = "test substitution"; + e->usage = "" + "Usage: test substitution\n" + " Test variable and function substitution.\n"; + return NULL; + case CLI_GENERATE: + return NULL; + } + + if (a->argc != e->args) { + return CLI_SHOWUSAGE; + } + + ast_cli(a->fd, "Testing variable substitution ...\n"); + c = ast_channel_alloc(0, 0, "", "", "", "", "", 0, "Test/substitution"); + + test_chan_integer(a->fd, c, &c->cid.cid_pres, "${CALLINGPRES}"); + test_chan_integer(a->fd, c, &c->cid.cid_ani2, "${CALLINGANI2}"); + test_chan_integer(a->fd, c, &c->cid.cid_ton, "${CALLINGTON}"); + test_chan_integer(a->fd, c, &c->cid.cid_tns, "${CALLINGTNS}"); + test_chan_integer(a->fd, c, &c->hangupcause, "${HANGUPCAUSE}"); + test_chan_integer(a->fd, c, &c->priority, "${PRIORITY}"); + test_chan_string(a->fd, c, c->context, sizeof(c->context), "${CONTEXT}"); + test_chan_string(a->fd, c, c->exten, sizeof(c->exten), "${EXTEN}"); + test_chan_variable(a->fd, c, "CHANNEL(language)"); + test_chan_variable(a->fd, c, "CHANNEL(musicclass)"); + test_chan_variable(a->fd, c, "CHANNEL(parkinglot)"); + test_chan_variable(a->fd, c, "CALLERID(name)"); + test_chan_variable(a->fd, c, "CURLOPT(proxyuserpwd)"); + test_chan_variable(a->fd, c, "CDR(foo)"); + test_chan_variable(a->fd, c, "ENV(foo)"); + test_chan_variable(a->fd, c, "GLOBAL(foo)"); + test_chan_variable(a->fd, c, "GROUP()"); + test_2way_function(a->fd, c, "${AES_ENCRYPT(abcdefghijklmnop,", ")}", "${AES_DECRYPT(abcdefghijklmnop,", ")}"); + test_2way_function(a->fd, c, "${BASE64_ENCODE(", ")}", "${BASE64_DECODE(", ")}"); + pbx_builtin_setvar_helper(c, "foo", "123"); + pbx_builtin_setvar_helper(c, "bar", "foo"); + pbx_builtin_setvar_helper(c, "baz", "fo"); + test_expected_result(a->fd, c, "${foo}${foo}", "123123"); + test_expected_result(a->fd, c, "A${foo}A${foo}A", "A123A123A"); + test_expected_result(a->fd, c, "A${${bar}}A", "A123A"); + test_expected_result(a->fd, c, "A${${baz}o}A", "A123A"); + test_expected_result(a->fd, c, "A${${baz}o:1}A", "A23A"); + test_expected_result(a->fd, c, "A${${baz}o:1:1}A", "A2A"); + test_expected_result(a->fd, c, "A${${baz}o:1:-1}A", "A2A"); + test_expected_result(a->fd, c, "A${${baz}o:-1:1}A", "A3A"); + test_expected_result(a->fd, c, "A${${baz}o:-2:1}A", "A2A"); + test_expected_result(a->fd, c, "A${${baz}o:-2:-1}A", "A2A"); + + /* For testing dialplan functions */ + for (i = 0; ; i++) { + char *cmd = ast_cli_generator("core show function", "", i); + if (cmd == NULL) { + break; + } + if (strcmp(cmd, "CHANNEL") && strcmp(cmd, "CALLERID") && strcmp(cmd, "CURLOPT") && strncmp(cmd, "AES", 3) && strncmp(cmd, "BASE64", 6) && strcmp(cmd, "CDR") && strcmp(cmd, "ENV") && strcmp(cmd, "GLOBAL") && strcmp(cmd, "GROUP") && strcmp(cmd, "CUT") && strcmp(cmd, "LISTFILTER") && strcmp(cmd, "PP_EACH_EXTENSION") && strcmp(cmd, "SET")) { + struct ast_custom_function *acf = ast_custom_function_find(cmd); + if (acf->read && acf->read2) { + char expression[80]; + snprintf(expression, sizeof(expression), "${%s(foo)}", cmd); + test_chan_function(a->fd, c, expression); + } + } + ast_free(cmd); + } + + ast_hangup(c); + + return CLI_SUCCESS; +} + +static struct ast_cli_entry cli_substitution[] = { + AST_CLI_DEFINE(handle_cli_test_substitution, "Test variable substitution"), +}; + +static int unload_module(void) +{ + ast_cli_unregister_multiple(cli_substitution, ARRAY_LEN(cli_substitution)); + return 0; +} + +static int load_module(void) +{ + ast_cli_register_multiple(cli_substitution, ARRAY_LEN(cli_substitution)); + return AST_MODULE_LOAD_SUCCESS; +} + +AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Substitution tests"); |