failtest: handle 2-argument open()

[ccan] / ccan / failtest / test / run-open.c

diff --git a/ccan/failtest/test/run-open.c b/ccan/failtest/test/run-open.c
new file mode 100644 (file)
index 0000000..dda7ddb
--- /dev/null
+++ b/ccan/failtest/test/run-open.c
@@ -0,0 +1,67 @@
+#include <stdlib.h>
+#include <setjmp.h>
+#include <stdio.h>
+#include <stdarg.h>
+#include <assert.h>
+#include <ccan/tap/tap.h>
+/* Include the C files directly. */
+#include <ccan/failtest/failtest.c>
+
+int main(void)
+{
+       int fd, pfd[2], err;
+       char buf[] = "Hello world!";
+       struct stat st;
+
+       plan_tests(12);
+
+       pipe(pfd);
+       fd = failtest_open("run-open-scratchpad", "run-open.c", 1,
+                          O_RDWR|O_CREAT, 0600);
+       if (fd == -1) {
+               /* We are the child: write error code for parent to check. */
+               err = errno;
+               write(pfd[1], &err, sizeof(err));
+               failtest_exit(0);
+       }
+       /* Check it is read-write. */
+       ok1(write(fd, buf, strlen(buf)) == strlen(buf));
+       lseek(fd, SEEK_SET, 0);
+       ok1(read(fd, buf, strlen("Hello world!")) == strlen("Hello world!"));
+       ok1(strcmp(buf, "Hello world!") == 0);
+
+       /* Check name and perms. */
+       ok1(stat("run-open-scratchpad", &st) == 0);
+       ok1(st.st_size == strlen(buf));
+       ok1(S_ISREG(st.st_mode));
+       ok1((st.st_mode & 0777) == 0600);
+
+       /* Check child got correct errno. */
+       ok1(read(pfd[0], &err, sizeof(err)) == sizeof(err));
+       ok1(err == EACCES);
+
+       /* Clean up. */
+       close(fd);
+       close(pfd[0]);
+       close(pfd[1]);
+
+       /* Two-arg open. */
+       pipe(pfd);
+       fd = failtest_open("run-open-scratchpad", "run-open.c", 1, O_RDONLY);
+       if (fd == -1) {
+               /* We are the child: write error code for parent to check. */
+               err = errno;
+               write(pfd[1], &err, sizeof(err));
+               failtest_exit(0);
+       }
+       /* Check it is read-only. */
+       ok1(write(fd, buf, strlen(buf)) == -1);
+       ok1(read(fd, buf, strlen("Hello world!")) == strlen("Hello world!"));
+       ok1(strcmp(buf, "Hello world!") == 0);
+       /* Clean up. */
+       close(fd);
+       close(pfd[0]);
+       close(pfd[1]);
+
+       return exit_status();
+}