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

     1  package keeper_test
     2  
     3  import (
     4  	"github.com/KiraCore/sekai/x/tokens/types"
     5  )
     6  
     7  func (suite *KeeperTestSuite) TestTokenBlacklist() {
     8  	ctx := suite.ctx
     9  
    10  	// check genesis
    11  	blackWhites := suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    12  	suite.Require().Equal(blackWhites.Blacklisted, []string{"frozen"})
    13  
    14  	// try adding two tokens
    15  	suite.app.TokensKeeper.AddTokensToBlacklist(ctx, []string{"frozen1", "frozen2"})
    16  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    17  	suite.Require().Equal(blackWhites.Blacklisted, []string{"frozen", "frozen1", "frozen2"})
    18  
    19  	// try adding one token
    20  	suite.app.TokensKeeper.AddTokensToBlacklist(ctx, []string{"frozen3"})
    21  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    22  	suite.Require().Equal(blackWhites.Blacklisted, []string{"frozen", "frozen1", "frozen2", "frozen3"})
    23  
    24  	// try adding no changes
    25  	suite.app.TokensKeeper.AddTokensToBlacklist(ctx, []string{})
    26  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    27  	suite.Require().Equal(blackWhites.Blacklisted, []string{"frozen", "frozen1", "frozen2", "frozen3"})
    28  
    29  	// try adding empty denom
    30  	suite.app.TokensKeeper.AddTokensToBlacklist(ctx, []string{""})
    31  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    32  	suite.Require().Equal(blackWhites.Blacklisted, []string{"frozen", "frozen1", "frozen2", "frozen3", ""})
    33  
    34  	// try removing blacklisted
    35  	suite.app.TokensKeeper.RemoveTokensFromBlacklist(ctx, []string{"frozen2"})
    36  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    37  	suite.Require().Equal(blackWhites.Blacklisted, []string{"frozen", "frozen1", "", "frozen3"})
    38  
    39  	// try removing not blacklisted
    40  	suite.app.TokensKeeper.RemoveTokensFromBlacklist(ctx, []string{"frozen4"})
    41  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    42  	suite.Require().Equal(blackWhites.Blacklisted, []string{"frozen", "frozen1", "", "frozen3"})
    43  }
    44  
    45  func (suite *KeeperTestSuite) TestTokenWhitelist() {
    46  	ctx := suite.ctx
    47  
    48  	// check genesis
    49  	blackWhites := suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    50  	suite.Require().Equal(blackWhites.Whitelisted, []string{"ukex"})
    51  
    52  	// try adding two tokens
    53  	suite.app.TokensKeeper.AddTokensToWhitelist(ctx, []string{"white1", "white2"})
    54  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    55  	suite.Require().Equal(blackWhites.Whitelisted, []string{"ukex", "white1", "white2"})
    56  
    57  	// try adding one token
    58  	suite.app.TokensKeeper.AddTokensToWhitelist(ctx, []string{"white3"})
    59  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    60  	suite.Require().Equal(blackWhites.Whitelisted, []string{"ukex", "white1", "white2", "white3"})
    61  
    62  	// try adding no changes
    63  	suite.app.TokensKeeper.AddTokensToWhitelist(ctx, []string{})
    64  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    65  	suite.Require().Equal(blackWhites.Whitelisted, []string{"ukex", "white1", "white2", "white3"})
    66  
    67  	// try adding empty denom
    68  	suite.app.TokensKeeper.AddTokensToWhitelist(ctx, []string{""})
    69  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    70  	suite.Require().Equal(blackWhites.Whitelisted, []string{"ukex", "white1", "white2", "white3", ""})
    71  
    72  	// try removing whitelisted
    73  	suite.app.TokensKeeper.RemoveTokensFromWhitelist(ctx, []string{"white2"})
    74  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    75  	suite.Require().Equal(blackWhites.Whitelisted, []string{"ukex", "white1", "", "white3"})
    76  
    77  	// try removing not whitelisted
    78  	suite.app.TokensKeeper.RemoveTokensFromWhitelist(ctx, []string{"white4"})
    79  	blackWhites = suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    80  	suite.Require().Equal(blackWhites.Whitelisted, []string{"ukex", "white1", "", "white3"})
    81  }
    82  
    83  func (suite *KeeperTestSuite) TestTokenBlackWhiteSetGet() {
    84  	ctx := suite.ctx
    85  
    86  	blackWhites := types.TokensWhiteBlack{
    87  		Whitelisted: []string{"newwhite"},
    88  		Blacklisted: []string{"newblack"},
    89  	}
    90  	suite.app.TokensKeeper.SetTokenBlackWhites(ctx, &blackWhites)
    91  	bw := suite.app.TokensKeeper.GetTokenBlackWhites(ctx)
    92  	suite.Require().NotNil(bw)
    93  	suite.Require().Equal(blackWhites, *bw)
    94  }