From dd8c2ad0efbd6bb4d1f942ce45e3b87d5ecb83b7 Mon Sep 17 00:00:00 2001

From: Jakub Hrozek <jhrozek@redhat.com>

Date: Fri, 3 Oct 2014 16:12:01 +0200

Subject: [PATCH 09/22] TESTS: Unit tests for server_setup

We changed server_setup, so we must make sure the function continues to

work as expected.

Reviewed-by: Pavel Reichl <preichl@redhat.com>

---

 src/tests/cwrap/Makefile.am   |  63 ++++++++++++-

 src/tests/cwrap/test_server.c | 204 ++++++++++++++++++++++++++++++++++++++++++

 2 files changed, 266 insertions(+), 1 deletion(-)

 create mode 100644 src/tests/cwrap/test_server.c

diff --git a/src/tests/cwrap/Makefile.am b/src/tests/cwrap/Makefile.am

index 96b9a52435d9e80bea3ec132cc86c3e6abd0adf2..28d60e7f5bc346543f51740ec4b2781a99a48909 100644

--- a/src/tests/cwrap/Makefile.am

+++ b/src/tests/cwrap/Makefile.am

@@ -3,7 +3,10 @@ AM_CPPFLAGS = \

     -Wall \

     -I$(top_srcdir)/src \

     -I. \

+    -DLOCALEDIR=\"$(localedir)\" \

+    -DVARDIR=\"$(localstatedir)\" \

     $(DBUS_CFLAGS) \

+    $(GLIB2_CFLAGS) \

     $(NULL)

 TESTS_ENVIRONMENT = \

@@ -16,6 +19,20 @@ dist_noinst_SCRIPTS = \

     cwrap_test_setup.sh \

     $(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)

+

 dist_noinst_DATA = \

     group \

     passwd \

@@ -25,7 +42,10 @@ check_PROGRAMS =

 if HAVE_CMOCKA

 if HAVE_NSS_WRAPPER

 if HAVE_UID_WRAPPER

-check_PROGRAMS += become_user-tests

+check_PROGRAMS = \

+	become_user-tests \

+	server-tests \

+	$(NULL)

 endif # HAVE_UID_WRAPPER

 endif # HAVE_NSS_WRAPPER

 endif # HAVE_CMOCKA

@@ -45,4 +65,45 @@ become_user_tests_LDADD = \

     $(abs_top_builddir)/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 \

+    ../../../src/db/sysdb_views.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 \

+    $(abs_top_builddir)/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 0000000000000000000000000000000000000000..26ecfee1e189b6a474ae52fdbfff6b8922b3f0d7

--- /dev/null

+++ b/src/tests/cwrap/test_server.c

@@ -0,0 +1,204 @@

+/*

+    Authors:

+        Jakub Hrozek <jhrozek@redhat.com>

+

+    Copyright (C) 2014 Red Hat

+

+    SSSD tests: Server instantiation

+

+    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;

+    pid_t pid;

+    char *pidfile;

+

+    /* Must root as root, real or fake */

+    assert_int_equal(geteuid(), 0);

+

+    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;

+}

-- 

1.9.3