mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-26 08:15:19 +00:00
790753f156
* wip: refactor to allow multiple bep3 assets * update tests * sims: validate asset before claiming * move asset supply to params * update tests * fix sims * fix tests * wip: add migration from v0.9 -> v0.10 bep3 * fix build and migration tests * nit: rename file * move asset supply out of params * update committee tests * fix sims * address review comments * address review comments * address review comments
304 lines
9.6 KiB
Go
304 lines
9.6 KiB
Go
package bep3_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
"github.com/kava-labs/kava/x/bep3"
|
|
)
|
|
|
|
type GenesisTestSuite struct {
|
|
suite.Suite
|
|
|
|
app app.TestApp
|
|
ctx sdk.Context
|
|
keeper bep3.Keeper
|
|
addrs []sdk.AccAddress
|
|
}
|
|
|
|
func (suite *GenesisTestSuite) SetupTest() {
|
|
tApp := app.NewTestApp()
|
|
suite.ctx = tApp.NewContext(true, abci.Header{Height: 1, Time: tmtime.Now()})
|
|
suite.keeper = tApp.GetBep3Keeper()
|
|
suite.app = tApp
|
|
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(3)
|
|
suite.addrs = addrs
|
|
}
|
|
|
|
func (suite *GenesisTestSuite) TestGenesisState() {
|
|
|
|
type GenState func() app.GenesisState
|
|
|
|
testCases := []struct {
|
|
name string
|
|
genState GenState
|
|
expectPass bool
|
|
}{
|
|
{
|
|
name: "default",
|
|
genState: func() app.GenesisState {
|
|
return NewBep3GenStateMulti(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 bep3.AtomicSwaps
|
|
var supplies bep3.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{"bep3": bep3.ModuleCdc.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{"bep3": bep3.ModuleCdc.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 = bep3.AtomicSwaps{swap}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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 = bep3.AssetSupplies{
|
|
bep3.AssetSupply{
|
|
IncomingSupply: c("bnb", 0),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", assetParam.SupplyLimit.Add(i(1)).Int64()),
|
|
},
|
|
}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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.Add(i(1))
|
|
|
|
// Set up an atomic swap with amount equal to the currently asset supply
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
timestamp := ts(0)
|
|
randomNumber, _ := bep3.GenerateSecureRandomNumber()
|
|
randomNumberHash := bep3.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := bep3.NewAtomicSwap(cs(c("bnb", overLimitAmount.Int64())), randomNumberHash,
|
|
bep3.DefaultMinBlockLock, timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, bep3.Open, true, bep3.Incoming)
|
|
gs.AtomicSwaps = bep3.AtomicSwaps{swap}
|
|
|
|
// Set up asset supply with overlimit current supply
|
|
gs.Supplies = bep3.AssetSupplies{
|
|
bep3.AssetSupply{
|
|
IncomingSupply: c("bnb", assetParam.SupplyLimit.Add(i(1)).Int64()),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", 0),
|
|
},
|
|
}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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.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, _ := bep3.GenerateSecureRandomNumber()
|
|
randomNumberHash := bep3.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := bep3.NewAtomicSwap(cs(c("bnb", halfLimit)), randomNumberHash,
|
|
uint64(360), timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, bep3.Open, true, bep3.Incoming)
|
|
gs.AtomicSwaps = bep3.AtomicSwaps{swap}
|
|
|
|
// Set up asset supply with overlimit current supply
|
|
gs.Supplies = bep3.AssetSupplies{
|
|
bep3.AssetSupply{
|
|
IncomingSupply: c("bnb", halfLimit),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", overHalfLimit),
|
|
},
|
|
}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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.Add(i(1))
|
|
|
|
// Set up an atomic swap with amount equal to the currently asset supply
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
timestamp := ts(0)
|
|
randomNumber, _ := bep3.GenerateSecureRandomNumber()
|
|
randomNumberHash := bep3.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := bep3.NewAtomicSwap(cs(c("bnb", overLimitAmount.Int64())), randomNumberHash,
|
|
bep3.DefaultMinBlockLock, timestamp, addrs[1], suite.addrs[0], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, bep3.Open, true, bep3.Outgoing)
|
|
gs.AtomicSwaps = bep3.AtomicSwaps{swap}
|
|
|
|
// Set up asset supply with overlimit current supply
|
|
gs.Supplies = bep3.AssetSupplies{
|
|
bep3.AssetSupply{
|
|
IncomingSupply: c("bnb", 0),
|
|
OutgoingSupply: c("bnb", 0),
|
|
CurrentSupply: c("bnb", assetParam.SupplyLimit.Add(i(1)).Int64()),
|
|
},
|
|
}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.MustMarshalJSON(gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "asset supply denom is not a supported asset",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Supplies = bep3.AssetSupplies{
|
|
bep3.AssetSupply{
|
|
IncomingSupply: c("fake", 0),
|
|
OutgoingSupply: c("fake", 0),
|
|
CurrentSupply: c("fake", 0),
|
|
},
|
|
}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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, _ := bep3.GenerateSecureRandomNumber()
|
|
randomNumberHash := bep3.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := bep3.NewAtomicSwap(cs(c("fake", 500000)), randomNumberHash,
|
|
uint64(360), timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, bep3.Open, true, bep3.Incoming)
|
|
|
|
gs.AtomicSwaps = bep3.AtomicSwaps{swap}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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, _ := bep3.GenerateSecureRandomNumber()
|
|
randomNumberHash := bep3.CalculateRandomHash(randomNumber[:], timestamp)
|
|
swap := bep3.NewAtomicSwap(cs(c("bnb", 5000)), randomNumberHash,
|
|
uint64(360), timestamp, suite.addrs[0], addrs[1], TestSenderOtherChain,
|
|
TestRecipientOtherChain, 0, bep3.NULL, true, bep3.Incoming)
|
|
|
|
gs.AtomicSwaps = bep3.AtomicSwaps{swap}
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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{"bep3": bep3.ModuleCdc.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{"bep3": bep3.ModuleCdc.MustMarshalJSON(gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
{
|
|
name: "negative supported asset limit",
|
|
genState: func() app.GenesisState {
|
|
gs := baseGenState(suite.addrs[0])
|
|
gs.Params.AssetParams[0].SupplyLimit = i(-100)
|
|
return app.GenesisState{"bep3": bep3.ModuleCdc.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{"bep3": bep3.ModuleCdc.MustMarshalJSON(gs)}
|
|
},
|
|
expectPass: false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
suite.Run(tc.name, func() {
|
|
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))
|
|
}
|