From 9d501b05b47a63e29c586c6dd3d40ac615a98886 Mon Sep 17 00:00:00 2001 From: Jakub Hrozek Date: Mon, 1 Dec 2014 12:04:25 +0100 Subject: TESTS: Basic child tests The child_common.c module had no unit tests, yet we need to amend it. Reviewed-by: Sumit Bose --- src/tests/cmocka/test_child.c | 83 +++++++++++++++++++ src/tests/cmocka/test_child_common.c | 150 +++++++++++++++++++++++++++++++++++ 2 files changed, 233 insertions(+) create mode 100644 src/tests/cmocka/test_child.c create mode 100644 src/tests/cmocka/test_child_common.c (limited to 'src') diff --git a/src/tests/cmocka/test_child.c b/src/tests/cmocka/test_child.c new file mode 100644 index 000000000..a857afce1 --- /dev/null +++ b/src/tests/cmocka/test_child.c @@ -0,0 +1,83 @@ +/* + SSSD + + Tests -- a simple test process that echoes input back + + Authors: + Jakub Hrozek + + Copyright (C) 2014 Red Hat + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . +*/ + +#include +#include +#include +#include + +#include "util/util.h" +#include "util/child_common.h" + +int main(int argc, const char *argv[]) +{ + int opt; + int debug_fd = -1; + poptContext pc; + const char *action; + ssize_t len = 0; + ssize_t written; + errno_t ret; + uint8_t *buf[IN_BUF_SIZE]; + uid_t uid; + gid_t gid; + + struct poptOption long_options[] = { + POPT_AUTOHELP + {"debug-level", 'd', POPT_ARG_INT, &debug_level, 0, + _("Debug level"), NULL}, + {"debug-timestamps", 0, POPT_ARG_INT, &debug_timestamps, 0, + _("Add debug timestamps"), NULL}, + {"debug-microseconds", 0, POPT_ARG_INT, &debug_microseconds, 0, + _("Show timestamps with microseconds"), NULL}, + {"debug-fd", 0, POPT_ARG_INT, &debug_fd, 0, + _("An open file descriptor for the debug logs"), NULL}, + {"debug-to-stderr", 0, POPT_ARG_NONE | POPT_ARGFLAG_DOC_HIDDEN, &debug_to_stderr, 0, \ + _("Send the debug output to stderr directly."), NULL }, \ + SSSD_SERVER_OPTS(uid, gid) + POPT_TABLEEND + }; + + /* Set debug level to invalid value so we can decide if -d 0 was used. */ + debug_level = SSSDBG_INVALID; + + pc = poptGetContext(argv[0], argc, argv, long_options, 0); + while((opt = poptGetNextOpt(pc)) != -1) { + switch(opt) { + default: + fprintf(stderr, "\nInvalid option %s: %s\n\n", + poptBadOption(pc, 0), poptStrerror(opt)); + poptPrintUsage(pc, stderr, 0); + _exit(1); + } + } + + DEBUG(SSSDBG_TRACE_FUNC, "test_child completed successfully\n"); + _exit(0); + +fail: + DEBUG(SSSDBG_TRACE_FUNC, "test_child completed successfully\n"); + close(STDOUT_FILENO); + _exit(-1); +} diff --git a/src/tests/cmocka/test_child_common.c b/src/tests/cmocka/test_child_common.c new file mode 100644 index 000000000..f2cd8c008 --- /dev/null +++ b/src/tests/cmocka/test_child_common.c @@ -0,0 +1,150 @@ +/* + Authors: + Jakub Hrozek + + Copyright (C) 2014 Red Hat + + SSSD tests: Child handlers + + This program is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . +*/ + +#include +#include +#include +#include + +#include "util/child_common.h" +#include "tests/cmocka/common_mock.h" + +#define TEST_BIN "test-child" +#define ECHO_STR "Hello child" + +struct child_test_ctx { + int pipefd_to_child[2]; + int pipefd_from_child[2]; + + struct sss_test_ctx *test_ctx; +}; + +void child_test_setup(void **state) +{ + struct child_test_ctx *child_tctx; + errno_t ret; + + check_leaks_push(global_talloc_context); + child_tctx = talloc(global_talloc_context, struct child_test_ctx); + assert_non_null(child_tctx); + + child_tctx->test_ctx = create_ev_test_ctx(child_tctx); + assert_non_null(child_tctx->test_ctx); + + ret = pipe(child_tctx->pipefd_from_child); + assert_int_not_equal(ret, -1); + DEBUG(SSSDBG_TRACE_LIBS, "from_child: %d:%d\n", + child_tctx->pipefd_from_child[0], + child_tctx->pipefd_from_child[1]); + + ret = pipe(child_tctx->pipefd_to_child); + assert_int_not_equal(ret, -1); + DEBUG(SSSDBG_TRACE_LIBS, "to_child: %d:%d\n", + child_tctx->pipefd_to_child[0], + child_tctx->pipefd_to_child[1]); + + *state = child_tctx; +} + +void child_test_teardown(void **state) +{ + struct child_test_ctx *child_tctx = talloc_get_type(*state, + struct child_test_ctx); + + talloc_free(child_tctx); + check_leaks_pop(global_talloc_context); +} + +/* Just make sure the exec works. The child does nothing but exits */ +void test_exec_child(void **state) +{ + errno_t ret; + pid_t child_pid; + int status; + struct child_test_ctx *child_tctx = talloc_get_type(*state, + struct child_test_ctx); + + child_pid = fork(); + assert_int_not_equal(child_pid, -1); + if (child_pid == 0) { + ret = exec_child(child_tctx, + child_tctx->pipefd_to_child, + child_tctx->pipefd_from_child, + CHILD_DIR"/"TEST_BIN, 2); + assert_int_equal(ret, EOK); + } else { + do { + errno = 0; + ret = waitpid(child_pid, &status, 0); + } while (ret == -1 && errno == EINTR); + + if (ret > 0) { + ret = EIO; + if (WIFEXITED(status)) { + ret = WEXITSTATUS(status); + assert_int_equal(ret, 0); + } + } else { + DEBUG(SSSDBG_FUNC_DATA, + "Failed to wait for children %d\n", child_pid); + ret = EIO; + } + } +} + +int main(int argc, const char *argv[]) +{ + int rv; + poptContext pc; + int opt; + struct poptOption long_options[] = { + POPT_AUTOHELP + SSSD_DEBUG_OPTS + POPT_TABLEEND + }; + + const UnitTest tests[] = { + unit_test_setup_teardown(test_exec_child, + child_test_setup, + child_test_teardown), + }; + + /* Set debug level to invalid value so we can deside if -d 0 was used. */ + debug_level = SSSDBG_INVALID; + + pc = poptGetContext(argv[0], argc, argv, long_options, 0); + while((opt = poptGetNextOpt(pc)) != -1) { + switch(opt) { + default: + fprintf(stderr, "\nInvalid option %s: %s\n\n", + poptBadOption(pc, 0), poptStrerror(opt)); + poptPrintUsage(pc, stderr, 0); + return 1; + } + } + poptFreeContext(pc); + + DEBUG_CLI_INIT(debug_level); + + rv = run_tests(tests); + return rv; +} -- cgit