Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

maintenance: add test to verify content of Snapshot

Browse Source 
Signed-off-by: Wei Fu <fuweid89@gmail.com>
This commit is contained in:
Wei Fu 2022-12-16 15:53:39 +08:00
parent 5879ee9bd2
commit e58c73cc18
1 changed files with 54 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
tests/integration/clientv3/maintenance_test.go
View File

@ -17,24 +17,27 @@ package clientv3test
import ( import (
"bytes" "bytes"
"context" "context"
"crypto/sha256"
"fmt" "fmt"
"io" "io"
"math" "math"
"os"
"path/filepath" "path/filepath"
"testing" "testing"
"time" "time"
"go.etcd.io/etcd/server/v3/storage/mvcc/testutil"
integration2 "go.etcd.io/etcd/tests/v3/framework/integration"
"go.uber.org/zap/zaptest"
"google.golang.org/grpc"
"go.etcd.io/etcd/api/v3/v3rpc/rpctypes" "go.etcd.io/etcd/api/v3/v3rpc/rpctypes"
"go.etcd.io/etcd/api/v3/version" "go.etcd.io/etcd/api/v3/version"
clientv3 "go.etcd.io/etcd/client/v3" clientv3 "go.etcd.io/etcd/client/v3"
"go.etcd.io/etcd/server/v3/lease" "go.etcd.io/etcd/server/v3/lease"
"go.etcd.io/etcd/server/v3/storage/backend" "go.etcd.io/etcd/server/v3/storage/backend"
"go.etcd.io/etcd/server/v3/storage/mvcc" "go.etcd.io/etcd/server/v3/storage/mvcc"
"go.etcd.io/etcd/server/v3/storage/mvcc/testutil"
integration2 "go.etcd.io/etcd/tests/v3/framework/integration"
"github.com/stretchr/testify/require"
"go.uber.org/zap/zaptest"
"google.golang.org/grpc"
) )
func TestMaintenanceHashKV(t *testing.T) { func TestMaintenanceHashKV(t *testing.T) {
@ -338,6 +341,52 @@ func TestMaintenanceSnapshotWithVersionVersion(t *testing.T) {
} }
} }
func TestMaintenanceSnapshotContentDigest(t *testing.T) {
integration2.BeforeTest(t)
clus := integration2.NewCluster(t, &integration2.ClusterConfig{Size: 1})
defer clus.Terminate(t)
populateDataIntoCluster(t, clus, 3, 1024*1024)
// reading snapshot with canceled context should error out
resp, err := clus.RandClient().SnapshotWithVersion(context.Background())
require.NoError(t, err)
defer resp.Snapshot.Close()
tmpDir := t.TempDir()
snapFile, err := os.Create(filepath.Join(tmpDir, t.Name()))
require.NoError(t, err)
defer snapFile.Close()
snapSize, err := io.Copy(snapFile, resp.Snapshot)
require.NoError(t, err)
// read the checksum
checksumSize := int64(sha256.Size)
_, err = snapFile.Seek(-checksumSize, io.SeekEnd)
require.NoError(t, err)
checksumInBytes, err := io.ReadAll(snapFile)
require.NoError(t, err)
require.Equal(t, int(checksumSize), len(checksumInBytes))
// remove the checksum part and rehash
err = snapFile.Truncate(snapSize - checksumSize)
require.NoError(t, err)
_, err = snapFile.Seek(0, io.SeekStart)
require.NoError(t, err)
hashWriter := sha256.New()
_, err = io.Copy(hashWriter, snapFile)
require.NoError(t, err)
// compare the checksum
actualChecksum := hashWriter.Sum(nil)
require.Equal(t, checksumInBytes, actualChecksum)
}
func TestMaintenanceStatus(t *testing.T) { func TestMaintenanceStatus(t *testing.T) {
integration2.BeforeTest(t) integration2.BeforeTest(t)