148 lines
3.9 KiB
C
148 lines
3.9 KiB
C
/* $NetBSD: peer_test.c,v 1.1.1.3 2014/12/10 03:34:43 christos Exp $ */
|
|
|
|
/*
|
|
* Copyright (C) 2014 Internet Systems Consortium, Inc. ("ISC")
|
|
*
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
|
|
* REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
* AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
* INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
* LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
|
|
* OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
* PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
/*! \file */
|
|
|
|
#include <config.h>
|
|
|
|
#include <atf-c.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <dns/peer.h>
|
|
|
|
#include "dnstest.h"
|
|
|
|
/*
|
|
* Individual unit tests
|
|
*/
|
|
ATF_TC(dscp);
|
|
ATF_TC_HEAD(dscp, tc) {
|
|
atf_tc_set_md_var(tc, "descr",
|
|
"Test DSCP set/get functions");
|
|
}
|
|
ATF_TC_BODY(dscp, tc) {
|
|
isc_result_t result;
|
|
isc_netaddr_t netaddr;
|
|
struct in_addr ina;
|
|
dns_peer_t *peer = NULL;
|
|
isc_dscp_t dscp;
|
|
|
|
result = dns_test_begin(NULL, ISC_TRUE);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
|
|
/*
|
|
* Create peer structure for the loopback address.
|
|
*/
|
|
ina.s_addr = INADDR_LOOPBACK;
|
|
isc_netaddr_fromin(&netaddr, &ina);
|
|
result = dns_peer_new(mctx, &netaddr, &peer);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
|
|
/*
|
|
* All should be not set on creation.
|
|
* 'dscp' should remain unchanged.
|
|
*/
|
|
dscp = 100;
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_NOTFOUND);
|
|
ATF_REQUIRE_EQ(dscp, 100);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_NOTFOUND);
|
|
ATF_REQUIRE_EQ(dscp, 100);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_NOTFOUND);
|
|
ATF_REQUIRE_EQ(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);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_NOTFOUND);
|
|
ATF_REQUIRE_EQ(dscp, 100);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_NOTFOUND);
|
|
ATF_REQUIRE_EQ(dscp, 100);
|
|
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
ATF_REQUIRE_EQ(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);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_NOTFOUND);
|
|
ATF_REQUIRE_EQ(dscp, 100);
|
|
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
ATF_REQUIRE_EQ(dscp, 1);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
ATF_REQUIRE_EQ(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);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
|
|
result = dns_peer_getquerydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
ATF_REQUIRE_EQ(dscp, 1);
|
|
|
|
result = dns_peer_getnotifydscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
ATF_REQUIRE_EQ(dscp, 2);
|
|
|
|
result = dns_peer_gettransferdscp(peer, &dscp);
|
|
ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
|
|
ATF_REQUIRE_EQ(dscp, 3);
|
|
|
|
dns_peer_detach(&peer);
|
|
dns_test_end();
|
|
}
|
|
|
|
/*
|
|
* Main
|
|
*/
|
|
ATF_TP_ADD_TCS(tp) {
|
|
ATF_TP_ADD_TC(tp, dscp);
|
|
return (atf_no_error());
|
|
}
|