lib-storage: test-mail-storage - Add framework for doing storage tests

diff --git a/src/lib-storage/test-mail-storage.c b/src/lib-storage/test-mail-storage.c
index 5eb8520dbcb6c7eb2ffe5465ca149db1e5149887..5af1fac7736cb49a307c95232a1d62c245de645d 100644 (file)
--- a/src/lib-storage/test-mail-storage.c
+++ b/src/lib-storage/test-mail-storage.c
@@ -1,7 +1,14 @@
 /* Copyright (c) 2017-2018 Dovecot authors, see the included COPYING file */
 
 #include "lib.h"
+#include "ioloop.h"
+#include "mkdir-parents.h"
+#include "unlink-directory.h"
+#include "hex-binary.h"
+#include "randgen.h"
 #include "test-common.h"
+#include "master-service.h"
+#include "mail-storage-service.h"
 #include "mail-storage-private.h"
 
 static void test_init_storage(struct mail_storage *storage_r)
@@ -22,6 +29,15 @@ static void test_deinit_storage(struct mail_storage *storage)
        event_unref(&storage->user->event);
 }
 
+struct test_mail_storage_ctx {
+       pool_t pool;
+       struct mail_storage_service_ctx *storage_service;
+       struct mail_user *user;
+       struct mail_storage_service_user *service_user;
+       struct ioloop *ioloop;
+       const char *mail_home;
+};
+
 static void test_mail_storage_errors(void)
 {
        struct mail_storage storage;
@@ -201,13 +217,127 @@ static void test_mail_storage_last_error_push_pop(void)
        test_end();
 }
 
-int main(void)
+static void test_mail_init(struct test_mail_storage_ctx *ctx)
+{
+       const char *error;
+       char path_buf[4096];
+       unsigned char rand[4];
+
+       ctx->pool = pool_allocfree_create("test pool");
+
+       if (getcwd(path_buf, sizeof(path_buf)) == NULL)
+               i_fatal("getcwd() failed: %m");
+
+       random_fill(rand, sizeof(rand));
+       ctx->mail_home = p_strdup_printf(ctx->pool, "%s/.test-dir%s/", path_buf,
+                                        binary_to_hex(rand, sizeof(rand)));
+
+       if (unlink_directory(ctx->mail_home, UNLINK_DIRECTORY_FLAG_RMDIR, &error) < 0 &&
+           errno != ENOENT)
+               i_warning("unlink_directory(%s) failed: %s", ctx->mail_home, error);
+
+       ctx->ioloop = io_loop_create();
+
+       ctx->storage_service = mail_storage_service_init(master_service, NULL,
+               MAIL_STORAGE_SERVICE_FLAG_NO_RESTRICT_ACCESS |
+               MAIL_STORAGE_SERVICE_FLAG_NO_LOG_INIT |
+               MAIL_STORAGE_SERVICE_FLAG_NO_PLUGINS);
+}
+
+static void test_mail_deinit(struct test_mail_storage_ctx *ctx)
 {
-       static void (*const test_functions[])(void) = {
+       const char *error;
+       mail_storage_service_deinit(&ctx->storage_service);
+
+       if (chdir(ctx->mail_home) < 0)
+               i_fatal("chdir(%s) failed: %m", ctx->mail_home);
+       if (chdir("..") < 0)
+               i_fatal("chdir(..) failed: %m");
+
+       if (unlink_directory(ctx->mail_home, UNLINK_DIRECTORY_FLAG_RMDIR,
+                            &error) < 0)
+               i_error("unlink_directory(%s) failed: %s", ctx->mail_home, error);
+
+       io_loop_destroy(&ctx->ioloop);
+
+       pool_unref(&ctx->pool);
+
+       i_zero(ctx);
+}
+
+static int test_mail_init_user(const char *user, const char *driver,
+                              const char *driver_opts, const char *sep,
+                              const char *const *extra_input,
+                              struct test_mail_storage_ctx *ctx)
+{
+       const char *error, *home;
+       ARRAY_TYPE(const_string) opts;
+
+       home = t_strdup_printf("%s%s", ctx->mail_home, user);
+
+       const char *const default_input[] = {
+               t_strdup_printf("mail=%s:~/%s", driver, driver_opts),
+               "postmaster_address=postmaster@localhost",
+               "namespace=inbox",
+               "namespace/inbox/prefix=",
+               "namespace/inbox/inbox=yes",
+               t_strdup_printf("namespace/inbox/separator=%s", sep),
+               t_strdup_printf("home=%s/%s", home, user),
+       };
+
+       i_assert(mkdir_parents(home, S_IRWXU)==0 || errno == EEXIST);
+
+       t_array_init(&opts, 20);
+       array_append(&opts, default_input, N_ELEMENTS(default_input));
+       if (extra_input != NULL)
+               while(*extra_input != NULL)
+                       array_append(&opts, extra_input++, 1);
+
+       array_append_zero(&opts);
+       struct mail_storage_service_input input = {
+               .userdb_fields = array_idx(&opts, 0),
+               .username = user,
+               .no_userdb_lookup = TRUE,
+               .debug = FALSE,
+       };
+
+       if (mail_storage_service_lookup_next(ctx->storage_service, &input,
+                                            &ctx->service_user, &ctx->user,
+                                            &error) < 0) {
+                i_error("mail_storage_service_lookup_next(%s) failed: %s",
+                        user, error);
+                return -1;
+       }
+
+       return 0;
+}
+
+#define test_mail_init_maildir_user(user) test_mail_init_user(user,"maildir","",NULL)
+static void test_mail_deinit_user(struct test_mail_storage_ctx *ctx)
+{
+       mail_user_unref(&ctx->user);
+       mail_storage_service_user_unref(&ctx->service_user);
+}
+
+int main(int argc, char **argv)
+{
+       int ret;
+       void (*const tests[])(void) = {
                test_mail_storage_errors,
                test_mail_storage_last_error_push_pop,
                NULL
        };
 
-       return test_run(test_functions);
+       master_service = master_service_init("test-mail-storage",
+                                            MASTER_SERVICE_FLAG_STANDALONE |
+                                            MASTER_SERVICE_FLAG_NO_CONFIG_SETTINGS |
+                                            MASTER_SERVICE_FLAG_NO_SSL_INIT |
+                                            MASTER_SERVICE_FLAG_NO_INIT_DATASTACK_FRAME,
+                                            &argc, &argv, "");
+
+       ret = test_run(tests);
+
+       master_service_deinit(&master_service);
+
+       return ret;
 }