From 36f14deae7f1ccf297e7297c68f8ede4e025d1c9 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Sun, 15 Sep 2013 01:03:55 +0200
Subject: o Adding a more comprehensive test of the heap file. Still needs a
 lot more work.

---
 src/test/java/io/trygvis/btree/HeapFileTest.java | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

(limited to 'src/test/java/io/trygvis/btree/HeapFileTest.java')

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());
     }
 }
-- 
cgit v1.2.3