e2e test: add schema version verification in mix_version_test.

Signed-off-by: Siyuan Zhang <[email protected]>

Author: siyuanfoundation

tests/e2e/cluster_downgrade_test.go

@@ -202,6 +202,9 @@ func newCluster(t *testing.T, clusterSize int, snapshotCount uint64) *e2e.EtcdPr
 		if errC := epc.Close(); errC != nil {
 			t.Fatalf("error closing etcd processes (%v)", errC)
 		}
+		for _, proc := range epc.Procs {
+			require.NoError(t, proc.VerifySchemaVersion(epc.Cfg.Logger))
+		}
 	})
 	return epc
 }

tests/e2e/etcd_mix_versions_test.go

@@ -92,11 +92,15 @@ func mixVersionsSnapshotTestByAddingMember(t *testing.T, cfg *e2e.EtcdProcessClu
 	epc, err := e2e.NewEtcdProcessCluster(context.TODO(), t,
 		e2e.WithConfig(cfg),
 		e2e.WithSnapshotCount(10),
+		e2e.WithKeepDataDir(true),
 	)
 	require.NoError(t, err, "failed to start etcd cluster: %v", err)
 	defer func() {
 		derr := epc.Close()
 		require.NoError(t, derr, "failed to close etcd cluster: %v", derr)
+		for _, proc := range epc.Procs {
+			require.NoError(t, proc.VerifySchemaVersion(epc.Cfg.Logger))
+		}
 	}()
 
 	t.Log("Writing 20 keys to the cluster (more than SnapshotCount entries to trigger at least a snapshot)")
@@ -135,11 +139,15 @@ func mixVersionsSnapshotTestByMockPartition(t *testing.T, cfg *e2e.EtcdProcessCl
 		e2e.WithConfig(cfg),
 		e2e.WithSnapshotCount(10),
 		e2e.WithSnapshotCatchUpEntries(10),
+		e2e.WithKeepDataDir(true),
 	)
 	require.NoError(t, err, "failed to start etcd cluster: %v", err)
 	defer func() {
 		derr := epc.Close()
 		require.NoError(t, derr, "failed to close etcd cluster: %v", derr)
+		for _, proc := range epc.Procs {
+			require.NoError(t, proc.VerifySchemaVersion(epc.Cfg.Logger))
+		}
 	}()
 	toPartitionedMember := epc.Procs[mockPartitionNodeIndex]
 

tests/framework/e2e/etcd_process.go

@@ -31,9 +31,13 @@ import (
 	"github.com/coreos/go-semver/semver"
 	"go.uber.org/zap"
 
+	"go.etcd.io/etcd/api/v3/version"
 	"go.etcd.io/etcd/client/pkg/v3/fileutil"
 	"go.etcd.io/etcd/pkg/v3/expect"
 	"go.etcd.io/etcd/pkg/v3/proxy"
+	"go.etcd.io/etcd/server/v3/storage/backend"
+	"go.etcd.io/etcd/server/v3/storage/datadir"
+	"go.etcd.io/etcd/server/v3/storage/schema"
 	"go.etcd.io/etcd/tests/v3/framework/config"
 )
 
@@ -60,6 +64,8 @@ type EtcdProcess interface {
 	LazyFS() *LazyFS
 	Logs() LogsExpect
 	Kill() error
+	// VerifySchemaVersion verifies the db file schema version is compatible with the binary after the process is closed.
+	VerifySchemaVersion(lg *zap.Logger) error
 }
 
 type LogsExpect interface {
@@ -253,6 +259,34 @@ func (ep *EtcdServerProcess) Close() error {
 	return nil
 }
 
+func (ep *EtcdServerProcess) VerifySchemaVersion(lg *zap.Logger) error {
+	dbPath := datadir.ToBackendFileName(ep.cfg.DataDirPath)
+	be := backend.NewDefaultBackend(lg, dbPath)
+	defer be.Close()
+	ver, err := schema.UnsafeDetectSchemaVersion(lg, be.BatchTx())
+	if err != nil {
+		return err
+	}
+	maxStorageVersion := version.V3_6
+	if ep.cfg.ExecPath == BinPath.EtcdLastRelease {
+		maxStorageVersion = version.V3_5
+	}
+	// in a mix version cluster, the storage version would be set to the cluster version,
+	// which could be lower than the server version by 1 minor version.
+	if !ver.LessThan(maxStorageVersion) && !ver.Equal(maxStorageVersion) {
+		return fmt.Errorf("expect backend schema version to be <= %s, but got %s", maxStorageVersion.String(), ver.String())
+	}
+	// check new fields introduced in V3_6 do not exist in V3_5 data file.
+	// V3_6 contains all the fields in V3_5, so no need to check for V3_6 servers.
+	if maxStorageVersion == version.V3_5 {
+		_, vs := be.BatchTx().UnsafeRange(schema.Meta, schema.MetaStorageVersionName, nil, 1)
+		if len(vs) != 0 {
+			return fmt.Errorf("expect storageVersion not exist in the meta bucket, but got %s", string(vs[0]))
+		}
+	}
+	return nil
+}
+
 func (ep *EtcdServerProcess) waitReady(ctx context.Context) error {
 	defer close(ep.donec)
 	return WaitReadyExpectProc(ctx, ep.proc, EtcdServerReadyLines)