github.com/Finschia/finschia-sdk@v0.48.1/client/keys/delete_test.go (about)

     1  package keys
     2  
     3  import (
     4  	"context"
     5  	"fmt"
     6  	"testing"
     7  
     8  	"github.com/stretchr/testify/require"
     9  
    10  	"github.com/Finschia/finschia-sdk/client"
    11  	"github.com/Finschia/finschia-sdk/client/flags"
    12  	"github.com/Finschia/finschia-sdk/crypto/hd"
    13  	"github.com/Finschia/finschia-sdk/crypto/keyring"
    14  	"github.com/Finschia/finschia-sdk/testutil"
    15  	"github.com/Finschia/finschia-sdk/testutil/testdata"
    16  	sdk "github.com/Finschia/finschia-sdk/types"
    17  )
    18  
    19  func Test_runDeleteCmd(t *testing.T) {
    20  	// Now add a temporary keybase
    21  	kbHome := t.TempDir()
    22  	cmd := DeleteKeyCommand()
    23  	cmd.Flags().AddFlagSet(Commands(kbHome).PersistentFlags())
    24  	mockIn := testutil.ApplyMockIODiscardOutErr(cmd)
    25  
    26  	yesF, _ := cmd.Flags().GetBool(flagYes)
    27  	forceF, _ := cmd.Flags().GetBool(flagForce)
    28  
    29  	require.False(t, yesF)
    30  	require.False(t, forceF)
    31  
    32  	fakeKeyName1 := "runDeleteCmd_Key1"
    33  	fakeKeyName2 := "runDeleteCmd_Key2"
    34  
    35  	path := sdk.GetConfig().GetFullBIP44Path()
    36  
    37  	cmd.SetArgs([]string{"blah", fmt.Sprintf("--%s=%s", flags.FlagHome, kbHome)})
    38  	kb, err := keyring.New(sdk.KeyringServiceName(), keyring.BackendTest, kbHome, mockIn)
    39  	require.NoError(t, err)
    40  
    41  	_, err = kb.NewAccount(fakeKeyName1, testdata.TestMnemonic, "", path, hd.Secp256k1)
    42  	require.NoError(t, err)
    43  
    44  	_, _, err = kb.NewMnemonic(fakeKeyName2, keyring.English, sdk.FullFundraiserPath, keyring.DefaultBIP39Passphrase, hd.Secp256k1)
    45  	require.NoError(t, err)
    46  
    47  	clientCtx := client.Context{}.
    48  		WithKeyringDir(kbHome).
    49  		WithKeyring(kb)
    50  
    51  	ctx := context.WithValue(context.Background(), client.ClientContextKey, &clientCtx)
    52  
    53  	err = cmd.ExecuteContext(ctx)
    54  	require.Error(t, err)
    55  	require.EqualError(t, err, "blah.info: key not found")
    56  
    57  	// User confirmation missing
    58  	cmd.SetArgs([]string{
    59  		fakeKeyName1,
    60  		fmt.Sprintf("--%s=%s", flags.FlagHome, kbHome),
    61  		fmt.Sprintf("--%s=%s", flags.FlagKeyringBackend, keyring.BackendTest),
    62  	})
    63  	err = cmd.Execute()
    64  	require.Error(t, err)
    65  	require.Equal(t, "EOF", err.Error())
    66  
    67  	_, err = kb.Key(fakeKeyName1)
    68  	require.NoError(t, err)
    69  
    70  	// Now there is a confirmation
    71  	cmd.SetArgs([]string{
    72  		fakeKeyName1,
    73  		fmt.Sprintf("--%s=%s", flags.FlagHome, kbHome),
    74  		fmt.Sprintf("--%s=true", flagYes),
    75  		fmt.Sprintf("--%s=%s", flags.FlagKeyringBackend, keyring.BackendTest),
    76  	})
    77  	require.NoError(t, cmd.Execute())
    78  
    79  	_, err = kb.Key(fakeKeyName1)
    80  	require.Error(t, err) // Key1 is gone
    81  
    82  	_, err = kb.Key(fakeKeyName2)
    83  	require.NoError(t, err)
    84  
    85  	cmd.SetArgs([]string{
    86  		fakeKeyName2,
    87  		fmt.Sprintf("--%s=%s", flags.FlagHome, kbHome),
    88  		fmt.Sprintf("--%s=true", flagYes),
    89  		fmt.Sprintf("--%s=%s", flags.FlagKeyringBackend, keyring.BackendTest),
    90  	})
    91  	require.NoError(t, cmd.Execute())
    92  
    93  	_, err = kb.Key(fakeKeyName2)
    94  	require.Error(t, err) // Key2 is gone
    95  }