github.com/Finschia/finschia-sdk@v0.49.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.NoError(t, err)
    55  
    56  	// User confirmation missing
    57  	cmd.SetArgs([]string{
    58  		fakeKeyName1,
    59  		fmt.Sprintf("--%s=%s", flags.FlagHome, kbHome),
    60  		fmt.Sprintf("--%s=%s", flags.FlagKeyringBackend, keyring.BackendTest),
    61  	})
    62  	err = cmd.Execute()
    63  	require.Error(t, err)
    64  	require.Equal(t, "EOF", err.Error())
    65  
    66  	_, err = kb.Key(fakeKeyName1)
    67  	require.NoError(t, err)
    68  
    69  	// Now there is a confirmation
    70  	cmd.SetArgs([]string{
    71  		fakeKeyName1,
    72  		fmt.Sprintf("--%s=%s", flags.FlagHome, kbHome),
    73  		fmt.Sprintf("--%s=true", flagYes),
    74  		fmt.Sprintf("--%s=%s", flags.FlagKeyringBackend, keyring.BackendTest),
    75  	})
    76  	require.NoError(t, cmd.Execute())
    77  
    78  	_, err = kb.Key(fakeKeyName1)
    79  	require.Error(t, err) // Key1 is gone
    80  
    81  	_, err = kb.Key(fakeKeyName2)
    82  	require.NoError(t, err)
    83  
    84  	cmd.SetArgs([]string{
    85  		fakeKeyName2,
    86  		fmt.Sprintf("--%s=%s", flags.FlagHome, kbHome),
    87  		fmt.Sprintf("--%s=true", flagYes),
    88  		fmt.Sprintf("--%s=%s", flags.FlagKeyringBackend, keyring.BackendTest),
    89  	})
    90  	require.NoError(t, cmd.Execute())
    91  
    92  	_, err = kb.Key(fakeKeyName2)
    93  	require.Error(t, err) // Key2 is gone
    94  }