test_common.go

package ammswap

import (
	sdk "github.com/FiboChain/fbc/libs/cosmos-sdk/types"
	swapkeeper "github.com/FiboChain/fbc/x/ammswap/keeper"
	"github.com/FiboChain/fbc/x/ammswap/types"
	"github.com/stretchr/testify/require"
	"testing"
	"time"
)

func NewTestSwapTokenPairWithInitLiquidity(t *testing.T, ctx sdk.Context, k swapkeeper.Keeper,
	baseToken, quoteToken sdk.DecCoin, addrList []sdk.AccAddress) SwapTokenPair {
	handler := NewHandler(k)

	createExchangeMsg := types.NewMsgCreateExchange(baseToken.Denom, quoteToken.Denom, addrList[0])
	_, err := handler(ctx, createExchangeMsg)
	require.Nil(t, err)
	deadLine := time.Now().Unix()
	addLiquidityMsg := types.NewMsgAddLiquidity(sdk.NewDec(0), baseToken, quoteToken, deadLine, addrList[0])
	_, err = handler(ctx, addLiquidityMsg)
	require.Nil(t, err)
	for _, addr := range addrList {
		baseToken1 := sdk.NewDecCoinFromDec(baseToken.Denom, baseToken.Amount.Mul(sdk.NewDec(100)))
		quoteToken1 := sdk.NewDecCoinFromDec(quoteToken.Denom, quoteToken.Amount.Mul(sdk.NewDec(100)))
		addLiquidityMsg := types.NewMsgAddLiquidity(sdk.NewDec(0), baseToken1, quoteToken1, deadLine, addr)
		_, err = handler(ctx, addLiquidityMsg)
		require.Nil(t, err)
	}

	swapTokenPair, err := k.GetSwapTokenPair(ctx, types.GetSwapTokenPairName(baseToken.Denom, quoteToken.Denom))
	require.Nil(t, err)

	return swapTokenPair
}