1 files changed, 4 insertions, 7 deletions
diff --git a/src/test/java/io/trygvis/btree/HeapFileTest.java b/src/test/java/io/trygvis/btree/HeapFileTest.java
index a9105bb..558f192 100644
--- a/src/test/java/io/trygvis/btree/HeapFileTest.java
+++ b/src/test/java/io/trygvis/btree/HeapFileTest.java
@@ -4,6 +4,7 @@ import org.junit.Test;
 
 import java.io.File;
 
+import static io.trygvis.btree.TestUtils.deleteFile;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertTrue;
 
@@ -12,16 +13,12 @@ public class HeapFileTest {
     @Test
     public void testBasic() throws Exception {
         int pageSize = 128;
-        File f = new File("target/heap");
-        if (f.exists()) {
-            assertTrue(f.delete());
-        }
-
-        HeapFile file = new HeapFile(pageSize, f);
+        HeapFile file = new HeapFile(pageSize, deleteFile(new File("target/heap")));
 
         assertEquals(0, file.pageCount());
         HeapPage page = file.blankPage();
-        file.writePage(page);
+        long pageNumber = file.writePage(page);
+        assertEquals(0, pageNumber);
         assertEquals(1, file.pageCount());
     }
 }