package db import ( "context" "flag" "os" "path" "testing" "github.com/prysmaticlabs/prysm/v5/beacon-chain/db/kv" "github.com/prysmaticlabs/prysm/v5/cmd" "github.com/prysmaticlabs/prysm/v5/consensus-types/blocks" "github.com/prysmaticlabs/prysm/v5/consensus-types/primitives" "github.com/prysmaticlabs/prysm/v5/testing/assert" "github.com/prysmaticlabs/prysm/v5/testing/require" "github.com/prysmaticlabs/prysm/v5/testing/util" logTest "github.com/sirupsen/logrus/hooks/test" "github.com/urfave/cli/v2" ) func TestRestore(t *testing.T) { logHook := logTest.NewGlobal() ctx := context.Background() backupDb, err := kv.NewKVStore(context.Background(), t.TempDir()) require.NoError(t, err) head := util.NewBeaconBlock() head.Block.Slot = 5000 wsb, err := blocks.NewSignedBeaconBlock(head) require.NoError(t, err) require.NoError(t, backupDb.SaveBlock(ctx, wsb)) root, err := head.Block.HashTreeRoot() require.NoError(t, err) st, err := util.NewBeaconState() require.NoError(t, err) require.NoError(t, backupDb.SaveState(ctx, st, root)) require.NoError(t, backupDb.SaveHeadBlockRoot(ctx, root)) require.NoError(t, err) require.NoError(t, backupDb.Close()) // We rename the backup file so that we can later verify // whether the restored db has been renamed correctly. require.NoError(t, os.Rename( path.Join(backupDb.DatabasePath(), kv.DatabaseFileName), path.Join(backupDb.DatabasePath(), "backup.db"))) restoreDir := t.TempDir() app := cli.App{} set := flag.NewFlagSet("test", 0) set.String(cmd.RestoreSourceFileFlag.Name, "", "") set.String(cmd.RestoreTargetDirFlag.Name, "", "") require.NoError(t, set.Set(cmd.RestoreSourceFileFlag.Name, path.Join(backupDb.DatabasePath(), "backup.db"))) require.NoError(t, set.Set(cmd.RestoreTargetDirFlag.Name, restoreDir)) cliCtx := cli.NewContext(&app, set, nil) assert.NoError(t, Restore(cliCtx)) files, err := os.ReadDir(path.Join(restoreDir, kv.BeaconNodeDbDirName)) require.NoError(t, err) assert.Equal(t, 1, len(files)) assert.Equal(t, kv.DatabaseFileName, files[0].Name()) restoredDb, err := kv.NewKVStore(context.Background(), path.Join(restoreDir, kv.BeaconNodeDbDirName)) defer func() { require.NoError(t, restoredDb.Close()) }() require.NoError(t, err) headBlock, err := restoredDb.HeadBlock(ctx) require.NoError(t, err) assert.Equal(t, primitives.Slot(5000), headBlock.Block().Slot(), "Restored database has incorrect data") assert.LogsContain(t, logHook, "Restore completed successfully") }