github.com/fibonacci-chain/fbc@v0.0.0-20231124064014-c7636198c1e9/x/farm/keeper/whitelist_test.go (about)

     1  //go:build ignore
     2  // +build ignore
     3  
     4  package keeper
     5  
     6  import (
     7  	"testing"
     8  
     9  	swaptypes "github.com/fibonacci-chain/fbc/x/ammswap/types"
    10  	"github.com/fibonacci-chain/fbc/x/farm/types"
    11  
    12  	sdk "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types"
    13  	"github.com/stretchr/testify/require"
    14  )
    15  
    16  func TestSatisfyWhiteListAdmittance(t *testing.T) {
    17  	ctx, k := GetKeeper(t)
    18  	quoteSymbol := types.DefaultParams().QuoteSymbol
    19  
    20  	baseSymbol := "xxb"
    21  	pool1 := types.FarmPool{
    22  		Name:          "pool1",
    23  		MinLockAmount: sdk.NewDecCoinFromDec(baseSymbol, sdk.ZeroDec()),
    24  	}
    25  	k.SetFarmPool(ctx, pool1)
    26  	err := k.satisfyWhiteListAdmittance(ctx, pool1)
    27  	require.NotNil(t, err)
    28  
    29  	// test add LPT
    30  	lockSymbol := swaptypes.PoolTokenPrefix + baseSymbol + "_" + quoteSymbol
    31  	pool2 := types.FarmPool{
    32  		Name:          "pool2",
    33  		MinLockAmount: sdk.NewDecCoinFromDec(lockSymbol, sdk.ZeroDec()),
    34  	}
    35  	k.SetFarmPool(ctx, pool2)
    36  	err = k.satisfyWhiteListAdmittance(ctx, pool2)
    37  	require.NotNil(t, err)
    38  
    39  	base1Symbol := "okb"
    40  	lockSymbol = swaptypes.PoolTokenPrefix + baseSymbol + "_" + base1Symbol
    41  	pool3 := types.FarmPool{
    42  		Name:          "pool3",
    43  		MinLockAmount: sdk.NewDecCoinFromDec(lockSymbol, sdk.ZeroDec()),
    44  	}
    45  	err = k.satisfyWhiteListAdmittance(ctx, pool3)
    46  	require.NotNil(t, err)
    47  
    48  	SetSwapTokenPair(ctx, k.Keeper, baseSymbol, quoteSymbol)
    49  
    50  	err = k.satisfyWhiteListAdmittance(ctx, pool1)
    51  	require.Nil(t, err)
    52  
    53  	err = k.satisfyWhiteListAdmittance(ctx, pool2)
    54  	require.Nil(t, err)
    55  
    56  	err = k.satisfyWhiteListAdmittance(ctx, pool3)
    57  	require.NotNil(t, err)
    58  
    59  	SetSwapTokenPair(ctx, k.Keeper, base1Symbol, quoteSymbol)
    60  	err = k.satisfyWhiteListAdmittance(ctx, pool3)
    61  	require.Nil(t, err)
    62  }
    63  
    64  func TestReadWriteWhiteList(t *testing.T) {
    65  	ctx, k := GetKeeper(t)
    66  
    67  	poolName := "pool"
    68  	require.False(t, k.isPoolNameExistedInWhiteList(ctx, poolName))
    69  	k.SetWhitelist(ctx, poolName)
    70  	require.True(t, k.isPoolNameExistedInWhiteList(ctx, poolName))
    71  	k.DeleteWhiteList(ctx, poolName)
    72  	require.False(t, k.isPoolNameExistedInWhiteList(ctx, poolName))
    73  }