mirror of
https://github.com/isc-projects/bind9.git
synced 2026-02-26 19:41:04 -05:00
04aff208fb
BIND 9.17 changed exit code of skipped test to meet Automake expectations infa505bfb0e. BIND 9.16 was not rewritten to Automake, but for consistency reasons, the same SKIPPED_TEST_EXIT_CODE preprocessor macro is used (though the actual exit code differs from the one in BIND 9.17). (cherry picked from commitfa505bfb0e)
173 lines
4 KiB
C
173 lines
4 KiB
C
/*
|
|
* Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* 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.
|
|
*/
|
|
|
|
#if HAVE_CMOCKA
|
|
|
|
#include <sched.h> /* IWYU pragma: keep */
|
|
#include <setjmp.h>
|
|
#include <stdarg.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#define UNIT_TESTING
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/util.h>
|
|
|
|
#include <dns/peer.h>
|
|
|
|
#include "dnstest.h"
|
|
|
|
static int
|
|
_setup(void **state) {
|
|
isc_result_t result;
|
|
|
|
UNUSED(state);
|
|
|
|
result = dns_test_begin(NULL, false);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
return (0);
|
|
}
|
|
|
|
static int
|
|
_teardown(void **state) {
|
|
UNUSED(state);
|
|
|
|
dns_test_end();
|
|
|
|
return (0);
|
|
}
|
|
|
|
/* Test DSCP set/get functions */
|
|
static void
|
|
dscp(void **state) {
|
|
isc_result_t result;
|
|
isc_netaddr_t netaddr;
|
|
struct in_addr ina;
|
|
dns_peer_t *peer = NULL;
|
|
isc_dscp_t dscp;
|
|
|
|
UNUSED(state);
|
|
|
|
/*
|
|
* Create peer structure for the loopback address.
|
|
*/
|
|
ina.s_addr = INADDR_LOOPBACK;
|
|
isc_netaddr_fromin(&netaddr, &ina);
|
|
result = dns_peer_new(dt_mctx, &netaddr, &peer);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
/*
|
|
* All should be not set on creation.
|
|
* 'dscp' should remain unchanged.
|
|
*/
|
|
dscp = 100;
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
assert_int_equal(dscp, 100);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
assert_int_equal(dscp, 100);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
assert_int_equal(dscp, 100);
|
|
|
|
/*
|
|
* Test that setting query dscp does not affect the other
|
|
* dscp values. 'dscp' should remain unchanged until
|
|
* dns_peer_getquerydscp is called.
|
|
*/
|
|
dscp = 100;
|
|
result = dns_peer_setquerydscp(peer, 1);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
assert_int_equal(dscp, 100);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
assert_int_equal(dscp, 100);
|
|
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(dscp, 1);
|
|
|
|
/*
|
|
* Test that setting notify dscp does not affect the other
|
|
* dscp values. 'dscp' should remain unchanged until
|
|
* dns_peer_getquerydscp is called then should change again
|
|
* on dns_peer_getnotifydscp.
|
|
*/
|
|
dscp = 100;
|
|
result = dns_peer_setnotifydscp(peer, 2);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_NOTFOUND);
|
|
assert_int_equal(dscp, 100);
|
|
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(dscp, 1);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(dscp, 2);
|
|
|
|
/*
|
|
* Test that setting notify dscp does not affect the other
|
|
* dscp values. Check that appropriate values are returned.
|
|
*/
|
|
dscp = 100;
|
|
result = dns_peer_settransferdscp(peer, 3);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(dscp, 1);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(dscp, 2);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
assert_int_equal(result, ISC_R_SUCCESS);
|
|
assert_int_equal(dscp, 3);
|
|
|
|
dns_peer_detach(&peer);
|
|
}
|
|
|
|
int
|
|
main(void) {
|
|
const struct CMUnitTest tests[] = {
|
|
cmocka_unit_test_setup_teardown(dscp, _setup, _teardown),
|
|
};
|
|
|
|
return (cmocka_run_group_tests(tests, NULL, NULL));
|
|
}
|
|
|
|
#else /* HAVE_CMOCKA */
|
|
|
|
#include <stdio.h>
|
|
|
|
int
|
|
main(void) {
|
|
printf("1..0 # Skipped: cmocka not available\n");
|
|
return (SKIPPED_TEST_EXIT_CODE);
|
|
}
|
|
|
|
#endif /* if HAVE_CMOCKA */
|