150ca963f2
Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Co-authored-by: Marko <marko@baricevic.me>
103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"time"
|
|
|
|
"cosmossdk.io/core/header"
|
|
"cosmossdk.io/x/epochs/types"
|
|
)
|
|
|
|
func (s *KeeperTestSuite) TestAddEpochInfo() {
|
|
defaultIdentifier := "default_add_epoch_info_id"
|
|
defaultDuration := time.Hour
|
|
startBlockHeight := int64(100)
|
|
startBlockTime := time.Unix(1656907200, 0).UTC()
|
|
tests := map[string]struct {
|
|
addedEpochInfo types.EpochInfo
|
|
expErr bool
|
|
expEpochInfo types.EpochInfo
|
|
}{
|
|
"simple_add": {
|
|
addedEpochInfo: types.EpochInfo{
|
|
Identifier: defaultIdentifier,
|
|
StartTime: time.Time{},
|
|
Duration: defaultDuration,
|
|
CurrentEpoch: 0,
|
|
CurrentEpochStartHeight: 0,
|
|
CurrentEpochStartTime: time.Time{},
|
|
EpochCountingStarted: false,
|
|
},
|
|
expErr: false,
|
|
expEpochInfo: types.EpochInfo{
|
|
Identifier: defaultIdentifier,
|
|
StartTime: startBlockTime,
|
|
Duration: defaultDuration,
|
|
CurrentEpoch: 0,
|
|
CurrentEpochStartHeight: startBlockHeight,
|
|
CurrentEpochStartTime: time.Time{},
|
|
EpochCountingStarted: false,
|
|
},
|
|
},
|
|
"zero_duration": {
|
|
addedEpochInfo: types.EpochInfo{
|
|
Identifier: defaultIdentifier,
|
|
StartTime: time.Time{},
|
|
Duration: time.Duration(0),
|
|
CurrentEpoch: 0,
|
|
CurrentEpochStartHeight: 0,
|
|
CurrentEpochStartTime: time.Time{},
|
|
EpochCountingStarted: false,
|
|
},
|
|
expErr: true,
|
|
},
|
|
}
|
|
for name, test := range tests {
|
|
s.Run(name, func() {
|
|
s.SetupTest()
|
|
s.Ctx = s.Ctx.WithHeaderInfo(header.Info{Height: startBlockHeight, Time: startBlockTime})
|
|
err := s.EpochsKeeper.AddEpochInfo(s.Ctx, test.addedEpochInfo)
|
|
if !test.expErr {
|
|
s.Require().NoError(err)
|
|
actualEpochInfo, err := s.EpochsKeeper.EpochInfo.Get(s.Ctx, test.addedEpochInfo.Identifier)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(test.expEpochInfo, actualEpochInfo)
|
|
} else {
|
|
s.Require().Error(err)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func (s *KeeperTestSuite) TestDuplicateAddEpochInfo() {
|
|
identifier := "duplicate_add_epoch_info"
|
|
epochInfo := types.NewGenesisEpochInfo(identifier, time.Hour*24*30)
|
|
err := s.EpochsKeeper.AddEpochInfo(s.Ctx, epochInfo)
|
|
s.Require().NoError(err)
|
|
err = s.EpochsKeeper.AddEpochInfo(s.Ctx, epochInfo)
|
|
s.Require().Error(err)
|
|
}
|
|
|
|
func (s *KeeperTestSuite) TestEpochLifeCycle() {
|
|
s.SetupTest()
|
|
|
|
epochInfo := types.NewGenesisEpochInfo("monthly", time.Hour*24*30)
|
|
err := s.EpochsKeeper.AddEpochInfo(s.Ctx, epochInfo)
|
|
s.Require().NoError(err)
|
|
epochInfoSaved, err := s.EpochsKeeper.EpochInfo.Get(s.Ctx, "monthly")
|
|
s.Require().NoError(err)
|
|
// setup expected epoch info
|
|
expectedEpochInfo := epochInfo
|
|
expectedEpochInfo.StartTime = s.Ctx.BlockTime()
|
|
expectedEpochInfo.CurrentEpochStartHeight = s.Ctx.BlockHeight()
|
|
s.Require().Equal(expectedEpochInfo, epochInfoSaved)
|
|
|
|
allEpochs, err := s.EpochsKeeper.AllEpochInfos(s.Ctx)
|
|
s.Require().NoError(err)
|
|
s.Require().Len(allEpochs, 5)
|
|
s.Require().Equal(allEpochs[0].Identifier, "day") // alphabetical order
|
|
s.Require().Equal(allEpochs[1].Identifier, "hour")
|
|
s.Require().Equal(allEpochs[2].Identifier, "minute")
|
|
s.Require().Equal(allEpochs[3].Identifier, "monthly")
|
|
s.Require().Equal(allEpochs[4].Identifier, "week")
|
|
}
|