prysm-pulse/validator/accounts/v2/import_test.go

60 lines
2.0 KiB
Go
Raw Normal View History

package v2
import (
"flag"
"io/ioutil"
"os"
"path/filepath"
"testing"
"github.com/prysmaticlabs/prysm/shared/testutil"
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
"github.com/prysmaticlabs/prysm/shared/testutil/require"
"github.com/prysmaticlabs/prysm/validator/flags"
"github.com/urfave/cli/v2"
)
func TestImport_Noninteractive(t *testing.T) {
testDir := testutil.TempDir()
walletDir := filepath.Join(testDir, walletDirName)
passwordsDir := filepath.Join(testDir, passwordDirName)
exportDir := filepath.Join(testDir, exportDirName)
importDir := filepath.Join(testDir, importDirName)
importPasswordDir := filepath.Join(testDir, importPasswordDirName)
passwordFilePath := filepath.Join(testDir, passwordFileName)
require.NoError(t, ioutil.WriteFile(passwordFilePath, []byte(password), os.ModePerm))
defer func() {
assert.NoError(t, os.RemoveAll(walletDir))
assert.NoError(t, os.RemoveAll(passwordsDir))
assert.NoError(t, os.RemoveAll(exportDir))
assert.NoError(t, os.RemoveAll(importDir))
assert.NoError(t, os.RemoveAll(importPasswordDir))
}()
wallet := setupWallet(t, testDir)
accounts, err := wallet.AccountNames()
require.NoError(t, err)
assert.Equal(t, len(accounts), 1)
require.NoError(t, wallet.zipAccounts(accounts, exportDir))
if _, err := os.Stat(filepath.Join(exportDir, archiveFilename)); os.IsNotExist(err) {
t.Fatal("Expected file to exist")
}
app := cli.App{}
set := flag.NewFlagSet("test", 0)
set.String(flags.WalletDirFlag.Name, importDir, "")
set.String(flags.WalletPasswordsDirFlag.Name, importPasswordDir, "")
set.String(flags.BackupPathFlag.Name, exportDir, "")
set.String(flags.PasswordFileFlag.Name, passwordFilePath, "")
assert.NoError(t, set.Set(flags.WalletDirFlag.Name, importDir))
assert.NoError(t, set.Set(flags.WalletPasswordsDirFlag.Name, importPasswordDir))
assert.NoError(t, set.Set(flags.BackupPathFlag.Name, exportDir))
assert.NoError(t, set.Set(flags.PasswordFileFlag.Name, passwordFilePath))
cliCtx := cli.NewContext(&app, set, nil)
require.NoError(t, ImportAccount(cliCtx))
}