mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-25 15:55:18 +00:00
349 lines
11 KiB
Go
349 lines
11 KiB
Go
package bep3_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
"github.com/kava-labs/kava/x/bep3/keeper"
|
|
"github.com/kava-labs/kava/x/bep3/types"
|
|
)
|
|
|
|
type GenesisTestSuite struct {
|
|
suite.Suite
|
|
|
|
app app.TestApp
|
|
ctx sdk.Context
|
|
keeper keeper.Keeper
|
|
addrs []sdk.AccAddress
|
|
}
|
|
|
|
func (suite *GenesisTestSuite) SetupTest() {
|
|
config := sdk.GetConfig()
|
|
app.SetBech32AddressPrefixes(config)
|
|
|
|
tApp := app.NewTestApp()
|
|
suite.ctx = tApp.NewContext(true, tmproto.Header{Height: 1, Time: tmtime.Now()})
|
|
suite.keeper = tApp.GetBep3Keeper()
|
|
suite.app = tApp
|
|
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(3)
|
|
suite.addrs = addrs
|
|
}
|
|
|
|
func (suite *GenesisTestSuite) TestModulePermissionsCheck() {
|
|
cdc := suite.app.AppCodec()
|
|
|
|
testCases := []struct {
|
|
name string
|
|
permissions []string
|
|
expectedPanic string
|
|
}{
|
|
{"no permissions", []string{}, "bep3 module account does not have burn permissions"},
|
|
{"mint permissions", []string{authtypes.Minter}, "bep3 module account does not have burn permissions"},
|
|
{"burn permissions", []string{authtypes.Burner}, "bep3 module account does not have mint permissions"},
|
|
{"burn and mint permissions", []string{authtypes.Burner, authtypes.Minter}, ""},
|
|
{"mint and burn permissions", []string{authtypes.Minter, authtypes.Burner}, ""},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
suite.Run(tc.name, func() {
|
|
suite.SetupTest()
|
|
authGenesis := authtypes.NewGenesisState(
|
|
authtypes.DefaultParams(),
|
|
authtypes.GenesisAccounts{authtypes.NewEmptyModuleAccount(types.ModuleName, tc.permissions...)},
|
|
)
|
|
bep3Genesis := types.DefaultGenesisState()
|
|
genState := app.GenesisState{
|
|
authtypes.ModuleName: cdc.MustMarshalJSON(authGenesis),
|
|
types.ModuleName: cdc.MustMarshalJSON(&bep3Genesis),
|
|
}
|
|
|
|
initApp := func() { suite.app.InitializeFromGenesisStates(genState) }
|
|
|
|
if tc.expectedPanic == "" {
|
|
suite.NotPanics(initApp)
|
|
} else {
|
|
suite.PanicsWithValue(tc.expectedPanic, initApp)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func (suite *GenesisTestSuite) TestGenesisState() {
|
|
type GenState func() app.GenesisState
|
|
|
|
cdc := suite.app.AppCodec()
|
|
|
|
testCases := []struct {
|
|
name string
|
|
genState GenState
|
|
expectPass bool
|
|
}{
|
|
{
|
|
name: "default",
|
|
genState: func() app.GenesisState {
|
|
return NewBep3GenStateMulti(cdc, suite.addrs[0])
|
|
},
|
|
expectPass: true,
|
|
},
|
|
{
|
|
name: "import atomic swaps and asset supplies",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
var swaps types.AtomicSwaps
|
|
var supplies types.AssetSupplies
|
|
for i := 0; i < 2; i++ {
|
|
swap, supply := loadSwapAndSupply(addrs[i], i)
|
|
swaps = append(swaps, swap)
|
|
supplies = append(supplies, supply)
|
|
}
|
|
gs.AtomicSwaps = swaps
|
|
gs.Supplies = supplies
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: true,
|
|
},
|
|
{
|
|
name: "0 deputy fees",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Params.AssetParams[0].FixedFee = sdk.ZeroInt()
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: true,
|
|
},
|
|
{
|
|
name: "incoming supply doesn't match amount in incoming atomic swaps",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(1)
|
|
swap, _ := loadSwapAndSupply(addrs[0], 2)
|
|
gs.AtomicSwaps = types.AtomicSwaps{swap}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "current supply above limit",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
assetParam, _ := suite.keeper.GetAsset(suite.ctx, "bnb")
|
|
gs.Supplies = types.AssetSupplies{
|
|
{
|
|
IncomingSupply: c("bnb", 0),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", assetParam.SupplyLimit.Limit.Add(i(1)).Int64()),
|
|
},
|
|
}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "incoming supply above limit",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
// Set up overlimit amount
|
|
assetParam, _ := suite.keeper.GetAsset(suite.ctx, "bnb")
|
|
overLimitAmount := assetParam.SupplyLimit.Limit.Add(i(1))
|
|
|
|
// Set up an atomic swap with amount equal to the currently asset supply
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
timestamp := ts(0)
|
|
randomNumber, _ := types.GenerateSecureRandomNumber()
|
|
randomNumberHash := types.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := types.NewAtomicSwap(cs(c("bnb", overLimitAmount.Int64())), randomNumberHash,
|
|
types.DefaultMinBlockLock, timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, types.SWAP_STATUS_OPEN, true, types.SWAP_DIRECTION_INCOMING)
|
|
gs.AtomicSwaps = types.AtomicSwaps{swap}
|
|
|
|
// Set up asset supply with overlimit current supply
|
|
gs.Supplies = types.AssetSupplies{
|
|
{
|
|
IncomingSupply: c("bnb", assetParam.SupplyLimit.Limit.Add(i(1)).Int64()),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", 0),
|
|
},
|
|
}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "incoming supply + current supply above limit",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
// Set up overlimit amount
|
|
assetParam, _ := suite.keeper.GetAsset(suite.ctx, "bnb")
|
|
halfLimit := assetParam.SupplyLimit.Limit.Int64() / 2
|
|
overHalfLimit := halfLimit + 1
|
|
|
|
// Set up an atomic swap with amount equal to the currently asset supply
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
timestamp := ts(0)
|
|
randomNumber, _ := types.GenerateSecureRandomNumber()
|
|
randomNumberHash := types.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := types.NewAtomicSwap(cs(c("bnb", halfLimit)), randomNumberHash,
|
|
uint64(360), timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, types.SWAP_STATUS_OPEN, true, types.SWAP_DIRECTION_INCOMING)
|
|
gs.AtomicSwaps = types.AtomicSwaps{swap}
|
|
|
|
// Set up asset supply with overlimit current supply
|
|
gs.Supplies = types.AssetSupplies{
|
|
{
|
|
IncomingSupply: c("bnb", halfLimit),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", overHalfLimit),
|
|
},
|
|
}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "outgoing supply above limit",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
// Set up overlimit amount
|
|
assetParam, _ := suite.keeper.GetAsset(suite.ctx, "bnb")
|
|
overLimitAmount := assetParam.SupplyLimit.Limit.Add(i(1))
|
|
|
|
// Set up an atomic swap with amount equal to the currently asset supply
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
timestamp := ts(0)
|
|
randomNumber, _ := types.GenerateSecureRandomNumber()
|
|
randomNumberHash := types.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := types.NewAtomicSwap(cs(c("bnb", overLimitAmount.Int64())), randomNumberHash,
|
|
types.DefaultMinBlockLock, timestamp, addrs[1], suite.addrs[0], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, types.SWAP_STATUS_OPEN, true, types.SWAP_DIRECTION_OUTGOING)
|
|
gs.AtomicSwaps = types.AtomicSwaps{swap}
|
|
|
|
// Set up asset supply with overlimit current supply
|
|
gs.Supplies = types.AssetSupplies{
|
|
{
|
|
IncomingSupply: c("bnb", 0),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", assetParam.SupplyLimit.Limit.Add(i(1)).Int64()),
|
|
},
|
|
}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "asset supply denom is not a supported asset",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Supplies = types.AssetSupplies{
|
|
{
|
|
IncomingSupply: c("fake", 0),
|
|
OutgoingSupply: c("fake", 0),
|
|
CurrentSupply: c("fake", 0),
|
|
},
|
|
}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "atomic swap asset type is unsupported",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
timestamp := ts(0)
|
|
randomNumber, _ := types.GenerateSecureRandomNumber()
|
|
randomNumberHash := types.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := types.NewAtomicSwap(cs(c("fake", 500000)), randomNumberHash,
|
|
uint64(360), timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, types.SWAP_STATUS_OPEN, true, types.SWAP_DIRECTION_INCOMING)
|
|
|
|
gs.AtomicSwaps = types.AtomicSwaps{swap}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "atomic swap status is invalid",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
timestamp := ts(0)
|
|
randomNumber, _ := types.GenerateSecureRandomNumber()
|
|
randomNumberHash := types.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := types.NewAtomicSwap(cs(c("bnb", 5000)), randomNumberHash,
|
|
uint64(360), timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, types.SWAP_STATUS_UNSPECIFIED, true, types.SWAP_DIRECTION_INCOMING)
|
|
|
|
gs.AtomicSwaps = types.AtomicSwaps{swap}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "minimum block lock cannot be > maximum block lock",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Params.AssetParams[0].MinBlockLock = 201
|
|
gs.Params.AssetParams[0].MaxBlockLock = 200
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "empty supported asset denom",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Params.AssetParams[0].Denom = ""
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "negative supported asset limit",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Params.AssetParams[0].SupplyLimit.Limit = i(-100)
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "duplicate supported asset denom",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Params.AssetParams[1].Denom = "bnb"
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
suite.Run(tc.name, func() {
|
|
suite.SetupTest()
|
|
if tc.expectPass {
|
|
suite.NotPanics(func() {
|
|
suite.app.InitializeFromGenesisStates(tc.genState())
|
|
}, tc.name)
|
|
} else {
|
|
suite.Panics(func() {
|
|
suite.app.InitializeFromGenesisStates(tc.genState())
|
|
}, tc.name)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGenesisTestSuite(t *testing.T) {
|
|
suite.Run(t, new(GenesisTestSuite))
|
|
}
|