lib/util/tests/util_str_escape.c

   1 /*
   2
   3    util_str_escape testing
   4
   5    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2017
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 */
  20
  21 #include "includes.h"
  22 #include "torture/torture.h"
  23 #include "torture/local/proto.h"
  24 #include "lib/util/util_str_escape.h"
  25
  26 static bool test_log_escape_empty_string(struct torture_context *tctx)
  27 {
  28         char *result = log_escape( tctx, "");
  29         torture_assert_str_equal(tctx, result, "", "Empty string handling");
  30         return true;
  31 }
  32
  33 static bool test_log_escape_null_string(struct torture_context *tctx)
  34 {
  35         char *result = log_escape( tctx, NULL);
  36         torture_assert(tctx, (result == NULL), "Empty string handling");
  37         return true;
  38 }
  39
  40 static bool test_log_escape_plain_string(struct torture_context *tctx)
  41 {
  42         const char *input    = "a plain string with no escapable characters";
  43         const char *expected = "a plain string with no escapable characters";
  44
  45         char *result = log_escape( tctx, input);
  46         torture_assert_str_equal(tctx, result, expected,
  47                                  "Plain string handling");
  48         return true;
  49 }
  50
  51 static bool test_log_escape_string(struct torture_context *tctx)
  52 {
  53         const char *input    = "\a\b\f\n\r\t\v\\\x01";
  54         const char *expected = "\\a\\b\\f\\n\\r\\t\\v\\\\\\x01";
  55
  56         char *result = log_escape( tctx, input);
  57         torture_assert_str_equal(tctx, result, expected,
  58                                  "Escapable characters in string");
  59         return true;
  60 }
  61
  62 static bool test_log_escape_hex_string(struct torture_context *tctx)
  63 {
  64         const char *input    = "\x01\x1F ";
  65         const char *expected = "\\x01\\x1F ";
  66
  67         char *result = log_escape( tctx, input);
  68         torture_assert_str_equal(tctx, result, expected,
  69                                  "hex escaping");
  70         return true;
  71 }
  72 struct torture_suite *torture_local_util_str_escape(TALLOC_CTX *mem_ctx)
  73 {
  74         struct torture_suite *suite = torture_suite_create(mem_ctx,
  75                                                            "util_str_escape");
  76
  77         torture_suite_add_simple_test(suite, "log_escape_empty_string",
  78                                       test_log_escape_empty_string);
  79         torture_suite_add_simple_test(suite, "log_escape_null_string",
  80                                       test_log_escape_null_string);
  81         torture_suite_add_simple_test(suite, "log_escape_plain_string",
  82                                       test_log_escape_plain_string);
  83         torture_suite_add_simple_test(suite, "log_escape_string",
  84                                       test_log_escape_string);
  85         torture_suite_add_simple_test(suite, "log_escape_hex_string",
  86                                       test_log_escape_hex_string);
  87
  88
  89         return suite;
  90 }