1.4.5.2

EdgeNode/internal/utils/kvstore/db_test.go

@@ -0,0 +1,55 @@
+// Copyright 2024 GoEdge CDN goedge.cdn@gmail.com. All rights reserved. Official site: https://goedge.cn .
+
+package kvstore_test
+
+import (
+	"github.com/TeaOSLab/EdgeNode/internal/utils/kvstore"
+	"github.com/TeaOSLab/EdgeNode/internal/utils/testutils"
+	"github.com/cockroachdb/pebble"
+	"testing"
+)
+
+func TestNewDB(t *testing.T) {
+	store, err := kvstore.OpenStore("test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		_ = store.Close()
+	}()
+
+	_, err = store.NewDB("TEST_DB")
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	testingStore = store
+	testInspectDB(t)
+}
+
+func testInspectDB(t *testing.T) {
+	if testingStore == nil {
+		return
+	}
+	it, err := testingStore.RawDB().NewIter(&pebble.IterOptions{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer func() {
+		_ = it.Close()
+	}()
+
+	var isSingleTesting = testutils.IsSingleTesting()
+
+	for it.First(); it.Valid(); it.Next() {
+		valueBytes, valueErr := it.ValueAndErr()
+		if valueErr != nil {
+			t.Fatal(valueErr)
+		}
+		t.Log(string(it.Key()), "=>", string(valueBytes))
+
+		if !isSingleTesting {
+			break
+		}
+	}
+}