github.com/KiraCore/sekai@v0.3.43/x/layer2/keeper/bridge_registrar_test.go (about)

     1  package keeper_test
     2  
     3  import (
     4  	"github.com/KiraCore/sekai/x/layer2/types"
     5  	sdk "github.com/cosmos/cosmos-sdk/types"
     6  )
     7  
     8  func (suite *KeeperTestSuite) TestBridgeRegistrarHelperSetGet() {
     9  	helper := types.BridgeRegistrarHelper{
    10  		NextUser:  1,
    11  		NextXam:   1,
    12  		NextToken: 1,
    13  	}
    14  	suite.app.Layer2Keeper.SetBridgeRegistrarHelper(suite.ctx, helper)
    15  
    16  	saved := suite.app.Layer2Keeper.GetBridgeRegistrarHelper(suite.ctx)
    17  	suite.Require().Equal(saved, helper)
    18  }
    19  
    20  func (suite *KeeperTestSuite) TestBridgeAccountSetGet() {
    21  	infos := []types.BridgeAccount{
    22  		{
    23  			Index:    1,
    24  			Address:  "kira15ky9du8a2wlstz6fpx3p4mqpjyrm5cgqzp4f3d",
    25  			DappName: "",
    26  			Balances: []types.BridgeBalance(nil),
    27  		},
    28  		{
    29  			Index:    2,
    30  			Address:  "kira1alzyfq40zjsveat87jlg8jxetwqmr0a29sgd0f",
    31  			DappName: "dapp1",
    32  			Balances: []types.BridgeBalance{
    33  				{
    34  					BridgeTokenIndex: 1,
    35  					Amount:           sdk.OneInt(),
    36  				},
    37  			},
    38  		},
    39  	}
    40  
    41  	for _, info := range infos {
    42  		suite.app.Layer2Keeper.SetBridgeAccount(suite.ctx, info)
    43  	}
    44  
    45  	for _, info := range infos {
    46  		c := suite.app.Layer2Keeper.GetBridgeAccount(suite.ctx, info.Index)
    47  		suite.Require().Equal(c, info)
    48  	}
    49  
    50  	accounts := suite.app.Layer2Keeper.GetBridgeAccounts(suite.ctx)
    51  	suite.Require().Len(accounts, 2)
    52  }
    53  
    54  func (suite *KeeperTestSuite) TestBridgeTokenSetGet() {
    55  	infos := []types.BridgeToken{
    56  		{
    57  			Index: 1,
    58  			Denom: "btc",
    59  		},
    60  		{
    61  			Index: 2,
    62  			Denom: "eth",
    63  		},
    64  	}
    65  
    66  	for _, info := range infos {
    67  		suite.app.Layer2Keeper.SetBridgeToken(suite.ctx, info)
    68  	}
    69  
    70  	for _, info := range infos {
    71  		c := suite.app.Layer2Keeper.GetBridgeToken(suite.ctx, info.Index)
    72  		suite.Require().Equal(c, info)
    73  	}
    74  
    75  	tokens := suite.app.Layer2Keeper.GetBridgeTokens(suite.ctx)
    76  	suite.Require().Len(tokens, 2)
    77  }
    78  
    79  func (suite *KeeperTestSuite) TestXAMSetGet() {
    80  	infos := []types.XAM{
    81  		{
    82  			Req: types.XAMRequest{
    83  				Amounts:         []types.BridgeBalance(nil),
    84  				SourceDapp:      1,
    85  				SourceAccount:   1,
    86  				DestDapp:        2,
    87  				DestBeneficiary: 1,
    88  				Xam:             "",
    89  			},
    90  			Res: types.XAMResponse{
    91  				Xid: 1,
    92  				Irc: 0,
    93  				Src: 0,
    94  				Drc: 0,
    95  				Irm: 0,
    96  				Srm: 0,
    97  				Drm: 0,
    98  			},
    99  		},
   100  		{
   101  			Req: types.XAMRequest{
   102  				Amounts: []types.BridgeBalance{
   103  					{
   104  						BridgeTokenIndex: 1,
   105  						Amount:           sdk.OneInt(),
   106  					},
   107  				},
   108  				SourceDapp:      1,
   109  				SourceAccount:   1,
   110  				DestDapp:        2,
   111  				DestBeneficiary: 1,
   112  				Xam:             "",
   113  			},
   114  			Res: types.XAMResponse{
   115  				Xid: 2,
   116  				Irc: 0,
   117  				Src: 0,
   118  				Drc: 0,
   119  				Irm: 0,
   120  				Srm: 0,
   121  				Drm: 0,
   122  			},
   123  		},
   124  	}
   125  
   126  	for _, info := range infos {
   127  		suite.app.Layer2Keeper.SetXAM(suite.ctx, info)
   128  	}
   129  
   130  	for _, info := range infos {
   131  		c := suite.app.Layer2Keeper.GetXAM(suite.ctx, info.Res.Xid)
   132  		suite.Require().Equal(c, info)
   133  	}
   134  
   135  	xams := suite.app.Layer2Keeper.GetXAMs(suite.ctx)
   136  	suite.Require().Len(xams, 2)
   137  }