-
Notifications
You must be signed in to change notification settings - Fork 376
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Basic tests for popen engine Signed-off-by: Cyrill Gorcunov <gorcunov@gmail.com>
- Loading branch information
Showing
3 changed files
with
284 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,253 @@ | ||
#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 -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 | check_plan(); | ||
} | ||
|
||
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 | 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 rc | check_plan(); | ||
} | ||
|
||
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(); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,28 @@ | ||
# write 5 bytes '12345' | ||
# read 9 bytes '1 2 3 4 5' | ||
|
||
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 12345 (5 bytes) | ||
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 1 2 3 4 5 (9 bytes) | ||
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 *** |