rbuf: new module.

[ccan] / ccan / rbuf / test / run-term-eof.c

diff --git a/ccan/rbuf/test/run-term-eof.c b/ccan/rbuf/test/run-term-eof.c
new file mode 100644 (file)
index 0000000..097dcbb
--- /dev/null
+++ b/ccan/rbuf/test/run-term-eof.c
@@ -0,0 +1,44 @@
+#include <ccan/rbuf/rbuf.h>
+/* Include the C files directly. */
+#include <ccan/rbuf/rbuf.c>
+#include <ccan/tap/tap.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <stdlib.h>
+
+int main(void)
+{
+       struct rbuf in;
+       char buf[4096], *p;
+       int fd = open("test/run-term-eof.c", O_RDONLY), len;
+
+       /* This is how many tests you plan to run */
+       plan_tests(6);
+
+       /* Grab ourselves for comparison. */
+       len = read(fd, buf, sizeof(buf));
+       buf[len] = '\0';
+       lseek(fd, SEEK_SET, 0);
+
+       /* We have exact-size buffer, which causes problems adding term. */
+       rbuf_init(&in, fd, malloc(len), len);
+       p = rbuf_read_str(&in, 64, NULL); /* At symbol does not appear. */
+       ok1(errno == ENOMEM);
+       ok1(!p);
+       /* This should succeed... */
+       p = rbuf_read_str(&in, 64, realloc);
+       ok1(p);
+       ok1(strcmp(p, buf) == 0);
+       free(in.buf);
+
+       /* Try again. */
+       lseek(fd, SEEK_SET, 0);
+       rbuf_init(&in, fd, malloc(len), len);
+       p = rbuf_read_str(&in, 64, realloc);
+       ok1(p);
+       ok1(strcmp(p, buf) == 0);
+       free(in.buf);
+
+       return exit_status();
+}