2
0
mirror of https://gitlab.isc.org/isc-projects/bind9 synced 2025-08-31 14:35:26 +00:00

Add unit test for keypair

This commit is contained in:
Mark Andrews
2020-08-01 01:25:37 +10:00
parent cd985d96e3
commit a6357d8b5c

View File

@@ -29,6 +29,8 @@
#include "isctest.h"
static bool debug = false;
static int
_setup(void **state) {
isc_result_t result;
@@ -119,13 +121,121 @@ lex_setline(void **state) {
isc_lex_destroy(&lex);
}
/*%
* keypair is <string>=<qstring>. This has implications double quotes
* in key names.
*/
static void
lex_keypair(void **state) {
isc_buffer_t buf;
isc_lex_t *lex = NULL;
isc_result_t result;
isc_token_t token;
size_t i;
struct {
const char *text;
const char *value;
isc_result_t result;
isc_tokentype_t type;
} tests[] = {
{ "", "", ISC_R_SUCCESS, isc_tokentype_eof },
{ "1234", "1234", ISC_R_SUCCESS, isc_tokentype_string },
{ "1234=", "1234=", ISC_R_SUCCESS, isc_tokentype_vpair },
{ "1234=foo", "1234=foo", ISC_R_SUCCESS, isc_tokentype_vpair },
{ "1234=\"foo", NULL, ISC_R_UNEXPECTEDEND, 0 },
{ "1234=\"foo\"", "1234=foo", ISC_R_SUCCESS,
isc_tokentype_qvpair },
{ "key", "key", ISC_R_SUCCESS, isc_tokentype_string },
{ "\"key=", "\"key=", ISC_R_SUCCESS, isc_tokentype_vpair },
{ "\"key=\"", NULL, ISC_R_UNEXPECTEDEND, 0 },
{ "key=\"\"", "key=", ISC_R_SUCCESS, isc_tokentype_qvpair },
{ "key=\"a b\"", "key=a b", ISC_R_SUCCESS,
isc_tokentype_qvpair },
{ "key=\"a\tb\"", "key=a\tb", ISC_R_SUCCESS,
isc_tokentype_qvpair },
/* double quote not immediately after '=' is not special. */
{ "key=c\"a b\"", "key=c\"a", ISC_R_SUCCESS,
isc_tokentype_vpair },
/* remove special meaning for '=' by escaping */
{ "key\\=", "key\\=", ISC_R_SUCCESS, isc_tokentype_string },
{ "key\\=\"a\"", "key\\=\"a\"", ISC_R_SUCCESS,
isc_tokentype_string },
{ "key\\=\"a \"", "key\\=\"a", ISC_R_SUCCESS,
isc_tokentype_string },
/* vpair with a key of 'key\=' (would need to be deescaped) */
{ "key\\==", "key\\==", ISC_R_SUCCESS, isc_tokentype_vpair },
/* qvpair with a key of 'key\=' (would need to be deescaped) */
{ "key\\==\"\"", "key\\==", ISC_R_SUCCESS,
isc_tokentype_qvpair },
};
UNUSED(state);
for (i = 0; i < ARRAY_SIZE(tests); i++) {
result = isc_lex_create(test_mctx, 1024, &lex);
assert_int_equal(result, ISC_R_SUCCESS);
isc_buffer_constinit(&buf, tests[i].text,
strlen(tests[i].text));
isc_buffer_add(&buf, strlen(tests[i].text));
result = isc_lex_openbuffer(lex, &buf);
assert_int_equal(result, ISC_R_SUCCESS);
result = isc_lex_setsourceline(lex, 100);
assert_int_equal(result, ISC_R_SUCCESS);
memset(&token, 0, sizeof(token));
result = isc_lex_getmastertoken(lex, &token,
isc_tokentype_qvpair, true);
if (debug) {
fprintf(stdout, "# '%s' -> result=%s/%s, type=%u/%u\n",
tests[i].text, isc_result_toid(result),
isc_result_toid(tests[i].result), token.type,
tests[i].type);
}
assert_int_equal(result, tests[i].result);
if (result == ISC_R_SUCCESS) {
switch (token.type) {
case isc_tokentype_string:
case isc_tokentype_qstring:
case isc_tokentype_vpair:
case isc_tokentype_qvpair:
if (debug) {
#define AS_STR(x) (x).value.as_textregion.base
fprintf(stdout, "# value='%s'\n",
AS_STR(token));
}
assert_int_equal(token.type, tests[i].type);
assert_string_equal(AS_STR(token),
tests[i].value);
break;
default:
assert_int_equal(token.type, tests[i].type);
break;
}
}
isc_lex_destroy(&lex);
}
}
int
main(void) {
main(int argc, char *argv[]) {
const struct CMUnitTest tests[] = {
cmocka_unit_test(lex_0xff),
cmocka_unit_test(lex_keypair),
cmocka_unit_test(lex_setline),
};
UNUSED(argv);
if (argc > 1) {
debug = true;
}
return (cmocka_run_group_tests(tests, _setup, _teardown));
}