github.com/prysmaticlabs/prysm@v1.4.4/validator/accounts/wallet/wallet_test.go (about)

     1  package wallet_test
     2  
     3  import (
     4  	"io/ioutil"
     5  	"os"
     6  	"path/filepath"
     7  	"testing"
     8  
     9  	"github.com/prysmaticlabs/prysm/shared/params"
    10  	"github.com/prysmaticlabs/prysm/shared/testutil/require"
    11  	"github.com/prysmaticlabs/prysm/validator/accounts/wallet"
    12  	"github.com/sirupsen/logrus"
    13  )
    14  
    15  func init() {
    16  	logrus.SetLevel(logrus.DebugLevel)
    17  	logrus.SetOutput(ioutil.Discard)
    18  }
    19  
    20  func Test_Exists_RandomFiles(t *testing.T) {
    21  	path := filepath.Join(t.TempDir(), "wallet")
    22  
    23  	exists, err := wallet.Exists(path)
    24  	require.Equal(t, false, exists)
    25  	require.NoError(t, err)
    26  	require.NoError(t, os.MkdirAll(path+"/direct", params.BeaconIoConfig().ReadWriteExecutePermissions), "Failed to create directory")
    27  
    28  	exists, err = wallet.Exists(path)
    29  	require.NoError(t, err)
    30  	require.Equal(t, true, exists)
    31  }
    32  
    33  func Test_IsValid_RandomFiles(t *testing.T) {
    34  	path := filepath.Join(t.TempDir(), "wallet")
    35  	valid, err := wallet.IsValid(path)
    36  	require.NoError(t, err)
    37  	require.Equal(t, false, valid)
    38  
    39  	require.NoError(t, os.MkdirAll(path, params.BeaconIoConfig().ReadWriteExecutePermissions), "Failed to create directory")
    40  
    41  	valid, err = wallet.IsValid(path)
    42  	require.ErrorContains(t, "no wallet found", err)
    43  	require.Equal(t, false, valid)
    44  
    45  	walletDir := filepath.Join(path, "direct")
    46  	require.NoError(t, os.MkdirAll(walletDir, params.BeaconIoConfig().ReadWriteExecutePermissions), "Failed to create directory")
    47  
    48  	valid, err = wallet.IsValid(path)
    49  	require.NoError(t, err)
    50  	require.Equal(t, true, valid)
    51  }