github.com/iotexproject/iotex-core@v1.14.1-rc1/ioctl/newcmd/alias/aliasremove_test.go (about)

     1  // Copyright (c) 2022 IoTeX Foundation
     2  // This source code is provided 'as is' and no warranties are given as to title or non-infringement, merchantability
     3  // or fitness for purpose and, to the extent permitted by law, all liability for your use of the code is disclaimed.
     4  // This source code is governed by Apache License 2.0 that can be found in the LICENSE file.
     5  package alias
     6  
     7  import (
     8  	"testing"
     9  
    10  	"github.com/golang/mock/gomock"
    11  	"github.com/iotexproject/iotex-core/ioctl/config"
    12  	"github.com/iotexproject/iotex-core/ioctl/util"
    13  	"github.com/iotexproject/iotex-core/test/mock/mock_ioctlclient"
    14  	"github.com/stretchr/testify/require"
    15  )
    16  
    17  func TestNewAliasRemoveCmd(t *testing.T) {
    18  	require := require.New(t)
    19  	// mock a client
    20  	ctrl := gomock.NewController(t)
    21  	client := mock_ioctlclient.NewMockClient(ctrl)
    22  	cfg := config.Config{
    23  		Aliases: map[string]string{
    24  			"io1uwnr55vqmhf3xeg5phgurlyl702af6eju542sx": "a",
    25  			"io1uwnr55vqmhf3xeg5phgurlyl702af6eju542se": "b",
    26  			"io1uwnr55vqmhf3xeg5phgurlyl702af6eju542s1": "c",
    27  		},
    28  	}
    29  	client.EXPECT().AliasMap().Return(map[string]string{
    30  		"io1uwnr55vqmhf3xeg5phgurlyl702af6eju542sx": "a",
    31  		"io1uwnr55vqmhf3xeg5phgurlyl702af6eju542se": "b",
    32  		"io1uwnr55vqmhf3xeg5phgurlyl702af6eju542s1": "c",
    33  	}).AnyTimes()
    34  	client.EXPECT().Config().Return(cfg).AnyTimes()
    35  
    36  	t.Run("remove alias", func(t *testing.T) {
    37  		client.EXPECT().SelectTranslation(gomock.Any()).Return("%s is removed", config.English).Times(5)
    38  		client.EXPECT().DeleteAlias("a").Return(nil)
    39  		cmd := NewAliasRemove(client)
    40  		result, err := util.ExecuteCmd(cmd, "a")
    41  		require.NoError(err)
    42  		require.Contains(result, "a is removed")
    43  	})
    44  
    45  	t.Run("invalid alias", func(t *testing.T) {
    46  		client.EXPECT().SelectTranslation(gomock.Any()).Return("invalid alias %s", config.English).AnyTimes()
    47  		cmd := NewAliasRemove(client)
    48  		_, err := util.ExecuteCmd(cmd, "io1uwnr55vqmhf3xeg5phgurlyl702af6eju542se")
    49  		require.Error(err)
    50  		require.Contains(err.Error(), "invalid alias io1uwnr55vqmhf3xeg5phgurlyl702af6eju542se")
    51  	})
    52  }