wp

freelist/freelist_test.go

@@ -0,0 +1,44 @@
+package freelist
+
+import (
+	"fmt"
+	"os"
+	"testing"
+)
+
+func createFreeList() (_ *FreeList, err error) {
+	file, err := os.OpenFile("test.free", os.O_CREATE|os.O_RDWR, 0666)
+	if err != nil {
+		return
+	}
+	return New(Options{
+		PageSize: 20,
+		File:     file,
+	})
+}
+
+func TestAddToFreeList(t *testing.T) {
+	freeList, err := createFreeList()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	pageNumbers := []uint32{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
+
+	err = freeList.AddPageNumbers(pageNumbers)
+	if err != nil {
+		t.Fatal(err)
+	}
+}
+
+func TestGetFromFreeList(t *testing.T) {
+	freeList, err := createFreeList()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	for range 4 {
+		pageNo, err := freeList.GetPageNumber()
+		fmt.Println(pageNo, err)
+	}
+}