test(server/v2): Add system-test for store's command

tests/systemtests/cli.go

@@ -169,6 +169,16 @@ func (c CLIWrapper) Run(args ...string) string {
 	return rsp
 }
 
+// RunCommandWithArgs use for run cli command, not tx
+func (c CLIWrapper) RunCommandWithArgs(args ...string) string {
+	c.t.Helper()
+	execOutput, ok := c.run(args)
+	if !ok {
+		return execOutput
+	}
+	return execOutput
+}
+
 // AwaitTxCommitted wait for tx committed on chain
 func (c CLIWrapper) AwaitTxCommitted(submitResp string, timeout ...time.Duration) (string, bool) {
 	c.t.Helper()

tests/systemtests/snapshots_test.go

@@ -0,0 +1,121 @@
+//go:build system_test
+
+package systemtests
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func getBinaryNameAndPrefix(isSnapshot bool) (string, string) {
+	if sut.execBinary == filepath.Join(WorkDir, "binaries", "simd") {
+		if isSnapshot {
+			return "simd", "snapshots"
+		}
+		return "simd", ""
+	}
+	return "simdv2", "store"
+}
+
+func TestSnapshots(t *testing.T) {
+	sut.ResetChain(t)
+	cli := NewCLIWrapper(t, sut, verbose)
+	// add genesis account with some tokens
+	account1Addr := cli.AddKey("account1")
+	sut.ModifyGenesisCLI(t,
+		[]string{"genesis", "add-genesis-account", account1Addr, "10000000stake"},
+	)
+
+	sut.StartChain(t)
+
+	binary, snapshotPrefix := getBinaryNameAndPrefix(true)
+	nodeDir := fmt.Sprintf("./testnet/node0/%s", binary)
+
+	// Wait for chain produce some blocks
+	time.Sleep(time.Second * 10)
+	// Stop 1 node
+	err := sut.StopSingleNode()
+	require.NoError(t, err)
+	time.Sleep(time.Second)
+
+	// export snapshot at height 5
+	res := cli.RunCommandWithArgs(snapshotPrefix, "export", "--height=5", fmt.Sprintf("--home=%s", nodeDir))
+	require.Contains(t, res, "Snapshot created at height 5")
+	require.DirExists(t, fmt.Sprintf("%s/data/snapshots/5/3", nodeDir))
+
+	// Check snapshots list
+	res = cli.RunCommandWithArgs(snapshotPrefix, "list", fmt.Sprintf("--home=%s", nodeDir))
+	require.Contains(t, res, "height: 5")
+
+	// Dump snapshot
+	res = cli.RunCommandWithArgs(snapshotPrefix, "dump", "5", "3", fmt.Sprintf("--home=%s", nodeDir), fmt.Sprintf("--output=%s/5-3.tar.gz", nodeDir))
+	// Check if output file exist
+	require.FileExists(t, fmt.Sprintf("%s/5-3.tar.gz", nodeDir))
+
+	// Delete snapshots
+	res = cli.RunCommandWithArgs(snapshotPrefix, "delete", "5", "3", fmt.Sprintf("--home=%s", nodeDir))
+	require.NoDirExists(t, fmt.Sprintf("%s/data/snapshots/5/3", nodeDir))
+
+	// Load snapshot from file
+	res = cli.RunCommandWithArgs(snapshotPrefix, "load", fmt.Sprintf("%s/5-3.tar.gz", nodeDir), fmt.Sprintf("--home=%s", nodeDir))
+	require.DirExists(t, fmt.Sprintf("%s/data/snapshots/5/3", nodeDir))
+
+	// Restore from snapshots
+
+	// Remove database
+	err = os.RemoveAll(fmt.Sprintf("%s/data/application.db", nodeDir))
+	require.NoError(t, err)
+
+	// Only v2 have ss database
+	if binary == "simdv2" {
+		err = os.RemoveAll(fmt.Sprintf("%s/data/ss", nodeDir))
+		require.NoError(t, err)
+	}
+
+	res = cli.RunCommandWithArgs(snapshotPrefix, "restore", "5", "3", fmt.Sprintf("--home=%s", nodeDir))
+	require.DirExists(t, fmt.Sprintf("%s/data/application.db", nodeDir))
+	if binary == "simdv2" {
+		require.DirExists(t, fmt.Sprintf("%s/data/ss", nodeDir))
+	}
+
+	// Start the node
+	sut.StartSingleNode(t, nodeDir)
+}
+
+func TestPrune(t *testing.T) {
+	sut.ResetChain(t)
+	cli := NewCLIWrapper(t, sut, verbose)
+	// add genesis account with some tokens
+	account1Addr := cli.AddKey("account1")
+	sut.ModifyGenesisCLI(t,
+		[]string{"genesis", "add-genesis-account", account1Addr, "10000000stake"},
+	)
+
+	sut.StartChain(t)
+
+	binary, prefix := getBinaryNameAndPrefix(false)
+	nodeDir := fmt.Sprintf("./testnet/node0/%s", binary)
+
+	// Wait for chain produce some blocks
+	time.Sleep(time.Second * 10)
+	// Stop 1 node
+	err := sut.StopSingleNode()
+	require.NoError(t, err)
+	time.Sleep(time.Second)
+
+	// prune
+	var res string
+	if binary == "simdv2" {
+		res = cli.RunCommandWithArgs(prefix, "prune", "--keep-recent=1", fmt.Sprintf("--home=%s", nodeDir))
+	} else {
+		res = cli.RunCommandWithArgs("prune", "everything", fmt.Sprintf("--home=%s", nodeDir))
+	}
+	require.Contains(t, res, "successfully pruned the application root multi stores")
+	// Start the node
+	sut.StartSingleNode(t, nodeDir)
+}

tests/systemtests/system.go

@@ -331,6 +331,37 @@
 	s.ChainStarted = false
 }
 
+// StopSingleNode stops a validator node without stop the chain running
+func (s *SystemUnderTest) StopSingleNode() error {
+	if !s.ChainStarted {
+		return nil
+	}
+
+	s.pidsLock.RLock()
+	pids := maps.Keys(s.pids)
+	s.pidsLock.RUnlock()
+
+	p, err := os.FindProcess(pids[0])
+	if err != nil {
+		return err
+	}
+
+	// Kill the 1st node
+	return p.Kill()
+}
+
+// StartSingleNode start running a validator node with dir input
+func (s *SystemUnderTest) StartSingleNode(t *testing.T, dir string) {
+	t.Helper()
+	cmd := exec.Command( //nolint:gosec // used by tests only
+		locateExecutable(s.execBinary),
+		[]string{"start", "--log_level=info", "--log_no_color"}...,
+	)
+	cmd.Dir = WorkDir
+	err := cmd.Start()
+	require.NoError(t, err)
+}
+
 func (s *SystemUnderTest) withEachPid(cb func(p *os.Process)) {
 	s.pidsLock.RLock()
 	pids := maps.Keys(s.pids)