mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 10:05:18 +00:00
8001cbbfd7
* fix: check asset supply * update begin blocker * add activation height and tests * fix: verify outgoing swap recipient * validate recipient address for outgoing swaps * update activation time * remove unused files Co-authored-by: rhuairahrighairigh <ruaridh.odonnell@gmail.com>
301 lines
9.4 KiB
Go
301 lines
9.4 KiB
Go
package bep3_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmbytes "github.com/tendermint/tendermint/libs/bytes"
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
"github.com/kava-labs/kava/x/bep3"
|
|
)
|
|
|
|
type ABCITestSuite struct {
|
|
suite.Suite
|
|
keeper bep3.Keeper
|
|
app app.TestApp
|
|
ctx sdk.Context
|
|
addrs []sdk.AccAddress
|
|
swapIDs []tmbytes.HexBytes
|
|
randomNumbers []tmbytes.HexBytes
|
|
}
|
|
|
|
func (suite *ABCITestSuite) SetupTest() {
|
|
tApp := app.NewTestApp()
|
|
ctx := tApp.NewContext(true, abci.Header{Height: 1, Time: tmtime.Now()})
|
|
|
|
// Set up auth GenesisState
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(11)
|
|
coins := []sdk.Coins{}
|
|
for j := 0; j < 11; j++ {
|
|
coins = append(coins, cs(c("bnb", 10000000000), c("ukava", 10000000000)))
|
|
}
|
|
authGS := app.NewAuthGenState(addrs, coins)
|
|
// Initialize test app
|
|
tApp.InitializeFromGenesisStates(authGS, NewBep3GenStateMulti(addrs[0]))
|
|
|
|
suite.ctx = ctx
|
|
suite.app = tApp
|
|
suite.addrs = addrs
|
|
suite.ResetKeeper()
|
|
}
|
|
|
|
func (suite *ABCITestSuite) ResetKeeper() {
|
|
suite.keeper = suite.app.GetBep3Keeper()
|
|
|
|
var swapIDs []tmbytes.HexBytes
|
|
var randomNumbers []tmbytes.HexBytes
|
|
for i := 0; i < 10; i++ {
|
|
// Set up atomic swap variables
|
|
expireHeight := bep3.DefaultMinBlockLock
|
|
amount := cs(c("bnb", int64(10000)))
|
|
timestamp := ts(i)
|
|
randomNumber, _ := bep3.GenerateSecureRandomNumber()
|
|
randomNumberHash := bep3.CalculateRandomHash(randomNumber[:], timestamp)
|
|
|
|
// Create atomic swap and check err to confirm creation
|
|
err := suite.keeper.CreateAtomicSwap(suite.ctx, randomNumberHash, timestamp, expireHeight,
|
|
suite.addrs[0], suite.addrs[i], TestSenderOtherChain, TestRecipientOtherChain,
|
|
amount, true)
|
|
suite.Nil(err)
|
|
|
|
// Store swap's calculated ID and secret random number
|
|
swapID := bep3.CalculateSwapID(randomNumberHash, suite.addrs[0], TestSenderOtherChain)
|
|
swapIDs = append(swapIDs, swapID)
|
|
randomNumbers = append(randomNumbers, randomNumber[:])
|
|
}
|
|
suite.swapIDs = swapIDs
|
|
suite.randomNumbers = randomNumbers
|
|
}
|
|
|
|
func (suite *ABCITestSuite) TestBeginBlocker_UpdateExpiredAtomicSwaps() {
|
|
testCases := []struct {
|
|
name string
|
|
firstCtx sdk.Context
|
|
secondCtx sdk.Context
|
|
expectedStatus bep3.SwapStatus
|
|
expectInStorage bool
|
|
}{
|
|
{
|
|
name: "normal",
|
|
firstCtx: suite.ctx,
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 10),
|
|
expectedStatus: bep3.Open,
|
|
expectInStorage: true,
|
|
},
|
|
{
|
|
name: "after expiration",
|
|
firstCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 400),
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 410),
|
|
expectedStatus: bep3.Expired,
|
|
expectInStorage: true,
|
|
},
|
|
{
|
|
name: "after completion",
|
|
firstCtx: suite.ctx,
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 10),
|
|
expectedStatus: bep3.Completed,
|
|
expectInStorage: true,
|
|
},
|
|
{
|
|
name: "after deletion",
|
|
firstCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 400),
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 400 + int64(bep3.DefaultLongtermStorageDuration)),
|
|
expectedStatus: bep3.NULL,
|
|
expectInStorage: false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
// Reset keeper and run the initial begin blocker
|
|
suite.ResetKeeper()
|
|
suite.Run(tc.name, func() {
|
|
bep3.BeginBlocker(tc.firstCtx, suite.keeper)
|
|
|
|
switch tc.expectedStatus {
|
|
case bep3.Completed:
|
|
for i, swapID := range suite.swapIDs {
|
|
err := suite.keeper.ClaimAtomicSwap(tc.firstCtx, suite.addrs[5], swapID, suite.randomNumbers[i])
|
|
suite.Nil(err)
|
|
}
|
|
case bep3.NULL:
|
|
for _, swapID := range suite.swapIDs {
|
|
err := suite.keeper.RefundAtomicSwap(tc.firstCtx, suite.addrs[5], swapID)
|
|
suite.Nil(err)
|
|
}
|
|
}
|
|
|
|
// Run the second begin blocker
|
|
bep3.BeginBlocker(tc.secondCtx, suite.keeper)
|
|
|
|
// Check each swap's availibility and status
|
|
for _, swapID := range suite.swapIDs {
|
|
storedSwap, found := suite.keeper.GetAtomicSwap(tc.secondCtx, swapID)
|
|
if tc.expectInStorage {
|
|
suite.True(found)
|
|
} else {
|
|
suite.False(found)
|
|
}
|
|
suite.Equal(tc.expectedStatus, storedSwap.Status)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func (suite *ABCITestSuite) TestBeginBlocker_DeleteClosedAtomicSwapsFromLongtermStorage() {
|
|
type Action int
|
|
const (
|
|
NULL Action = 0x00
|
|
Refund Action = 0x01
|
|
Claim Action = 0x02
|
|
)
|
|
|
|
testCases := []struct {
|
|
name string
|
|
firstCtx sdk.Context
|
|
action Action
|
|
secondCtx sdk.Context
|
|
expectInStorage bool
|
|
}{
|
|
{
|
|
name: "no action with long storage duration",
|
|
firstCtx: suite.ctx,
|
|
action: NULL,
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + int64(bep3.DefaultLongtermStorageDuration)),
|
|
expectInStorage: true,
|
|
},
|
|
{
|
|
name: "claim with short storage duration",
|
|
firstCtx: suite.ctx,
|
|
action: Claim,
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 5000),
|
|
expectInStorage: true,
|
|
},
|
|
{
|
|
name: "claim with long storage duration",
|
|
firstCtx: suite.ctx,
|
|
action: Claim,
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + int64(bep3.DefaultLongtermStorageDuration)),
|
|
expectInStorage: false,
|
|
},
|
|
{
|
|
name: "refund with short storage duration",
|
|
firstCtx: suite.ctx,
|
|
action: Refund,
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + 5000),
|
|
expectInStorage: true,
|
|
},
|
|
{
|
|
name: "refund with long storage duration",
|
|
firstCtx: suite.ctx,
|
|
action: Refund,
|
|
secondCtx: suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + int64(bep3.DefaultLongtermStorageDuration)),
|
|
expectInStorage: false,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
// Reset keeper and run the initial begin blocker
|
|
suite.ResetKeeper()
|
|
suite.Run(tc.name, func() {
|
|
bep3.BeginBlocker(tc.firstCtx, suite.keeper)
|
|
|
|
switch tc.action {
|
|
case Claim:
|
|
for i, swapID := range suite.swapIDs {
|
|
err := suite.keeper.ClaimAtomicSwap(tc.firstCtx, suite.addrs[5], swapID, suite.randomNumbers[i])
|
|
suite.Nil(err)
|
|
}
|
|
case Refund:
|
|
for _, swapID := range suite.swapIDs {
|
|
swap, _ := suite.keeper.GetAtomicSwap(tc.firstCtx, swapID)
|
|
refundCtx := suite.ctx.WithBlockHeight(suite.ctx.BlockHeight() + int64(swap.ExpireHeight))
|
|
bep3.BeginBlocker(refundCtx, suite.keeper)
|
|
err := suite.keeper.RefundAtomicSwap(refundCtx, suite.addrs[5], swapID)
|
|
suite.Nil(err)
|
|
// Add expire height to second ctx block height
|
|
tc.secondCtx = tc.secondCtx.WithBlockHeight(tc.secondCtx.BlockHeight() + int64(swap.ExpireHeight))
|
|
}
|
|
}
|
|
|
|
// Run the second begin blocker
|
|
bep3.BeginBlocker(tc.secondCtx, suite.keeper)
|
|
|
|
// Check each swap's availability and status
|
|
for _, swapID := range suite.swapIDs {
|
|
_, found := suite.keeper.GetAtomicSwap(tc.secondCtx, swapID)
|
|
if tc.expectInStorage {
|
|
suite.True(found)
|
|
} else {
|
|
suite.False(found)
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func (suite *ABCITestSuite) TestBeginBlocker_UpdateAssetSupplies() {
|
|
// set new asset limit in the params
|
|
newBnbLimit := c("bnb", 100)
|
|
params := suite.keeper.GetParams(suite.ctx)
|
|
for i := range params.SupportedAssets {
|
|
if params.SupportedAssets[i].Denom != newBnbLimit.Denom {
|
|
continue
|
|
}
|
|
params.SupportedAssets[i].Limit = newBnbLimit.Amount
|
|
}
|
|
suite.keeper.SetParams(suite.ctx, params)
|
|
|
|
// store the old limit for future reference
|
|
supply, found := suite.keeper.GetAssetSupply(suite.ctx, []byte(newBnbLimit.Denom))
|
|
suite.True(found)
|
|
oldBnbLimit := supply.SupplyLimit
|
|
|
|
// run before the upgrade time, check limit was not changed
|
|
bep3.BeginBlocker(suite.ctx.WithBlockTime(bep3.SupplyLimitUpgradeTime.Add(-time.Hour)), suite.keeper)
|
|
|
|
supply, found = suite.keeper.GetAssetSupply(suite.ctx, []byte(newBnbLimit.Denom))
|
|
suite.True(found)
|
|
suite.True(supply.SupplyLimit.IsEqual(oldBnbLimit))
|
|
|
|
// run at precise upgrade time, check limit was not changed
|
|
bep3.BeginBlocker(suite.ctx.WithBlockTime(bep3.SupplyLimitUpgradeTime), suite.keeper)
|
|
|
|
supply, found = suite.keeper.GetAssetSupply(suite.ctx, []byte(newBnbLimit.Denom))
|
|
suite.True(found)
|
|
suite.True(supply.SupplyLimit.IsEqual(oldBnbLimit))
|
|
|
|
// run after upgrade time, check limit was changed
|
|
bep3.BeginBlocker(suite.ctx.WithBlockTime(bep3.SupplyLimitUpgradeTime.Add(time.Nanosecond)), suite.keeper)
|
|
|
|
supply, found = suite.keeper.GetAssetSupply(suite.ctx, []byte(newBnbLimit.Denom))
|
|
suite.True(found)
|
|
suite.True(supply.SupplyLimit.IsEqual(newBnbLimit))
|
|
|
|
// run again with new params, check limit was updated to new param limit
|
|
finalBnbLimit := c("bnb", 5000000000000)
|
|
params = suite.keeper.GetParams(suite.ctx)
|
|
for i := range params.SupportedAssets {
|
|
if params.SupportedAssets[i].Denom != finalBnbLimit.Denom {
|
|
continue
|
|
}
|
|
params.SupportedAssets[i].Limit = finalBnbLimit.Amount
|
|
}
|
|
suite.keeper.SetParams(suite.ctx, params)
|
|
bep3.BeginBlocker(suite.ctx.WithBlockTime(bep3.SupplyLimitUpgradeTime.Add(time.Hour)), suite.keeper)
|
|
|
|
supply, found = suite.keeper.GetAssetSupply(suite.ctx, []byte(newBnbLimit.Denom))
|
|
suite.True(found)
|
|
suite.True(supply.SupplyLimit.IsEqual(finalBnbLimit))
|
|
}
|
|
|
|
func TestABCITestSuite(t *testing.T) {
|
|
suite.Run(t, new(ABCITestSuite))
|
|
}
|