ccan/io: test read after hangup.

[ccan] / ccan / io / test / run-08-hangup-on-idle.c

diff --git a/ccan/io/test/run-08-hangup-on-idle.c b/ccan/io/test/run-08-hangup-on-idle.c
new file mode 100644 (file)
index 0000000..364d794
--- /dev/null
+++ b/ccan/io/test/run-08-hangup-on-idle.c
@@ -0,0 +1,45 @@
+#include <ccan/io/io.h>
+/* Include the C files directly. */
+#include <ccan/io/poll.c>
+#include <ccan/io/io.c>
+#include <ccan/tap/tap.h>
+#include <sys/wait.h>
+#include <stdio.h>
+
+static int fds2[2];
+
+static struct io_plan timeout_wakeup(struct io_conn *conn, char *buf)
+{
+       /* This kills the dummy connection. */
+       close(fds2[1]);
+       return io_read(buf, 16, io_close, NULL);
+}
+
+int main(void)
+{
+       int fds[2];
+       struct io_conn *conn;
+       char buf[16];
+
+       plan_tests(4);
+
+       ok1(pipe(fds) == 0);
+
+       /* Write then close. */
+       io_new_conn(fds[1], io_write("hello there world", 16, io_close, NULL),
+                   NULL, NULL);
+       conn = io_new_conn(fds[0], io_idle(), NULL, NULL);
+
+       /* To avoid assert(num_waiting) */
+       ok1(pipe(fds2) == 0);
+       io_new_conn(fds2[0], io_read(buf, 16, io_close, NULL), NULL, NULL);
+
+       /* After half a second, it will read. */
+       io_timeout(conn, time_from_msec(500), timeout_wakeup, buf);
+
+       ok1(io_loop() == NULL);
+       ok1(memcmp(buf, "hello there world", 16) == 0);
+
+       /* This exits depending on whether all tests passed */
+       return exit_status();
+}