github.com/KiraCore/sekai@v0.3.43/x/tokens/keeper/alias_test.go (about)

     1  package keeper_test
     2  
     3  import (
     4  	"github.com/KiraCore/sekai/x/tokens/types"
     5  )
     6  
     7  func (suite *KeeperTestSuite) TestTokenAlias() {
     8  	suite.SetupTest()
     9  	ctx := suite.ctx
    10  
    11  	// check initial token alias before registration
    12  	alias := suite.app.TokensKeeper.GetTokenAlias(ctx, "stake")
    13  	suite.Require().Nil(alias)
    14  	aliases := suite.app.TokensKeeper.ListTokenAlias(ctx)
    15  	suite.Require().Len(aliases, 1)
    16  	aliasMap := suite.app.TokensKeeper.GetTokenAliasesByDenom(ctx, []string{"stake"})
    17  	suite.Require().Equal(len(aliasMap), 0)
    18  	suite.Require().Nil(aliasMap["stake"])
    19  
    20  	// upsert token alias and check
    21  	newAlias := types.TokenAlias{
    22  		Symbol:   "stake",
    23  		Name:     "test token",
    24  		Icon:     "icon_link",
    25  		Decimals: 6,
    26  		Denoms:   []string{"stake"},
    27  	}
    28  	suite.app.TokensKeeper.UpsertTokenAlias(ctx, newAlias)
    29  	alias = suite.app.TokensKeeper.GetTokenAlias(ctx, "stake")
    30  	suite.Require().NotNil(alias)
    31  	aliases = suite.app.TokensKeeper.ListTokenAlias(ctx)
    32  	suite.Require().Len(aliases, 2)
    33  	aliasMap = suite.app.TokensKeeper.GetTokenAliasesByDenom(ctx, []string{"stake"})
    34  	suite.Require().Equal(len(aliasMap), 1)
    35  	suite.Require().NotNil(aliasMap["stake"])
    36  
    37  	// delete token alias and check
    38  	suite.app.TokensKeeper.DeleteTokenAlias(ctx, "stake")
    39  	alias = suite.app.TokensKeeper.GetTokenAlias(ctx, "stake")
    40  	suite.Require().Nil(alias)
    41  	aliases = suite.app.TokensKeeper.ListTokenAlias(ctx)
    42  	suite.Require().Len(aliases, 1)
    43  	aliasMap = suite.app.TokensKeeper.GetTokenAliasesByDenom(ctx, []string{"stake"})
    44  	suite.Require().Equal(len(aliasMap), 0)
    45  	suite.Require().Nil(aliasMap["stake"])
    46  }