mirror of
https://github.com/isc-projects/bind9.git
synced 2026-05-22 18:17:05 -04:00
22d664aa15
These functions don't need to be called from multiple places and by making them static we will detect when they are not added to the list functions to be tested.
281 lines
12 KiB
C
281 lines
12 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* SPDX-License-Identifier: MPL-2.0
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
/*! \file */
|
|
|
|
#include <inttypes.h>
|
|
#include <stdbool.h>
|
|
|
|
#include <isc/buffer.h>
|
|
#include <isc/commandline.h>
|
|
#include <isc/file.h>
|
|
#include <isc/hash.h>
|
|
#include <isc/log.h>
|
|
#include <isc/mem.h>
|
|
#include <isc/netmgr.h>
|
|
#include <isc/result.h>
|
|
#include <isc/string.h>
|
|
#include <isc/timer.h>
|
|
#include <isc/util.h>
|
|
#include <isc/uv.h>
|
|
|
|
extern int ncpus;
|
|
extern unsigned int workers;
|
|
extern bool debug;
|
|
|
|
int
|
|
setup_mctx(void **state);
|
|
int
|
|
teardown_mctx(void **state);
|
|
|
|
int
|
|
setup_workers(void **state);
|
|
|
|
int
|
|
setup_loopmgr(void **state);
|
|
int
|
|
teardown_loopmgr(void **state);
|
|
|
|
int
|
|
setup_netmgr(void **state);
|
|
int
|
|
teardown_netmgr(void **state);
|
|
|
|
int
|
|
setup_managers(void **state);
|
|
int
|
|
teardown_managers(void **state);
|
|
|
|
isc_result_t
|
|
file_path_to_groupname(const char *path, char *out, size_t outlen);
|
|
/*%<
|
|
* Example:
|
|
* "/.../tests/dns/dbdiff" -> "dns_dbdiff"
|
|
*
|
|
* Returns ISC_R_SUCCESS on success, ISC_R_NOSPACE if outlen is too small,
|
|
* or other error codes returned by isc_file_splitpath.
|
|
*/
|
|
|
|
#ifndef TESTS_DIR
|
|
#define TESTS_DIR "./"
|
|
#endif
|
|
|
|
/* cmocka<2.0.0 compatibility */
|
|
#ifndef assert_int_in_range
|
|
#define assert_int_in_range(value, min, max) \
|
|
assert_in_range((value), (min), (max))
|
|
#endif
|
|
#ifndef assert_uint_in_range
|
|
#define assert_uint_in_range(value, min, max) \
|
|
assert_in_range((value), (min), (max))
|
|
#endif
|
|
|
|
/* clang-format off */
|
|
/* Copied from cmocka */
|
|
#define ISC_TEST_ENTRY(name) \
|
|
{ #name, run_test_##name, NULL, NULL, NULL },
|
|
#define ISC_TEST_ENTRY_SETUP(name) \
|
|
{ #name, run_test_##name, setup_test_##name, NULL, NULL },
|
|
#define ISC_TEST_ENTRY_TEARDOWN(name) \
|
|
{ #name, run_test_##name, NULL, teardown_test_##name, NULL },
|
|
#define ISC_TEST_ENTRY_SETUP_TEARDOWN(name) \
|
|
{ #name, run_test_##name, setup_test_##name, teardown_test_##name, NULL },
|
|
#define ISC_TEST_ENTRY_CUSTOM(name, setup, teardown) \
|
|
{ #name, run_test_##name, setup, teardown, NULL },
|
|
/* clang-format on */
|
|
|
|
#define ISC_SETUP_TEST_DECLARE(name) \
|
|
int setup_test_##name(void **state ISC_ATTR_UNUSED);
|
|
|
|
#define ISC_RUN_TEST_DECLARE(name) \
|
|
static void run_test_##name(void **state ISC_ATTR_UNUSED);
|
|
|
|
#define ISC_TEARDOWN_TEST_DECLARE(name) \
|
|
int teardown_test_##name(void **state ISC_ATTR_UNUSED)
|
|
|
|
#define ISC_LOOP_TEST_CUSTOM_DECLARE(name, setup, teardown) \
|
|
void run_test_##name(void **state ISC_ATTR_UNUSED); \
|
|
void loop_test_##name(void *arg ISC_ATTR_UNUSED);
|
|
|
|
#define ISC_LOOP_TEST_DECLARE(name) \
|
|
ISC_LOOP_TEST_CUSTOM_DECLARE(name, NULL, NULL)
|
|
|
|
#define ISC_LOOP_TEST_SETUP_DECLARE(name) \
|
|
ISC_LOOP_TEST_CUSTOM_DECLARE(name, setup_loop_##name, NULL)
|
|
|
|
#define ISC_LOOP_TEST_SETUP_TEARDOWN_DECLARE(name) \
|
|
ISC_LOOP_TEST_CUSTOM_DECLARE(name, setup_loop_##name, \
|
|
teardown_loop_##name)
|
|
|
|
#define ISC_LOOP_TEST_TEARDOWN_DECLARE(name) \
|
|
ISC_LOOP_TEST_CUSTOM_DECLARE(name, NULL, teardown_loop_##name)
|
|
|
|
#define ISC_LOOP_SETUP_DECLARE(name) \
|
|
void setup_loop_##name(void *arg ISC_ATTR_UNUSED);
|
|
|
|
#define ISC_SETUP_TEST_IMPL(name) \
|
|
int setup_test_##name(void **state ISC_ATTR_UNUSED); \
|
|
int setup_test_##name(void **state ISC_ATTR_UNUSED)
|
|
|
|
#define ISC_RUN_TEST_IMPL(name) \
|
|
static void run_test_##name(void **state ISC_ATTR_UNUSED); \
|
|
static void run_test_##name(void **state ISC_ATTR_UNUSED)
|
|
|
|
#define ISC_TEARDOWN_TEST_IMPL(name) \
|
|
int teardown_test_##name(void **state ISC_ATTR_UNUSED); \
|
|
int teardown_test_##name(void **state ISC_ATTR_UNUSED)
|
|
|
|
#define ISC_TEST_LIST_START const struct CMUnitTest tests[] = {
|
|
#define ISC_TEST_LIST_END \
|
|
} \
|
|
;
|
|
|
|
#define ISC_LOOP_TEST_CUSTOM_IMPL(name, setup, teardown) \
|
|
static void run_test_##name(void **state ISC_ATTR_UNUSED); \
|
|
static void loop_test_##name(void *arg ISC_ATTR_UNUSED); \
|
|
static void run_test_##name(void **state ISC_ATTR_UNUSED) { \
|
|
isc_job_cb setup_loop = setup; \
|
|
isc_job_cb teardown_loop = teardown; \
|
|
if (setup_loop != NULL) { \
|
|
isc_loop_setup(isc_loop_main(), setup_loop, state); \
|
|
} \
|
|
if (teardown_loop != NULL) { \
|
|
isc_loop_teardown(isc_loop_main(), teardown_loop, \
|
|
state); \
|
|
} \
|
|
isc_loop_setup(isc_loop_main(), loop_test_##name, state); \
|
|
isc_loopmgr_run(); \
|
|
} \
|
|
static void loop_test_##name(void *arg ISC_ATTR_UNUSED)
|
|
|
|
#define ISC_LOOP_TEST_IMPL(name) ISC_LOOP_TEST_CUSTOM_IMPL(name, NULL, NULL)
|
|
|
|
#define ISC_LOOP_TEST_SETUP_IMPL(name) \
|
|
ISC_LOOP_TEST_CUSTOM_IMPL(name, setup_loop_##name, NULL)
|
|
|
|
#define ISC_LOOP_TEST_SETUP_TEARDOWN_IMPL(name) \
|
|
ISC_LOOP_TEST_CUSTOM_IMPL(name, setup_loop_##name, teardown_loop_##name)
|
|
|
|
#define ISC_LOOP_TEST_TEARDOWN_IMPL(name) \
|
|
ISC_LOOP_TEST_CUSTOM_IMPL(name, NULL, teardown_loop_##name)
|
|
|
|
#define ISC_LOOP_SETUP_IMPL(name) \
|
|
void setup_loop_##name(void *arg ISC_ATTR_UNUSED); \
|
|
void setup_loop_##name(void *arg ISC_ATTR_UNUSED)
|
|
|
|
#define ISC_LOOP_TEARDOWN_IMPL(name) \
|
|
void teardown_loop_##name(void *arg ISC_ATTR_UNUSED); \
|
|
void teardown_loop_##name(void *arg ISC_ATTR_UNUSED)
|
|
|
|
#define ISC_TEST_DECLARE(name) void run_test_##name(void **state);
|
|
|
|
#define ISC_TEST_LIST_START const struct CMUnitTest tests[] = {
|
|
#define ISC_TEST_LIST_END \
|
|
} \
|
|
;
|
|
|
|
#define ISC_TEST_MAIN ISC_TEST_MAIN_CUSTOM(NULL, NULL)
|
|
|
|
#define ISC_TEST_MAIN_CUSTOM(setup, teardown) \
|
|
int main(int argc, char **argv) { \
|
|
int c, r; \
|
|
size_t i, j; \
|
|
struct CMUnitTest selected[ARRAY_SIZE(tests)]; \
|
|
\
|
|
signal(SIGPIPE, SIG_IGN); \
|
|
\
|
|
memset(selected, 0, sizeof(selected)); \
|
|
\
|
|
setup_mctx(NULL); \
|
|
setup_workers(NULL); \
|
|
\
|
|
while ((c = isc_commandline_parse(argc, argv, "dlt:")) != \
|
|
-1) { \
|
|
switch (c) { \
|
|
case 'd': \
|
|
debug = true; \
|
|
break; \
|
|
case 'l': \
|
|
for (i = 0; \
|
|
i < (sizeof(tests) / sizeof(tests[0])); \
|
|
i++) \
|
|
{ \
|
|
if (tests[i].name != NULL) { \
|
|
fprintf(stdout, "%s\n", \
|
|
tests[i].name); \
|
|
} \
|
|
} \
|
|
return (0); \
|
|
case 't': \
|
|
for (i = 0; i < ARRAY_SIZE(tests) && \
|
|
tests[i].name != NULL; \
|
|
i++) \
|
|
{ \
|
|
if (strcmp(tests[i].name, \
|
|
isc_commandline_argument) != \
|
|
0) \
|
|
{ \
|
|
continue; \
|
|
} \
|
|
for (j = 0; \
|
|
j < ARRAY_SIZE(selected) && \
|
|
selected[j].name != NULL; \
|
|
j++) \
|
|
{ \
|
|
if (strcmp(tests[j].name, \
|
|
isc_commandline_argument) == \
|
|
0) \
|
|
{ \
|
|
break; \
|
|
} \
|
|
} \
|
|
if (j < ARRAY_SIZE(selected) && \
|
|
selected[j].name == NULL) \
|
|
{ \
|
|
selected[j] = tests[i]; \
|
|
break; \
|
|
} \
|
|
} \
|
|
if (i == ARRAY_SIZE(tests)) { \
|
|
fprintf(stderr, "unknown test '%s'\n", \
|
|
isc_commandline_argument); \
|
|
exit(1); \
|
|
} \
|
|
break; \
|
|
default: \
|
|
fprintf(stderr, "Usage: %s [-dl] [-t test]\n", \
|
|
argv[0]); \
|
|
exit(1); \
|
|
} \
|
|
} \
|
|
\
|
|
char group_name[1024]; \
|
|
isc_result_t res = file_path_to_groupname(argv[0], group_name, \
|
|
sizeof(group_name)); \
|
|
if (res != ISC_R_SUCCESS) { \
|
|
strncpy(group_name, "tests", sizeof(group_name)); \
|
|
} \
|
|
if (selected[0].name != NULL) { \
|
|
r = cmocka_run_group_tests_name(group_name, selected, \
|
|
setup, teardown); \
|
|
} else { \
|
|
r = cmocka_run_group_tests_name(group_name, tests, \
|
|
setup, teardown); \
|
|
} \
|
|
\
|
|
teardown_mctx(NULL); \
|
|
\
|
|
return (r); \
|
|
}
|