auth: Add auth-client-connection test

diff --git a/src/auth/Makefile.am b/src/auth/Makefile.am
index e6e4cd7ed81a96c7ed8938dabea0bba4f8d87a6d..ca558d8ff67c2ada9ac799a581aef41ca62f44bf 100644 (file)
--- a/src/auth/Makefile.am
+++ b/src/auth/Makefile.am
@@ -208,6 +208,8 @@ pkginc_lib_HEADERS = $(headers)
 
 test_programs = \
        test-auth-cache \
+       test-auth-client \
+       test-auth-master \
        test-auth \
        test-mech
 
@@ -245,6 +247,14 @@ test_mech_SOURCES = \
 test_mech_LDADD = $(test_libs) $(auth_libs) $(AUTH_LIBS) $(LUA_LIBS)
 test_mech_DEPENDENCIES = $(pkglibexec_PROGRAMS) $(test_libs)
 
+test_auth_client_SOURCES = \
+       test-auth.c \
+       test-mock.c \
+       test-auth-client.c
+
+test_auth_client_LDADD = $(test_libs) $(auth_libs) $(AUTH_LIBS) $(LUA_LIBS)
+test_auth_client_DEPENDENCIES = $(pkglibexec_PROGRAMS) $(test_libs)
+
 check-local:
        for bin in $(test_programs); do \
          if ! $(RUN_TEST) ./$$bin; then exit 1; fi; \

diff --git a/src/auth/test-auth-client.c b/src/auth/test-auth-client.c
new file mode 100644 (file)
index 0000000..2254d20
--- /dev/null
+++ b/src/auth/test-auth-client.c
@@ -0,0 +1,99 @@
+/* Copyright (c) 2005-2018 Dovecot authors, see the included COPYING file */
+
+#include "test-auth.h"
+#include "ioloop.h"
+#include "net.h"
+#include "master-service.h"
+#include "auth-common.h"
+#include "auth-client-connection.h"
+#include "auth-client.h"
+
+#define TEST_AUTH_CLIENT_SOCKET "test-auth-client-socket"
+
+static void auth_client_connected(int *server_fd)
+{
+       struct auth *auth = auth_default_service();
+       int fd = net_accept(*server_fd, NULL, NULL);
+       auth_client_connection_create(auth, fd, TEST_AUTH_CLIENT_SOCKET, FALSE, FALSE);
+}
+
+static void
+test_callback(struct auth_client_request *req, enum auth_request_status status,
+             const char *b64, const char *const *args, void *context)
+{
+       struct auth_client *client ATTR_UNUSED = context;
+       if (status == AUTH_REQUEST_STATUS_CONTINUE) {
+               test_assert_strcmp(b64, "");
+               /* continue with \0testuser\0testpass */
+               auth_client_request_continue(req, "AHRlc3R1c2VyAHRlc3RwYXNz");
+               return;
+       }
+       test_assert_strcmp(args[0], "user=testuser");
+       test_assert_cmp(status, ==, AUTH_REQUEST_STATUS_OK);
+       io_loop_stop(current_ioloop);
+}
+
+static void test_auth_client(void)
+{
+       test_begin("auth client");
+       struct ioloop *loop = io_loop_create();
+       test_auth_init();
+
+       i_unlink_if_exists(TEST_AUTH_CLIENT_SOCKET);
+       int fd = net_listen_unix(TEST_AUTH_CLIENT_SOCKET, 10);
+       struct io *io = io_add(fd, IO_READ, auth_client_connected, &fd);
+
+       struct auth_client *client =
+               auth_client_init(TEST_AUTH_CLIENT_SOCKET, getpid(), FALSE);
+       auth_client_connect(client);
+
+       while (!auth_client_is_connected(client)) {
+               io_loop_set_running(current_ioloop);
+               io_loop_handler_run(current_ioloop);
+       }
+
+       struct auth_request_info reqinfo = {
+               .mech = "plain",
+               .client_id = "1",
+               .service = "default",
+               .session_id = "1",
+       };
+       (void)auth_client_request_new(client, &reqinfo, test_callback, client);
+
+       io_loop_run(current_ioloop);
+
+       auth_client_deinit(&client);
+       io_remove(&io);
+       auth_client_connections_destroy_all();
+       test_auth_deinit();
+       io_loop_destroy(&loop);
+       i_unlink_if_exists(TEST_AUTH_CLIENT_SOCKET);
+       test_end();
+}
+
+int main(int argc, char *argv[])
+{
+       static void (*const test_functions[])(void) = {
+               test_auth_client,
+               NULL
+       };
+       const enum master_service_flags service_flags =
+               MASTER_SERVICE_FLAG_NO_CONFIG_SETTINGS |
+               MASTER_SERVICE_FLAG_STANDALONE |
+               MASTER_SERVICE_FLAG_STD_CLIENT |
+               MASTER_SERVICE_FLAG_DONT_SEND_STATS;
+       int ret;
+
+       master_service = master_service_init("test-auth-client",
+                                            service_flags, &argc, &argv, "");
+
+       master_service_init_finish(master_service);
+
+       struct ioloop *ioloop = io_loop_create();
+       io_loop_set_current(ioloop);
+       ret = test_run(test_functions);
+       io_loop_destroy(&ioloop);
+
+       master_service_deinit(&master_service);
+       return ret;
+}

diff --git a/src/auth/test-auth.c b/src/auth/test-auth.c
index 0ba5f8f49d16b98f993d01d284adb905668cfcfa..5fb4eee7a07f356d8e15f25c55aa3f24f9cdb58c 100644 (file)
--- a/src/auth/test-auth.c
+++ b/src/auth/test-auth.c
@@ -5,6 +5,7 @@
 #include "settings-parser.h"
 #include "auth-settings.h"
 #include "auth-token.h"
+#include "auth-penalty.h"
 #include "mech.h"
 #include "otp.h"
 #include "mech-otp-common.h"
@@ -60,10 +61,13 @@ void test_auth_init(void)
        auths_preinit(&test_auth_set, mech_reg, protocols);
        auths_init();
        auth_token_init();
+
+       auth_penalty = auth_penalty_init("missing");
 }
 
 void test_auth_deinit(void)
 {
+       auth_penalty_deinit(&auth_penalty);
        mech_otp_deinit();
        auths_deinit();
        auth_token_deinit();