[Tarantool-patches] [PATCH 7/7] test/unit: add popen test
Cyrill Gorcunov
gorcunov at gmail.com
Mon Mar 2 23:12:27 MSK 2020
Basic test for popen engine
Signed-off-by: Cyrill Gorcunov <gorcunov at gmail.com>
---
test/unit/CMakeLists.txt | 3 +
test/unit/popen.c | 250 +++++++++++++++++++++++++++++++++++++++
test/unit/popen.result | 25 ++++
3 files changed, 278 insertions(+)
create mode 100644 test/unit/popen.c
create mode 100644 test/unit/popen.result
diff --git a/test/unit/CMakeLists.txt b/test/unit/CMakeLists.txt
index c037ac539..4ac08de8d 100644
--- a/test/unit/CMakeLists.txt
+++ b/test/unit/CMakeLists.txt
@@ -240,3 +240,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..7656ebe73
--- /dev/null
+++ b/test/unit/popen.c
@@ -0,0 +1,250 @@
+#include <stdio.h>
+#include <stdlib.h>
+
+#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 $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);
+ ok(rc == (int)strlen(data), "write to pipe");
+ 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 check_plan();
+}
+
+static int
+popen_read_exit(void)
+{
+ struct popen_handle *handle;
+ char *child_argv[] = {
+ "/bin/sh", "-c",
+ "echo 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);
+ ok(rc >= 9 && !strncmp(data, "1 2 3 4 5", 9), "read from pipe");
+
+out_kill:
+ rc = popen_delete(handle);
+ ok(rc == 0, "popen_delete");
+
+out:
+ footer();
+ return check_plan();
+}
+
+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 check_plan();
+}
+
+static int
+main_f(va_list ap)
+{
+ int rc = 0;
+
+ rc = popen_write_exit();
+ if (rc == 0)
+ rc = popen_read_exit();
+ if (rc == 0)
+ 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 0;
+}
diff --git a/test/unit/popen.result b/test/unit/popen.result
new file mode 100644
index 000000000..14fd3e43e
--- /dev/null
+++ b/test/unit/popen.result
@@ -0,0 +1,25 @@
+1..7
+ *** popen_write_exit ***
+ok 1 - popen_new
+ok 2 - popen_state
+ok 3 - state alive
+ok 4 - write flag check
+ok 5 - write to pipe
+ok 6 - child exited
+ok 7 - popen_delete
+ *** popen_write_exit: done ***
+1..5
+ *** popen_read_exit ***
+ok 1 - popen_new
+ok 2 - child exited
+ok 3 - read flag check
+ok 4 - read from pipe
+ok 5 - popen_delete
+ *** popen_read_exit: done ***
+1..4
+ *** popen_kill ***
+ok 1 - popen_new
+ok 2 - popen_send_signal
+ok 3 - child terminated
+ok 4 - popen_delete
+ *** popen_kill: done ***
--
2.20.1
More information about the Tarantool-patches
mailing list