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:
@@ -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));
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user