0g-chain/x/issuance/abci_test.go
2024-08-03 15:04:48 +08:00

115 lines
3.4 KiB
Go

package issuance_test
import (
"testing"
"time"
"github.com/stretchr/testify/suite"
sdkmath "cosmossdk.io/math"
sdk "github.com/cosmos/cosmos-sdk/types"
tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
tmtime "github.com/cometbft/cometbft/types/time"
"github.com/0glabs/0g-chain/app"
"github.com/0glabs/0g-chain/x/issuance"
"github.com/0glabs/0g-chain/x/issuance/keeper"
"github.com/0glabs/0g-chain/x/issuance/types"
)
// Test suite used for all keeper tests
type ABCITestSuite struct {
suite.Suite
keeper keeper.Keeper
app app.TestApp
ctx sdk.Context
addrs []sdk.AccAddress
modAccount sdk.AccAddress
blockTime time.Time
}
// The default state used by each test
func (suite *ABCITestSuite) SetupTest() {
tApp := app.NewTestApp()
blockTime := tmtime.Now()
ctx := tApp.NewContext(true, tmproto.Header{Height: 1, Time: blockTime})
tApp.InitializeFromGenesisStates()
_, addrs := app.GeneratePrivKeyAddressPairs(5)
keeper := tApp.GetIssuanceKeeper()
modAccount, err := sdk.AccAddressFromBech32("0g1ffv7nhd3z6sych2qpqkk03ec6hzkmufyhp5hf8")
suite.Require().NoError(err)
suite.app = tApp
suite.ctx = ctx
suite.keeper = keeper
suite.addrs = addrs
suite.modAccount = modAccount
suite.blockTime = blockTime
}
func (suite *ABCITestSuite) TestRateLimitingTimePassage() {
type args struct {
assets []types.Asset
supplies []types.AssetSupply
blockTimes []time.Duration
expectedSupply types.AssetSupply
}
testCases := []struct {
name string
args args
}{
{
"time passage same period",
args{
assets: []types.Asset{
types.NewAsset(suite.addrs[0].String(), "usdtoken", []string{suite.addrs[1].String()}, false, true, types.NewRateLimit(true, sdkmath.NewInt(10000000000), time.Hour*24)),
},
supplies: []types.AssetSupply{
types.NewAssetSupply(sdk.NewCoin("usdtoken", sdk.ZeroInt()), time.Hour),
},
blockTimes: []time.Duration{time.Hour},
expectedSupply: types.NewAssetSupply(sdk.NewCoin("usdtoken", sdk.ZeroInt()), time.Hour*2),
},
},
{
"time passage new period",
args{
assets: []types.Asset{
types.NewAsset(suite.addrs[0].String(), "usdtoken", []string{suite.addrs[1].String()}, false, true, types.NewRateLimit(true, sdkmath.NewInt(10000000000), time.Hour*24)),
},
supplies: []types.AssetSupply{
types.NewAssetSupply(sdk.NewCoin("usdtoken", sdk.ZeroInt()), time.Hour),
},
blockTimes: []time.Duration{time.Hour * 12, time.Hour * 12},
expectedSupply: types.NewAssetSupply(sdk.NewCoin("usdtoken", sdk.ZeroInt()), time.Duration(0)),
},
},
}
for _, tc := range testCases {
suite.Run(tc.name, func() {
suite.SetupTest()
params := types.NewParams(tc.args.assets)
suite.keeper.SetParams(suite.ctx, params)
for _, supply := range tc.args.supplies {
suite.keeper.SetAssetSupply(suite.ctx, supply, supply.GetDenom())
}
suite.keeper.SetPreviousBlockTime(suite.ctx, suite.blockTime)
for _, bt := range tc.args.blockTimes {
nextBlockTime := suite.ctx.BlockTime().Add(bt)
suite.ctx = suite.ctx.WithBlockTime(nextBlockTime)
suite.Require().NotPanics(func() {
issuance.BeginBlocker(suite.ctx, suite.keeper)
})
}
actualSupply, found := suite.keeper.GetAssetSupply(suite.ctx, tc.args.expectedSupply.GetDenom())
suite.Require().True(found)
suite.Require().Equal(tc.args.expectedSupply, actualSupply)
})
}
}
func TestABCITestSuite(t *testing.T) {
suite.Run(t, new(ABCITestSuite))
}