X-Git-Url: http://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=ccan%2Ffailtest%2Ftest%2Frun-write.c;h=9f1f1d7fc8005fdda59cbe46684157329f2469be;hp=5425517030a5d77f1db0b22926c30956d54bf183;hb=b7dedf4344c7e573c048a36e7b456f2a4bdbebb7;hpb=f0002cb9e4f6f403a25ad50252c06694439900f0

diff --git a/ccan/failtest/test/run-write.c b/ccan/failtest/test/run-write.c
index 54255170..9f1f1d7f 100644
--- a/ccan/failtest/test/run-write.c
+++ b/ccan/failtest/test/run-write.c
@@ -7,26 +7,33 @@
 /* Include the C files directly. */
 #include <ccan/failtest/failtest.c>
 
-int main(void)
+int main(int argc, char *argv[])
 {
 	int fd;
 	char *p;
 	char buf[] = "Hello world!";
 
 	plan_tests(5);
+	failtest_init(argc, argv);
 
-	fd = open("run-write-scratchpad", O_RDWR|O_CREAT, 0600);
+	fd = failtest_open("run-write-scratchpad", __FILE__, __LINE__,
+			   O_RDWR|O_CREAT, 0600);
+	/* Child will fail, ignore. */
+	if (fd < 0)
+		failtest_exit(0);
 	write(fd, buf, strlen(buf));
 	ok1(lseek(fd, 0, SEEK_CUR) == strlen(buf));
 
-	p = failtest_malloc(100, "run-write.c", 1);
+	p = failtest_malloc(100, __FILE__, __LINE__);
 	if (!p) {
 		/* We are the child.  Do a heap of writes. */
 		unsigned int i;
 
 		for (i = 0; i < strlen(buf)+1; i++)
-			if (failtest_write(fd, "x", 1, "run-write.c", 1) == 1)
+			if (failtest_write(fd, "x", 1, __FILE__, __LINE__)
+			    == 1)
 				break;
+		failtest_close(fd, __FILE__, __LINE__);
 		failtest_exit(0);
 	}
 
@@ -37,6 +44,7 @@ int main(void)
 	lseek(fd, 0, SEEK_SET);
 	ok1(read(fd, buf, strlen(buf)) == strlen("Hello world!"));
 	ok1(strcmp(buf, "Hello world!") == 0);
+	failtest_close(fd, __FILE__, __LINE__);
 
 	return exit_status();
 }