TESTS: Basic child tests

The child_common.c module had no unit tests, yet we need to amend it.

Reviewed-by: Sumit Bose <sbose@redhat.com>

2 files changed, 233 insertions, 0 deletions

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 <jhrozek@redhat.com>
+
+    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 <http://www.gnu.org/licenses/>.
+*/
+
+#include <sys/types.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <popt.h>
+
+#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 <jhrozek@redhat.com>
+
+    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 <http://www.gnu.org/licenses/>.
+*/
+
+#include <talloc.h>
+#include <tevent.h>
+#include <errno.h>
+#include <popt.h>
+
+#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;
+}