X-Git-Url: https://git.ozlabs.org/?p=ccan;a=blobdiff_plain;f=ccan%2Ffailtest%2Ftest%2Frun-write.c;h=ada185a643004ba961ce3698026a822824ad9e84;hp=d3d7c6041871c96884e28c488a0f86a1982a9f24;hb=36264d9553f924c8a58ae967f659601d1ff6793a;hpb=5adceea6af51c93d3b6fc39849ff05340cd87253

diff --git a/ccan/failtest/test/run-write.c b/ccan/failtest/test/run-write.c
index d3d7c604..ada185a6 100644
--- a/ccan/failtest/test/run-write.c
+++ b/ccan/failtest/test/run-write.c
@@ -7,30 +7,34 @@
 /* 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 = failtest_open("run-write-scratchpad", "run-write.c", 1,
+	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));
+	if (write(fd, buf, strlen(buf)) != strlen(buf))
+		abort();
 	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);
 	}
 
@@ -41,6 +45,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();
 }