139 lines
2.9 KiB
C
139 lines
2.9 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 http://mozilla.org/MPL/2.0/.
|
|
*
|
|
* See the COPYRIGHT file distributed with this work for additional
|
|
* information regarding copyright ownership.
|
|
*/
|
|
|
|
#if HAVE_CMOCKA
|
|
|
|
#include <stdarg.h>
|
|
#include <stdbool.h>
|
|
#include <stddef.h>
|
|
#include <setjmp.h>
|
|
#include <cmocka.h>
|
|
|
|
#include <isc/mem.h>
|
|
#include <isc/result.h>
|
|
|
|
#include <dns/tkey.h>
|
|
|
|
#if LD_WRAP
|
|
static isc_mem_t mock_mctx = {
|
|
.impmagic = 0,
|
|
.magic = ISCAPI_MCTX_MAGIC,
|
|
.methods = NULL
|
|
};
|
|
|
|
static void *
|
|
__wrap_isc__mem_get(isc_mem_t *mctx __attribute__ ((unused)),
|
|
size_t size)
|
|
{
|
|
bool has_enough_memory = mock_type(bool);
|
|
if (!has_enough_memory) {
|
|
return (NULL);
|
|
}
|
|
return (malloc(size));
|
|
}
|
|
|
|
static void
|
|
__wrap_isc__mem_put(isc_mem_t *ctx0 __attribute__ ((unused)),
|
|
void *ptr,
|
|
size_t size __attribute__ ((unused)))
|
|
{
|
|
free(ptr);
|
|
}
|
|
|
|
static void
|
|
__wrap_isc_mem_attach(isc_mem_t *source0, isc_mem_t **targetp) {
|
|
*targetp = source0;
|
|
}
|
|
|
|
static void
|
|
__wrap_isc_mem_detach(isc_mem_t **ctxp) {
|
|
*ctxp = NULL;
|
|
}
|
|
|
|
static int
|
|
_setup(void **state) {
|
|
dns_tkeyctx_t *tctx = NULL;
|
|
will_return(__wrap_isc__mem_get, true);
|
|
if (dns_tkeyctx_create(&mock_mctx, &tctx) != ISC_R_SUCCESS) {
|
|
return (-1);
|
|
}
|
|
*state = tctx;
|
|
return (0);
|
|
}
|
|
|
|
static int
|
|
_teardown(void **state) {
|
|
dns_tkeyctx_t *tctx = *state;
|
|
if (tctx != NULL) {
|
|
dns_tkeyctx_destroy(&tctx);
|
|
}
|
|
return (0);
|
|
}
|
|
|
|
static void
|
|
dns_tkeyctx_create_test(void **state) {
|
|
dns_tkeyctx_t *tctx;
|
|
|
|
tctx = NULL;
|
|
will_return(__wrap_isc__mem_get, false);
|
|
assert_int_equal(dns_tkeyctx_create(&mock_mctx, &tctx), ISC_R_NOMEMORY);
|
|
|
|
tctx = NULL;
|
|
will_return(__wrap_isc__mem_get, true);
|
|
assert_int_equal(dns_tkeyctx_create(&mock_mctx, &tctx), ISC_R_SUCCESS);
|
|
*state = tctx;
|
|
}
|
|
|
|
static void
|
|
dns_tkeyctx_destroy_test(void **state) {
|
|
dns_tkeyctx_t *tctx = *state;
|
|
|
|
assert_non_null(tctx);
|
|
dns_tkeyctx_destroy(&tctx);
|
|
}
|
|
|
|
#endif /* LD_WRAP */
|
|
|
|
int
|
|
main(void) {
|
|
#if LD_WRAP
|
|
const struct CMUnitTest tkey_tests[] = {
|
|
cmocka_unit_test_teardown(dns_tkeyctx_create_test, _teardown),
|
|
cmocka_unit_test_setup(dns_tkeyctx_destroy_test, _setup),
|
|
#if 0 /* not yet */
|
|
cmocka_unit_test(dns_tkey_processquery_test),
|
|
cmocka_unit_test(dns_tkey_builddhquery_test),
|
|
cmocka_unit_test(dns_tkey_buildgssquery_test),
|
|
cmocka_unit_test(dns_tkey_builddeletequery_test),
|
|
cmocka_unit_test(dns_tkey_processdhresponse_test),
|
|
cmocka_unit_test(dns_tkey_processgssresponse_test),
|
|
cmocka_unit_test(dns_tkey_processdeleteresponse_test),
|
|
cmocka_unit_test(dns_tkey_gssnegotiate_test),
|
|
#endif
|
|
};
|
|
return (cmocka_run_group_tests(tkey_tests, NULL, NULL));
|
|
#else
|
|
print_message("1..0 # Skip tkey_test requires LD_WRAP\n");
|
|
#endif /* LD_WRAP */
|
|
}
|
|
|
|
#else
|
|
|
|
#include <stdio.h>
|
|
|
|
int
|
|
main(void) {
|
|
printf("1..0 # Skipped: cmocka not available\n");
|
|
return (0);
|
|
}
|
|
|
|
#endif
|