Clean up temporary files in unit tests when all goes well.

1 files changed, 5 insertions, 0 deletions

diff --git a/unittests.c b/unittests.c
index 7c59e6e..8671a5b 100644
--- a/unittests.c
+++ b/unittests.c
@@ -1004,6 +1004,8 @@ static int write_test(long (*callback)(char *filename, Dynstr *dynstr))
     if (read_bytes == -1)
         abort();
     close(fd);
+    remove(tempname);
+
     dynstr2 = dynstr_new_from_memory(buf, read_bytes);
     FAIL_UNLESS_EQUAL(read_bytes, dynstr_len(dynstr));
     FAIL_UNLESS_EQUAL(dynstr_cmp(dynstr, dynstr2), 0);
@@ -1065,6 +1067,8 @@ static int read_test(Dynstr *(*callback)(char *filename, size_t size))
     close(fd);
 
     dynstr = callback(tempname, sizeof(canonical) * 2);
+    remove(tempname);
+
     FAIL_UNLESS(dynstr != NULL);
     FAIL_UNLESS_EQUAL(dynstr_len(dynstr), sizeof(canonical));
     
@@ -1137,6 +1141,7 @@ static int readline_test(Dynstr *(*callback)(char *filename))
     close(fd);
     
     result = callback(tempname);
+    remove(tempname);
 
     FAIL_UNLESS_EQUAL(dynstr_cmp(line1, result), 0);