From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from mail-lf1-f66.google.com (mail-lf1-f66.google.com [209.85.167.66]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by dev.tarantool.org (Postfix) with ESMTPS id 6DAAA46970E for ; Fri, 31 Jan 2020 22:25:44 +0300 (MSK) Received: by mail-lf1-f66.google.com with SMTP id 203so5612055lfa.12 for ; Fri, 31 Jan 2020 11:25:44 -0800 (PST) From: Cyrill Gorcunov Date: Fri, 31 Jan 2020 22:25:04 +0300 Message-Id: <20200131192504.12142-4-gorcunov@gmail.com> In-Reply-To: <20200131192504.12142-1-gorcunov@gmail.com> References: <20200131192504.12142-1-gorcunov@gmail.com> MIME-Version: 1.0 Content-Transfer-Encoding: 8bit Subject: [Tarantool-patches] [PATCH v10 3/3] test: unit/popen List-Id: Tarantool development patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , To: tml Basic tests for popen engine Signed-off-by: Cyrill Gorcunov --- test/unit/CMakeLists.txt | 3 + test/unit/popen.c | 253 +++++++++++++++++++++++++++++++++++++++ 2 files changed, 256 insertions(+) create mode 100644 test/unit/popen.c diff --git a/test/unit/CMakeLists.txt b/test/unit/CMakeLists.txt index 4a57597e9..14cbb1c9d 100644 --- a/test/unit/CMakeLists.txt +++ b/test/unit/CMakeLists.txt @@ -236,3 +236,6 @@ target_link_libraries(swim_errinj.test unit swim) add_executable(merger.test merger.test.c) target_link_libraries(merger.test unit core box) + +add_executable(popen.test popen.c) +target_link_libraries(popen.test misc unit core) diff --git a/test/unit/popen.c b/test/unit/popen.c new file mode 100644 index 000000000..d529a7322 --- /dev/null +++ b/test/unit/popen.c @@ -0,0 +1,253 @@ +#include +#include + +#include "trivia/util.h" +#include "unit.h" + +#include "coio.h" +#include "coio_task.h" +#include "memory.h" +#include "fiber.h" +#include "popen.h" +#include "say.h" + +#define TEST_POPEN_COMMON_FLAGS \ + (POPEN_FLAG_SETSID | \ + POPEN_FLAG_SHELL | \ + POPEN_FLAG_RESTORE_SIGNALS) + +static int +wait_exit(struct popen_handle *handle, int *state, int *exit_code) +{ + for (;;) { + if (popen_state(handle, state, exit_code)) + return -1; + if (*state == POPEN_STATE_EXITED || + *state == POPEN_STATE_SIGNALED) + break; + fiber_sleep(0.1); + } + return 0; +} + +static int +popen_write_exit(void) +{ + struct popen_handle *handle; + char *child_argv[] = { + "/bin/sh", "-c", + "prompt=''; read -n 5 prompt; echo -n $prompt", + NULL, + }; + + const char data[] = "12345"; + int state, exit_code; + + struct popen_opts opts = { + .argv = child_argv, + .nr_argv = lengthof(child_argv), + .env = NULL, + .flags = + POPEN_FLAG_FD_STDIN | + POPEN_FLAG_FD_STDOUT | + POPEN_FLAG_FD_STDERR | + TEST_POPEN_COMMON_FLAGS, + }; + int rc; + + plan(7); + header(); + + handle = popen_new(&opts); + ok(handle != NULL, "popen_new"); + if (handle == NULL) + goto out; + + rc = popen_state(handle, &state, &exit_code); + ok(rc == 0, "popen_state"); + + ok(state == POPEN_STATE_ALIVE, "state %s", + popen_state_str(state)); + + rc = popen_write_timeout(handle, (void *)data, + (int)strlen(data), + POPEN_FLAG_FD_STDOUT, 180); + ok(rc == -1, "write flag check"); + + rc = popen_write_timeout(handle, (void *)data, + (int)strlen(data), + POPEN_FLAG_FD_STDIN, 180); + diag("write %d bytes '%s'", (int)strlen(data), data); + ok(rc == (int)strlen(data), "write %s (%d bytes)", + data, (int)strlen(data)); + if (rc != (int)strlen(data)) + goto out_kill; + + rc = wait_exit(handle, &state, &exit_code); + if (rc) { + ok(false, "child wait"); + goto out_kill; + } + + ok(state == POPEN_STATE_EXITED, "child exited"); + +out_kill: + rc = popen_delete(handle); + ok(rc == 0, "popen_delete"); + +out: + footer(); + return rc; +} + +static int +popen_read_exit(void) +{ + struct popen_handle *handle; + char *child_argv[] = { + "/bin/sh", "-c", + "echo -n 1 2 3 4 5", + NULL, + }; + + int state, exit_code; + char data[32] = { }; + + struct popen_opts opts = { + .argv = child_argv, + .nr_argv = lengthof(child_argv), + .env = NULL, + .flags = + POPEN_FLAG_FD_STDIN | + POPEN_FLAG_FD_STDOUT | + POPEN_FLAG_FD_STDERR | + TEST_POPEN_COMMON_FLAGS, + }; + int rc; + + plan(5); + header(); + + handle = popen_new(&opts); + ok(handle != NULL, "popen_new"); + if (handle == NULL) + goto out; + + rc = wait_exit(handle, &state, &exit_code); + if (rc) { + ok(false, "child wait"); + goto out_kill; + } + ok(state == POPEN_STATE_EXITED, "child exited"); + + rc = popen_read_timeout(handle, data, sizeof(data), + POPEN_FLAG_FD_STDIN, 180); + ok(rc == -1, "read flag check"); + + rc = popen_read_timeout(handle, data, sizeof(data), + POPEN_FLAG_FD_STDOUT, 180); + diag("read %d bytes '%s'\n", rc, data); + ok(rc == 9 && !strcmp(data, "1 2 3 4 5"), + "read %s (%d bytes)", data, rc); + +out_kill: + rc = popen_delete(handle); + ok(rc == 0, "popen_delete"); + +out: + footer(); + return rc; +} + +static int +popen_kill(void) +{ + struct popen_handle *handle; + char *child_argv[] = { + "/bin/sh", "-c", + "while [ 1 ]; do sleep 10; done", + NULL, + }; + + int state, exit_code; + + struct popen_opts opts = { + .argv = child_argv, + .nr_argv = lengthof(child_argv), + .env = NULL, + .flags = + POPEN_FLAG_FD_STDIN | + POPEN_FLAG_FD_STDOUT | + POPEN_FLAG_FD_STDERR | + TEST_POPEN_COMMON_FLAGS, + }; + int rc; + + plan(4); + header(); + + handle = popen_new(&opts); + ok(handle != NULL, "popen_new"); + if (handle == NULL) + goto out; + + rc = popen_send_signal(handle, SIGTERM); + ok(rc == 0, "popen_send_signal"); + if (rc != 0) + goto out_kill; + + rc = wait_exit(handle, &state, &exit_code); + if (rc) { + ok(false, "child wait"); + goto out_kill; + } + ok(state == POPEN_STATE_SIGNALED, "child terminated"); + +out_kill: + rc = popen_delete(handle); + ok(rc == 0, "popen_delete"); + +out: + footer(); + return rc; +} + +static int +main_f(va_list ap) +{ + int rc = 0; + + rc |= popen_write_exit(); + rc |= popen_read_exit(); + rc |= popen_kill(); + + ev_break(loop(), EVBREAK_ALL); + return 0; +} + + +int +main(int argc, char *argv[]) +{ + //say_logger_init(NULL, S_DEBUG, 0, "plain", 0); + memory_init(); + + fiber_init(fiber_c_invoke); + popen_init(); + coio_init(); + coio_enable(); + + if (!loop()) + panic("%s", "can't init event loop"); + + struct fiber *test = fiber_new("coio_stat", main_f); + fiber_wakeup(test); + + ev_now_update(loop()); + ev_run(loop(), 0); + popen_free(); + fiber_free(); + memory_free(); + + return check_plan(); +} -- 2.20.1