mirror of
https://github.com/isc-projects/bind9.git
synced 2026-06-11 04:30:00 -04:00
837fc382b4
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.
(cherry picked from commit 22d664aa15)
199 lines
8.4 KiB
C
199 lines
8.4 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 <uv.h>
|
|
|
|
#include <isc/buffer.h>
|
|
#include <isc/commandline.h>
|
|
#include <isc/hash.h>
|
|
#include <isc/log.h>
|
|
#include <isc/mem.h>
|
|
#include <isc/netmgr.h>
|
|
#include <isc/print.h>
|
|
#include <isc/result.h>
|
|
#include <isc/string.h>
|
|
#include <isc/task.h>
|
|
#include <isc/timer.h>
|
|
#include <isc/util.h>
|
|
|
|
#include "netmgr_p.h"
|
|
#include "task_p.h"
|
|
#include "timer_p.h"
|
|
|
|
#define CHECK(r) \
|
|
do { \
|
|
result = (r); \
|
|
if (result != ISC_R_SUCCESS) \
|
|
goto cleanup; \
|
|
} while (0)
|
|
|
|
extern isc_mem_t *mctx;
|
|
extern isc_nm_t *netmgr;
|
|
extern isc_taskmgr_t *taskmgr;
|
|
extern isc_timermgr_t *timermgr;
|
|
extern unsigned int workers;
|
|
extern isc_task_t *maintask;
|
|
extern bool debug;
|
|
|
|
#define isc_test_nap(ms) uv_sleep(ms)
|
|
|
|
int
|
|
setup_managers(void **state);
|
|
int
|
|
teardown_managers(void **state);
|
|
|
|
#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 __attribute__((unused)));
|
|
|
|
#define ISC_RUN_TEST_DECLARE(name) \
|
|
static void run_test_##name(void **state __attribute__((unused)));
|
|
|
|
#define ISC_TEARDOWN_TEST_DECLARE(name) \
|
|
int teardown_test_##name(void **state __attribute__((unused)))
|
|
|
|
#define ISC_SETUP_TEST_IMPL(name) \
|
|
int setup_test_##name(void **state __attribute__((unused))); \
|
|
int setup_test_##name(void **state __attribute__((unused)))
|
|
|
|
#define ISC_RUN_TEST_IMPL(name) \
|
|
static void run_test_##name(void **state __attribute__((unused))); \
|
|
static void run_test_##name(void **state __attribute__((unused)))
|
|
|
|
#define ISC_TEARDOWN_TEST_IMPL(name) \
|
|
int teardown_test_##name(void **state __attribute__((unused))); \
|
|
int teardown_test_##name(void **state __attribute__((unused)))
|
|
|
|
#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)); \
|
|
\
|
|
isc_mem_debugging |= ISC_MEM_DEBUGRECORD; \
|
|
isc_mem_create(&mctx); \
|
|
\
|
|
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); \
|
|
} \
|
|
} \
|
|
\
|
|
if (selected[0].name != NULL) { \
|
|
r = cmocka_run_group_tests(selected, setup, teardown); \
|
|
} else { \
|
|
r = cmocka_run_group_tests(tests, setup, teardown); \
|
|
} \
|
|
\
|
|
isc_mem_destroy(&mctx); \
|
|
\
|
|
return (r); \
|
|
}
|