github.com/fibonacci-chain/fbc@v0.0.0-20231124064014-c7636198c1e9/x/ammswap/types/test_common.go (about)

     1  package types
     2  
     3  import (
     4  	"time"
     5  
     6  	sdk "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types"
     7  	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/x/supply"
     8  	"github.com/fibonacci-chain/fbc/x/common"
     9  	"github.com/fibonacci-chain/fbc/x/token"
    10  	tokentypes "github.com/fibonacci-chain/fbc/x/token/types"
    11  )
    12  
    13  // nolint
    14  const TestBasePooledToken = "aab"
    15  const TestBasePooledToken2 = "ccb"
    16  const TestBasePooledToken3 = "ddb"
    17  const TestQuotePooledToken = common.NativeToken
    18  const TestSwapTokenPairName = TestBasePooledToken + "_" + TestQuotePooledToken
    19  
    20  // GetTestSwapTokenPair just for test
    21  func GetTestSwapTokenPair() SwapTokenPair {
    22  	return SwapTokenPair{
    23  		QuotePooledCoin: sdk.NewDecCoinFromDec(TestQuotePooledToken, sdk.NewDec(0)),
    24  		BasePooledCoin:  sdk.NewDecCoinFromDec(TestBasePooledToken, sdk.NewDec(0)),
    25  		PoolTokenName:   GetPoolTokenName(TestBasePooledToken, TestQuotePooledToken),
    26  	}
    27  }
    28  
    29  func SetTestTokens(ctx sdk.Context, tokenKeeper token.Keeper, supplyKeeper supply.Keeper, addr sdk.AccAddress, coins sdk.DecCoins) error {
    30  	for _, coin := range coins {
    31  		name := coin.Denom
    32  		tokenKeeper.NewToken(ctx, tokentypes.Token{"", name, name, name, coin.Amount, 1, addr, true})
    33  	}
    34  	err := supplyKeeper.MintCoins(ctx, tokentypes.ModuleName, coins)
    35  	if err != nil {
    36  		return err
    37  	}
    38  	err = supplyKeeper.SendCoinsFromModuleToAccount(ctx, tokentypes.ModuleName, addr, coins)
    39  	if err != nil {
    40  		return err
    41  	}
    42  	return nil
    43  }
    44  
    45  func CreateTestMsgs(addr sdk.AccAddress) []sdk.Msg {
    46  	return []sdk.Msg{
    47  		NewMsgCreateExchange(TestBasePooledToken, TestQuotePooledToken, addr),
    48  		NewMsgCreateExchange(TestBasePooledToken2, TestQuotePooledToken, addr),
    49  		NewMsgAddLiquidity(sdk.ZeroDec(),
    50  			sdk.NewDecCoin(TestBasePooledToken, sdk.OneInt()), sdk.NewDecCoin(TestQuotePooledToken, sdk.OneInt()),
    51  			time.Now().Add(time.Hour).Unix(), addr),
    52  		NewMsgAddLiquidity(sdk.ZeroDec(),
    53  			sdk.NewDecCoin(TestBasePooledToken2, sdk.OneInt()), sdk.NewDecCoin(TestQuotePooledToken, sdk.OneInt()),
    54  			time.Now().Add(time.Hour).Unix(), addr),
    55  		NewMsgRemoveLiquidity(sdk.OneDec(),
    56  			sdk.NewDecCoin(TestBasePooledToken2, sdk.OneInt()), sdk.NewDecCoin(TestQuotePooledToken, sdk.OneInt()),
    57  			time.Now().Add(time.Hour).Unix(), addr),
    58  	}
    59  }