tests & example

diskv_wrapper.go

@@ -0,0 +1,16 @@
+package diskv_wrapper
+
+import (
+	"github.com/peterbourgon/diskv/v3"
+)
+
+func Init(basePath string) *diskv.Diskv {
+	disk := diskv.New(diskv.Options{
+		BasePath:          basePath,
+		AdvancedTransform: AdvanceTransform,
+		InverseTransform:  InverseTransform,
+		CacheSizeMax:      1024 * 1024,
+	})
+
+	return disk
+}

example/main.go

@@ -11,10 +11,11 @@ import (
 
 func main() {
 	var version, key, value string
+	var basePath = "example_data"
 
 	fmt.Println("::Using Diskv")
 
-	currentDisk := diskv_utils.Init()
+	currentDisk := diskv_utils.Init(basePath)
 
 	fmt.Println("::::Existing")
 	existingVersionUse := diskv_utils.VersionUse(currentDisk)

tests/disk_test/version

@@ -0,0 +1 @@
+v1

tests/diskv_wrapper_test.go

@@ -0,0 +1,24 @@
+package diskv_wrapper
+
+import (
+	"testing"
+
+	"github.com/mkhuda/diskv_wrapper"
+	"github.com/peterbourgon/diskv/v3"
+)
+
+var (
+	disk *diskv.Diskv = diskv_wrapper.Init("disk_test")
+)
+
+func write_disk_version() {
+	diskv_wrapper.WriteVersion(disk, "v1")
+}
+
+func TestVersionUse(t *testing.T) {
+	write_disk_version()
+	existingVersionUse := diskv_wrapper.VersionUse(disk)
+	if len(existingVersionUse) == 0 {
+		t.Errorf("Expected has version to use")
+	}
+}

writer.go

@@ -4,17 +4,6 @@ import (
 	"github.com/peterbourgon/diskv/v3"
 )
 
-func Init() *diskv.Diskv {
-	disk := diskv.New(diskv.Options{
-		BasePath:          "localdisk",
-		AdvancedTransform: AdvanceTransform,
-		InverseTransform:  InverseTransform,
-		CacheSizeMax:      1024 * 1024,
-	})
-
-	return disk
-}
-
 func WriteVersion(d *diskv.Diskv, version string) {
 	d.WriteString("version", version)
 }