stateloader: echo test for WriteInitialRangeState

Epic: none
Release note: none

Author: pav-kv

pkg/kv/kvserver/stateloader/BUILD.bazel

@@ -30,15 +30,19 @@ go_test(
         "initial_test.go",
         "stateloader_test.go",
     ],
+    data = glob(["testdata/**"]),
     embed = [":stateloader"],
     deps = [
         "//pkg/kv/kvpb",
         "//pkg/kv/kvserver/kvserverpb",
+        "//pkg/kv/kvserver/print",
         "//pkg/raft/raftpb",
         "//pkg/roachpb",
         "//pkg/storage",
         "//pkg/testutils",
+        "//pkg/testutils/echotest",
         "//pkg/util/leaktest",
+        "//pkg/util/log",
         "//pkg/util/stop",
         "@com_github_stretchr_testify//require",
     ],

pkg/kv/kvserver/stateloader/initial_test.go

@@ -7,19 +7,55 @@ package stateloader
 
 import (
 	"context"
+	"path/filepath"
 	"reflect"
+	"strings"
 	"testing"
 
 	"github.com/cockroachdb/cockroach/pkg/kv/kvpb"
 	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/kvserverpb"
+	"github.com/cockroachdb/cockroach/pkg/kv/kvserver/print"
 	"github.com/cockroachdb/cockroach/pkg/raft/raftpb"
 	"github.com/cockroachdb/cockroach/pkg/roachpb"
 	"github.com/cockroachdb/cockroach/pkg/storage"
 	"github.com/cockroachdb/cockroach/pkg/testutils"
+	"github.com/cockroachdb/cockroach/pkg/testutils/echotest"
 	"github.com/cockroachdb/cockroach/pkg/util/leaktest"
+	"github.com/cockroachdb/cockroach/pkg/util/log"
 	"github.com/cockroachdb/cockroach/pkg/util/stop"
+	"github.com/stretchr/testify/require"
 )
 
+// TestWriteInitialRangeState captures the typical initial range state written
+// to storage at cluster bootstrap.
+func TestWriteInitialRangeState(t *testing.T) {
+	defer leaktest.AfterTest(t)()
+	defer log.Scope(t).Close(t)
+
+	eng := storage.NewDefaultInMemForTesting()
+	defer eng.Close()
+	b := eng.NewBatch() // TODO(pav-kv): make it write-only batch
+	defer b.Close()
+
+	require.NoError(t, WriteInitialRangeState(context.Background(), b,
+		roachpb.RangeDescriptor{
+			RangeID:       5,
+			StartKey:      roachpb.RKey("a"),
+			EndKey:        roachpb.RKey("z"),
+			NextReplicaID: 4,
+		},
+		roachpb.ReplicaID(3),
+		// Use arbitrary version instead of things like clusterversion.Latest, so
+		// that the test doesn't sporadically fail when version bumps occur.
+		roachpb.Version{Major: 10, Minor: 2, Patch: 17},
+	))
+
+	str, err := print.DecodeWriteBatch(b.Repr())
+	require.NoError(t, err)
+	str = strings.ReplaceAll(str, "\n\n", "\n")
+	echotest.Require(t, str, filepath.Join("testdata", t.Name()+".txt"))
+}
+
 func TestSynthesizeHardState(t *testing.T) {
 	defer leaktest.AfterTest(t)()
 	stopper := stop.NewStopper()

pkg/kv/kvserver/stateloader/testdata/TestWriteInitialRangeState.txt

@@ -0,0 +1,10 @@
+echo
+----
+Put: 0,0 /Local/RangeID/5/u/RaftTruncatedState (0x01698d757266747400): index:10 term:5 
+Put: 0,0 /Local/RangeID/5/r/RangeLease (0x01698d72726c6c2d00): <empty>
+Put: 0,0 /Local/RangeID/5/r/RangeGCThreshold (0x01698d726c67632d00): 0,0
+Put: 0,0 /Local/RangeID/5/r/RangeGCHint (0x01698d727267636800): meta={id=00000000 key=/Min iso=Serializable pri=0.00000000 epo=0 ts=0,0 min=0,0 seq=0} lock=false stat=PENDING rts=0,0 gul=0,0
+Put: 0,0 /Local/RangeID/5/r/RangeVersion (0x01698d727276657200): 10.2
+Put: 0,0 /Local/RangeID/5/r/RangeAppliedState (0x01698d727261736b00): raft_applied_index:10 lease_applied_index:10 range_stats:<sys_bytes:142 sys_count:4 > raft_closed_timestamp:<> raft_applied_index_term:5 
+Put: 0,0 /Local/RangeID/5/u/RaftHardState (0x01698d757266746800): term:5 vote:0 commit:10 lead:0 lead_epoch:0 
+Put: 0,0 /Local/RangeID/5/u"rftr" (0x01698d757266747200): txn={<nil>} ts=0,0 del=false klen=0 vlen=0 rawlen=7 mergeTs=<nil> txnDidNotUpdateMeta=false