io: add io_flush_sync().
[ccan] / ccan / io / test / run-30-io_flush_sync.c
1 #include <ccan/io/io.h>
2 /* Include the C files directly. */
3 #include <ccan/io/poll.c>
4 #include <ccan/io/io.c>
5 #include <ccan/tap/tap.h>
6 #include <sys/wait.h>
7 #include <stdio.h>
8
9 static size_t bytes_written;
10
11 /* Should be called multiple times, since only writes 1 byte. */
12 static int do_controlled_write(int fd, struct io_plan_arg *arg)
13 {
14         ssize_t ret;
15
16         ret = write(fd, arg->u1.cp, 1);
17         if (ret < 0)
18                 return -1;
19         bytes_written += ret;
20         arg->u1.cp += ret;
21         arg->u2.s -= ret;
22         return arg->u2.s == 0;
23 }
24
25 static int do_error(int fd, struct io_plan_arg *arg)
26 {
27         errno = 1001;
28         return -1;
29 }
30
31 static struct io_plan *conn_wait(struct io_conn *conn, void *unused)
32 {
33         return io_wait(conn, conn, io_never, NULL);
34 }
35
36 static struct io_plan *init_conn_writer(struct io_conn *conn, const char *str)
37 {
38         struct io_plan_arg *arg = io_plan_arg(conn, IO_OUT);
39
40         arg->u1.const_vp = str;
41         arg->u2.s = strlen(str);
42
43         return io_set_plan(conn, IO_OUT, do_controlled_write, conn_wait, NULL);
44 }
45
46 static struct io_plan *init_conn_reader(struct io_conn *conn, void *dst)
47 {
48         /* Never actually succeeds. */
49         return io_read(conn, dst, 1000, io_never, NULL);
50 }
51
52 static struct io_plan *init_conn_error(struct io_conn *conn, void *unused)
53 {
54         io_plan_arg(conn, IO_OUT);
55         return io_set_plan(conn, IO_OUT, do_error, io_never, NULL);
56 }
57
58 int main(void)
59 {
60         int fd = open("/dev/null", O_RDWR);
61         const tal_t *ctx = tal(NULL, char);
62         struct io_conn *conn;
63
64         /* This is how many tests you plan to run */
65         plan_tests(9);
66
67         conn = io_new_conn(ctx, fd, init_conn_writer, "hello");
68         ok1(bytes_written == 0);
69
70         ok1(io_flush_sync(conn));
71         ok1(bytes_written == strlen("hello"));
72
73         /* This won't do anything */
74         ok1(io_flush_sync(conn));
75         ok1(bytes_written == strlen("hello"));
76
77         /* It's reading, this won't do anything. */
78         conn = io_new_conn(ctx, fd, init_conn_reader, ctx);
79         ok1(io_flush_sync(conn));
80         ok1(bytes_written == strlen("hello"));
81
82         /* Now test error state. */
83         conn = io_new_conn(ctx, fd, init_conn_error, ctx);
84         ok1(!io_flush_sync(conn));
85         ok1(errno == 1001);
86
87         tal_free(ctx);
88
89         /* This exits depending on whether all tests passed */
90         return exit_status();
91 }