TESTS: Unit tests for server_setup

FIXME - check the PID was created as the user in question.

2 files changed, 266 insertions, 0 deletions

diff --git a/src/tests/cwrap/Makefile.am b/src/tests/cwrap/Makefile.am
index 3bf2ff9f8..2297dc8b0 100644
--- a/src/tests/cwrap/Makefile.am
+++ b/src/tests/cwrap/Makefile.am
@@ -3,9 +3,26 @@ AM_CPPFLAGS = \
     -Wall \
     -I$(top_srcdir)/src \
     -I. \
+    -DLOCALEDIR=\"$(localedir)\" \
+    -DVARDIR=\"$(localstatedir)\" \
     $(DBUS_CFLAGS) \
+    $(GLIB2_CFLAGS) \
     $(NULL)
 
+SSSD_LIBS = \
+    $(TALLOC_LIBS) \
+    $(TEVENT_LIBS) \
+    $(POPT_LIBS) \
+    $(LDB_LIBS) \
+    $(DBUS_LIBS) \
+    $(PCRE_LIBS) \
+    $(INI_CONFIG_LIBS) \
+    $(COLLECTION_LIBS) \
+    $(DHASH_LIBS) \
+    $(SSS_CRYPT_LIBS) \
+    $(OPENLDAP_LIBS) \
+    $(TDB_LIBS)
+
 AM_TESTS_ENVIRONMENT = \
     SSSD_SRCDIR=$(top_srcdir) \
 	. $(top_srcdir)/src/tests/cwrap/cwrap_test_setup.sh; \
@@ -13,6 +30,7 @@ AM_TESTS_ENVIRONMENT = \
 
 check_PROGRAMS = \
 	become_user-tests \
+	server-tests \
 	$(NULL)
 
 check_LTLIBRARIES = \
@@ -42,4 +60,44 @@ become_user_tests_LDADD = \
     libsss_test_common.la \
     $(NULL)
 
+server_tests_SOURCES = \
+    test_server.c \
+    ../../../src/util/server.c \
+    ../../../src/util/become_user.c \
+    ../../../src/util/backup_file.c \
+    ../../../src/util/domain_info_utils.c \
+    ../../../src/util/atomic_io.c \
+    ../../../src/util/signal.c \
+    ../../../src/util/util.c \
+    ../../../src/util/strtonum.c \
+    ../../../src/util/util_errors.c \
+    ../../../src/util/safe-format-string.c \
+    ../../../src/util/sss_tc_utf8.c \
+    ../../../src/util/sss_utf8.c \
+    ../../../src/util/usertools.c \
+    ../../../src/confdb/confdb.c \
+    ../../../src/db/sysdb.c \
+    ../../../src/db/sysdb_upgrade.c \
+    ../../../src/db/sysdb_ops.c \
+    ../../../src/db/sysdb_search.c \
+    ../../../src/db/sysdb_autofs.c \
+    ../../../src/db/sysdb_services.c \
+    $(NULL)
+server_tests_CFLAGS = \
+    $(AM_CFLAGS) \
+    $(LIBCAPNG_CFLAGS) \
+    -DTEST_DB_PATH=\"server_tests\" \
+    -DTEST_PID_PATH=\"server_tests\" \
+    -DUNIT_TESTING \
+    $(NULL)
+server_tests_LDADD = \
+    $(CMOCKA_LIBS) \
+    $(LIBCAPNG_LIBS) \
+    $(UNICODE_LIBS) \
+    $(SSSD_LIBS) \
+    $(abs_top_builddir)/libsss_debug.la \
+    $(abs_top_builddir)/libsss_crypt.la \
+    libsss_test_common.la \
+    $(NULL)
+
 tests: $(check_PROGRAMS)
diff --git a/src/tests/cwrap/test_server.c b/src/tests/cwrap/test_server.c
new file mode 100644
index 000000000..b49b72a45
--- /dev/null
+++ b/src/tests/cwrap/test_server.c
@@ -0,0 +1,208 @@
+/*
+    Authors:
+        Jakub Hrozek <jhrozek@redhat.com>
+
+    Copyright (C) 2014 Red Hat
+
+    SSSD tests: User switching
+
+    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 <sys/stat.h>
+#include <fcntl.h>
+
+#include <popt.h>
+#include "util/util.h"
+#include "tests/cmocka/common_mock.h"
+
+static void wait_for_fg_server(pid_t pid)
+{
+    pid_t wpid;
+    int status;
+
+    assert_int_not_equal(pid, -1);
+
+    wpid = waitpid(pid, &status, 0);
+    assert_int_equal(wpid, pid);
+    assert_true(WIFEXITED(status));
+    assert_int_equal(WEXITSTATUS(status), 0);
+}
+
+static void wait_for_bg_server(const char *pidfile)
+{
+    int fd;
+    long tmp;
+    char buf[16];
+    pid_t pid;
+    int ret;
+    int count;
+
+    count = 0;
+    do {
+        struct stat sb;
+
+        count++;
+        if (count > 100) {
+            break;
+        }
+
+        ret = stat(pidfile, &sb);
+        usleep(50);
+    } while (ret != 0);
+
+    /* read the pidfile */
+    fd = open(pidfile, O_RDONLY);
+    assert_false(fd < 0);
+
+    ret = read(fd, buf, sizeof(buf));
+    close(fd);
+    assert_false(ret <= 0);
+
+    buf[sizeof(buf) - 1] = '\0';
+
+    tmp = strtol(buf, NULL, 10);
+    assert_false(tmp == 0 || tmp > 0xFFFF || errno == ERANGE);
+
+    pid = (pid_t) (tmp & 0xFFFF);
+
+    /* Make sure the daemon goes away! */
+    ret = kill(pid, SIGTERM);
+    fprintf(stderr, "killing %u\n", pid);
+    assert_true(ret == 0);
+
+    unlink(pidfile);
+}
+
+void test_run_as_root_fg(void **state)
+{
+    int ret;
+    struct main_context *main_ctx;
+    pid_t pid;
+
+    /* Must root as root, real or fake */
+    assert_int_equal(geteuid(), 0);
+
+    pid = fork();
+    if (pid == 0) {
+        ret = server_setup(__FUNCTION__, 0, 0, 0,
+                           __FUNCTION__, &main_ctx);
+        assert_int_equal(ret, 0);
+        exit(0);
+    }
+    wait_for_fg_server(pid);
+}
+
+void test_run_as_sssd_fg(void **state)
+{
+    int ret;
+    struct main_context *main_ctx;
+    struct passwd *sssd;
+    pid_t pid;
+
+    /* Must root as root, real or fake */
+    assert_int_equal(geteuid(), 0);
+
+    sssd = getpwnam("sssd");
+    assert_non_null(sssd);
+
+    pid = fork();
+    if (pid == 0) {
+        ret = server_setup(__FUNCTION__, 0, sssd->pw_uid, sssd->pw_gid,
+                           __FUNCTION__, &main_ctx);
+        assert_int_equal(ret, 0);
+        exit(0);
+    }
+    wait_for_fg_server(pid);
+}
+
+void test_run_as_root_daemon(void **state)
+{
+    int ret;
+    struct main_context *main_ctx;
+    struct passwd *sssd;
+    pid_t pid;
+    char *pidfile;
+
+    /* Must root as root, real or fake */
+    assert_int_equal(geteuid(), 0);
+
+    sssd = getpwnam("sssd");
+    assert_non_null(sssd);
+
+    pidfile = talloc_asprintf(NULL, "%s/%s.pid", TEST_PID_PATH, __FUNCTION__);
+
+    /* Make sure there are no leftovers */
+    unlink(pidfile);
+
+    pid = fork();
+    if (pid == 0) {
+        ret = server_setup(__FUNCTION__, FLAGS_PID_FILE,
+                           0, 0, __FUNCTION__, &main_ctx);
+        assert_int_equal(ret, 0);
+
+        server_loop(main_ctx);
+        exit(0);
+    }
+
+    wait_for_bg_server(pidfile);
+    talloc_free(pidfile);
+}
+
+int main(int argc, const char *argv[])
+{
+    poptContext pc;
+    int opt;
+    int rv;
+    struct poptOption long_options[] = {
+        POPT_AUTOHELP
+        SSSD_DEBUG_OPTS
+        POPT_TABLEEND
+    };
+
+    const UnitTest tests[] = {
+        unit_test(test_run_as_root_fg),
+        unit_test(test_run_as_sssd_fg),
+        unit_test(test_run_as_root_daemon),
+    };
+
+    /* 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);
+
+    /* Even though normally the tests should clean up after themselves
+     * they might not after a failed run. Remove the old db to be sure */
+    tests_set_cwd();
+    test_dom_suite_cleanup(TEST_DB_PATH, CONFDB_FILE, NULL);
+    test_dom_suite_setup(TEST_DB_PATH);
+
+    rv = run_tests(tests);
+    test_dom_suite_cleanup(TEST_DB_PATH, CONFDB_FILE, NULL);
+
+    return rv;
+}