Merged revisions 378935 via svnmerge from
[asterisk/asterisk.git] / tests / test_xml_escape.c
1 /*
2  * Asterisk -- An open source telephony toolkit.
3  *
4  * Copyright (C) 2013, Digium, Inc.
5  *
6  * David M. Lee, II <dlee@digium.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 /*!
20  * \file
21  * \brief Test ast_xml_escape
22  *
23  * \author\verbatim David M. Lee, II <dlee@digium.com> \endverbatim
24  *
25  * \ingroup tests
26  */
27
28 /*** MODULEINFO
29         <depend>TEST_FRAMEWORK</depend>
30         <support_level>core</support_level>
31  ***/
32
33 #include "asterisk.h"
34
35 ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
36
37 #include "asterisk/utils.h"
38 #include "asterisk/module.h"
39 #include "asterisk/test.h"
40
41 static enum ast_test_result_state test_res = AST_TEST_PASS;
42
43 static void test_xml(struct ast_test *test, const char *input, const char *expected, int max_len, int expected_res)
44 {
45         char actual[256] = "";
46         int res;
47
48         if (max_len == -1) {
49                 max_len = sizeof(actual);
50         }
51
52         res = ast_xml_escape(input, actual, max_len);
53         if (res != expected_res) {
54                 ast_test_status_update(test, "Expected result '%d', got '%d'\n", expected_res, res);
55                 test_res = AST_TEST_FAIL;
56         }
57
58         if (strcmp(expected, actual) != 0) {
59                 ast_test_status_update(test, "Expected output '%s', got '%s'\n", expected, actual);
60                 test_res = AST_TEST_FAIL;
61         }
62 }
63
64 AST_TEST_DEFINE(xml_escape_test)
65 {
66         char *input;
67         char *expected;
68
69         switch (cmd) {
70         case TEST_INIT:
71                 info->name = "xml_escape_test";
72                 info->category = "/main/xml_escape/";
73                 info->summary = "Test XML escaping";
74                 info->description =
75                         "Test XML escaping";
76                 return AST_TEST_NOT_RUN;
77         case TEST_EXECUTE:
78                 break;
79         }
80
81         test_res = AST_TEST_PASS;
82
83         /* happy path */
84         input = "encode me: <&>'\"";
85         expected = "encode me: &lt;&amp;&gt;&apos;&quot;";
86         test_xml(test, input, expected, -1, 0);
87
88         /* size 0 should fail without changing anything */
89         input = "foo";
90         expected = "";
91         test_xml(test, input, expected, 0, -1);
92
93         /* truncate chars */
94         input = "<truncated>";
95         expected = "&lt;trunc";
96         test_xml(test, input, expected, 10, -1);
97
98         /* truncate entity */
99         input = "trunc<";
100         expected = "trunc";
101         test_xml(test, input, expected, 9, -1);
102
103         return test_res;
104 }
105
106 static int unload_module(void)
107 {
108         AST_TEST_UNREGISTER(xml_escape_test);
109         return 0;
110 }
111
112 static int load_module(void)
113 {
114         AST_TEST_REGISTER(xml_escape_test);
115         return AST_MODULE_LOAD_SUCCESS;
116 }
117
118 AST_MODULE_INFO_STANDARD(ASTERISK_GPL_KEY, "Skeleton (sample) Test");