tests: refactor create_dom_test_ctx()

Separate the function into more functions as a preparation for
creating a multi domain test environment.

Reviewed-by: Michal Židek <mzidek@redhat.com>

1 files changed, 193 insertions, 61 deletions

diff --git a/src/tests/common_dom.c b/src/tests/common_dom.c
index bc69c5a09..95732b4b8 100644
--- a/src/tests/common_dom.c
+++ b/src/tests/common_dom.c
@@ -25,104 +25,236 @@
 
 #include "tests/common.h"
 
-struct sss_test_ctx *
-create_dom_test_ctx(TALLOC_CTX *mem_ctx,
-                    const char *tests_path,
-                    const char *confdb_path,
-                    const char *domain_name,
-                    const char *id_provider,
-                    struct sss_test_conf_param *params)
+static errno_t
+mock_confdb(TALLOC_CTX *mem_ctx,
+            const char *tests_path,
+            const char *cdb_file,
+            struct confdb_ctx **_cdb)
 {
-    struct sss_test_ctx *test_ctx;
-    size_t i;
-    const char *val[2];
-    val[1] = NULL;
+    TALLOC_CTX *tmp_ctx = NULL;
+    struct confdb_ctx *cdb = NULL;
+    char *cdb_path = NULL;
     errno_t ret;
-    char *dompath;
-
-    test_ctx = create_ev_test_ctx(mem_ctx);
-    if (test_ctx == NULL) {
-        DEBUG(SSSDBG_CRIT_FAILURE, "talloc_zero failed\n");
-        goto fail;
-    }
 
-    test_ctx->confdb_path = talloc_asprintf(test_ctx, "%s/%s",
-                                            tests_path, confdb_path);
-    if (test_ctx->confdb_path == NULL) {
-        DEBUG(SSSDBG_CRIT_FAILURE, "talloc_asprintf failed\n");
-        goto fail;
+    tmp_ctx = talloc_new(NULL);
+    if (tmp_ctx == NULL) {
+        DEBUG(SSSDBG_CRIT_FAILURE, ("talloc_new() failed\n"));
+        return ENOMEM;
     }
 
-    test_ctx->conf_dom_path = talloc_asprintf(test_ctx,
-                                              CONFDB_DOMAIN_PATH_TMPL,
-                                              domain_name);
-    if (test_ctx->conf_dom_path == NULL) {
+    cdb_path = talloc_asprintf(tmp_ctx, "%s/%s", tests_path, cdb_file);
+    if (cdb_path == NULL) {
         DEBUG(SSSDBG_CRIT_FAILURE, "talloc_asprintf failed\n");
-        goto fail;
+        ret = ENOMEM;
+        goto done;
     }
 
-    /* Connect to the conf db */
-    ret = confdb_init(test_ctx, &test_ctx->confdb, test_ctx->confdb_path);
+    /* connect to the confdb */
+    ret = confdb_init(tmp_ctx, &cdb, cdb_path);
     if (ret != EOK) {
         DEBUG(SSSDBG_CRIT_FAILURE, "confdb_init failed: %d\n", ret);
-        goto fail;
+        goto done;
     }
 
-    val[0] = domain_name;
-    ret = confdb_add_param(test_ctx->confdb, true,
-                           "config/sssd", "domains", val);
+    *_cdb = talloc_steal(mem_ctx, cdb);
+    ret = EOK;
+
+done:
+    talloc_free(tmp_ctx);
+    return ret;
+}
+
+static errno_t
+mock_confdb_domain(TALLOC_CTX *mem_ctx,
+                   struct confdb_ctx *cdb,
+                   const char *db_path,
+                   const char *name,
+                   const char *id_provider,
+                   struct sss_test_conf_param *params,
+                   char **_cdb_path)
+{
+    TALLOC_CTX *tmp_ctx = NULL;
+    const char *val[2] = {NULL, NULL};
+    char *cdb_path = NULL;
+    char **array = NULL;
+    char *list = NULL;
+    bool exists = false;
+    errno_t ret;
+    int i;
+
+    tmp_ctx = talloc_new(NULL);
+    if (tmp_ctx == NULL) {
+        DEBUG(SSSDBG_CRIT_FAILURE, ("talloc_new() failed\n"));
+        return ENOMEM;
+    }
+
+    /* get current domain list */
+    ret = confdb_get_string(cdb, tmp_ctx, "config/sssd", "domains",
+                            NULL, &list);
     if (ret != EOK) {
-        DEBUG(SSSDBG_CRIT_FAILURE, "cannot add domain: %d\n", ret);
-        goto fail;
+        goto done;
     }
 
-    dompath = talloc_asprintf(test_ctx, "config/domain/%s", domain_name);
-    if (dompath == NULL) {
-        DEBUG(SSSDBG_CRIT_FAILURE, "talloc_asprintf failed\n");
-        goto fail;
+    /* check if the domain is already in */
+    if (list != NULL) {
+        ret = split_on_separator(tmp_ctx, list, ',', true, true, &array, NULL);
+        if (ret != EOK) {
+            goto done;
+        }
+
+        for (i = 0; array[i] != NULL; i++) {
+            if (strcmp(array[i], name) == 0) {
+                exists = true;
+                break;
+            }
+        }
+    }
+
+    /* add domain to the list of enabled domains */
+    if (!exists) {
+        if (list == NULL) {
+            list = talloc_strdup(tmp_ctx, name);
+        } else {
+            list = talloc_asprintf_append(list, ", %s", name);
+        }
+
+        if (list == NULL) {
+            ret = ENOMEM;
+            goto done;
+        }
+
+        val[0] = list;
+        ret = confdb_add_param(cdb, true, "config/sssd", "domains", val);
+        if (ret != EOK) {
+            DEBUG(SSSDBG_CRIT_FAILURE, "Unable to change domain list [%d]: %s\n",
+                  ret, sss_strerror(ret));
+            goto done;
+        }
+    }
+
+    /* create domain section */
+    cdb_path = talloc_asprintf(tmp_ctx, CONFDB_DOMAIN_PATH_TMPL, name);
+    if (cdb_path == NULL) {
+        ret = ENOMEM;
+        goto done;
     }
 
     val[0] = id_provider;
-    ret = confdb_add_param(test_ctx->confdb, true,
-                           dompath, "id_provider", val);
+    ret = confdb_add_param(cdb, true, cdb_path, "id_provider", val);
     if (ret != EOK) {
-        DEBUG(SSSDBG_CRIT_FAILURE, "cannot add id_provider: %d\n", ret);
-        goto fail;
+        DEBUG(SSSDBG_CRIT_FAILURE, "Unable to add id_provider [%d]: %s\n",
+              ret, sss_strerror(ret));
+        goto done;
     }
 
-    if (params) {
-        for (i=0; params[i].key; i++) {
+    if (params != NULL) {
+        for (i = 0; params[i].key != NULL; i++) {
             val[0] = params[i].value;
-            ret = confdb_add_param(test_ctx->confdb, true,
-                                   dompath, params[i].key,
-                                   val);
+            ret = confdb_add_param(cdb, true, cdb_path, params[i].key, val);
             if (ret != EOK) {
-                DEBUG(SSSDBG_CRIT_FAILURE,
-                      "cannot add parameter %s: %d\n", params[i].key, ret);
-                goto fail;
+                DEBUG(SSSDBG_CRIT_FAILURE, "Unable to add parameter %s [%d]: "
+                      "%s\n", params[i].key, ret, sss_strerror(ret));
+                goto done;
             }
         }
     }
 
-    ret = sssd_domain_init(test_ctx, test_ctx->confdb, domain_name,
-                           tests_path, &test_ctx->dom);
+    if (_cdb_path != NULL) {
+        *_cdb_path = talloc_steal(mem_ctx, cdb_path);
+    }
+
+    ret = EOK;
+
+done:
+    talloc_free(tmp_ctx);
+    return ret;
+}
+
+static errno_t
+mock_domain(TALLOC_CTX *mem_ctx,
+            struct confdb_ctx *cdb,
+            const char *db_path,
+            const char *name,
+            struct sss_domain_info **_domain)
+{
+    struct sss_domain_info *domain = NULL;
+    errno_t ret;
+
+    /* initialize sysdb */
+    ret = sssd_domain_init(mem_ctx, cdb, name, db_path, &domain);
     if (ret != EOK) {
-        DEBUG(SSSDBG_CRIT_FAILURE, "cannot add id_provider: %d\n", ret);
-        goto fail;
+        DEBUG(SSSDBG_CRIT_FAILURE, "sssd_domain_init() of %s failed "
+              "[%d]: %s\n", name, ret, sss_strerror(ret));
+        goto done;
     }
-    test_ctx->sysdb = test_ctx->dom->sysdb;
 
-    /* Init with an AD-style regex to be able to test flat name */
-    ret = sss_names_init_from_args(test_ctx,
+    /* init with an AD-style regex to be able to test flat name */
+    ret = sss_names_init_from_args(domain,
                                    "(((?P<domain>[^\\\\]+)\\\\(?P<name>.+$))|" \
                                    "((?P<name>[^@]+)@(?P<domain>.+$))|" \
                                    "(^(?P<name>[^@\\\\]+)$))",
-                                   "%1$s@%2$s", &test_ctx->nctx);
+                                   "%1$s@%2$s", &domain->names);
     if (ret != EOK) {
         DEBUG(SSSDBG_CRIT_FAILURE, "cannot create names context\n");
+        goto done;
+    }
+
+    if (_domain != NULL) {
+        *_domain = domain;
+    }
+
+    ret = EOK;
+
+done:
+    if (ret != EOK) {
+        talloc_free(domain);
+    }
+    return ret;
+}
+
+struct sss_test_ctx *
+create_dom_test_ctx(TALLOC_CTX *mem_ctx,
+                    const char *tests_path,
+                    const char *confdb_path,
+                    const char *domain_name,
+                    const char *id_provider,
+                    struct sss_test_conf_param *params)
+{
+    struct sss_test_ctx *test_ctx;
+    errno_t ret;
+
+    test_ctx = create_ev_test_ctx(mem_ctx);
+    if (test_ctx == NULL) {
+        DEBUG(SSSDBG_CRIT_FAILURE, "create_ev_test_ctx() failed\n");
+        goto fail;
+    }
+
+    ret = mock_confdb(test_ctx, tests_path, confdb_path, &test_ctx->confdb);
+    if (ret != EOK) {
+        DEBUG(SSSDBG_CRIT_FAILURE, "Unable to initialize confdb [%d]: %s\n",
+              ret, sss_strerror(ret));
+        goto fail;
+    }
+
+    ret = mock_confdb_domain(test_ctx, test_ctx->confdb, tests_path,
+                             domain_name, id_provider, params,
+                             &test_ctx->conf_dom_path);
+    if (ret != EOK) {
+        DEBUG(SSSDBG_CRIT_FAILURE, "Unable to initialize confdb domain "
+              "[%d]: %s\n", ret, sss_strerror(ret));
         goto fail;
     }
-    test_ctx->dom->names = test_ctx->nctx;
+
+    ret = mock_domain(test_ctx, test_ctx->confdb, tests_path, domain_name,
+                      &test_ctx->dom);
+    if (ret != EOK) {
+        DEBUG(SSSDBG_CRIT_FAILURE, "Unable to initialize sss domain "
+              "[%d]: %s\n", ret, sss_strerror(ret));
+        goto fail;
+    }
+
+    test_ctx->sysdb = test_ctx->dom->sysdb;
+    test_ctx->nctx = test_ctx->dom->names;
 
     return test_ctx;