[Testing] Load latest version of IAVL tree right after flushed

store/iavlstore_test.go

@@ -2,7 +2,9 @@ package store
 
 import (
 	"bytes"
+	"fmt"
 	"os"
+	"strconv"
 	"testing"
 
 	"github.com/stretchr/testify/require"
@@ -144,6 +146,43 @@ func testNormal(t *testing.T) {
 	diskDb.Close()
 }
 
+func TestLoadIAVLVersion(t *testing.T) {
+	flushInterval = int64(5)
+	memDB := db.NewMemDB()
+	store, err := NewIAVLStore(memDB, 0, 0, flushInterval)
+	require.NoError(t, err)
+	var lastVersion int64
+	var s string
+	ch := make(chan int64)
+	go func() {
+		for i := int64(1); i <= flushInterval*10000; i++ {
+			s = strconv.FormatInt(i, 10)
+			store.Set([]byte("key"+s), []byte("value"+s))
+
+			// Broken Unpredictably
+			_, lastVersion, err = store.SaveVersion(nil)
+
+			// Works fine
+			// _, lastVersion, err = store.tree.SaveVersion()
+			ch <- lastVersion
+			require.NoError(t, err)
+			fmt.Printf("%d\n", lastVersion)
+			require.Equal(t, i, lastVersion)
+		}
+	}()
+
+	for {
+		select {
+		case lv := <-ch:
+			fmt.Println("lv ", lv)
+			it, err := store.tree.GetImmutable(lv)
+			require.NoError(t, err)
+			require.NotNil(t, it)
+		}
+	}
+
+}
+
 func testFlush(t *testing.T) {
 	diskDb := getDiskDb(t, "testFlush")
 	store, err := NewIAVLStore(diskDb, 0, 0, flushInterval)