b8cdd96ff4
ISC_TEST_MAIN now supports turning on/off debugging and
running individual tests.
(cherry picked from commit d8a6ff5c3e)
190 lines
8.1 KiB
C
190 lines
8.1 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
|
|
|
|
/* 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) \
|
|
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) \
|
|
void run_test_##name(void **state __attribute__((unused))); \
|
|
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); \
|
|
}
|