mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-27 16:55:21 +00:00
84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
|
package dasigners_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
|
||
|
|
tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
|
||
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||
|
|
|
||
|
|
"github.com/0glabs/0g-chain/app"
|
||
|
|
"github.com/0glabs/0g-chain/x/dasigners/v1"
|
||
|
|
"github.com/0glabs/0g-chain/x/dasigners/v1/keeper"
|
||
|
|
"github.com/0glabs/0g-chain/x/dasigners/v1/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
type GenesisTestSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
|
||
|
|
app app.TestApp
|
||
|
|
ctx sdk.Context
|
||
|
|
keeper keeper.Keeper
|
||
|
|
addresses []sdk.AccAddress
|
||
|
|
}
|
||
|
|
|
||
|
|
func (suite *GenesisTestSuite) SetupTest() {
|
||
|
|
suite.app = app.NewTestApp()
|
||
|
|
suite.keeper = suite.app.GetDASignersKeeper()
|
||
|
|
suite.ctx = suite.app.NewContext(true, tmproto.Header{})
|
||
|
|
_, suite.addresses = app.GeneratePrivKeyAddressPairs(10)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (suite *GenesisTestSuite) TestInitGenesis() {
|
||
|
|
// Most genesis validation tests are located in the types directory. The 'invalid' test cases are
|
||
|
|
// randomly selected subset of those tests.
|
||
|
|
testCases := []struct {
|
||
|
|
name string
|
||
|
|
genState *types.GenesisState
|
||
|
|
expectPass bool
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "normal",
|
||
|
|
genState: types.DefaultGenesisState(),
|
||
|
|
expectPass: true,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tc := range testCases {
|
||
|
|
suite.Run(tc.name, func() {
|
||
|
|
// Setup (note: suite.SetupTest is not run before every suite.Run)
|
||
|
|
suite.app = app.NewTestApp()
|
||
|
|
suite.keeper = suite.app.GetDASignersKeeper()
|
||
|
|
suite.ctx = suite.app.NewContext(true, tmproto.Header{})
|
||
|
|
|
||
|
|
// Run
|
||
|
|
var exportedGenState *types.GenesisState
|
||
|
|
run := func() {
|
||
|
|
dasigners.InitGenesis(suite.ctx, suite.keeper, *tc.genState)
|
||
|
|
exportedGenState = dasigners.ExportGenesis(suite.ctx, suite.keeper)
|
||
|
|
}
|
||
|
|
if tc.expectPass {
|
||
|
|
suite.Require().NotPanics(run)
|
||
|
|
} else {
|
||
|
|
suite.Require().Panics(run)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Check
|
||
|
|
if tc.expectPass {
|
||
|
|
fmt.Printf("expected: %v\n", tc.genState)
|
||
|
|
expectedJson, err := suite.app.AppCodec().MarshalJSON(tc.genState)
|
||
|
|
suite.Require().NoError(err)
|
||
|
|
fmt.Printf("actual: %v\n", exportedGenState)
|
||
|
|
actualJson, err := suite.app.AppCodec().MarshalJSON(exportedGenState)
|
||
|
|
suite.Require().NoError(err)
|
||
|
|
suite.Equal(expectedJson, actualJson)
|
||
|
|
}
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenesisTestSuite(t *testing.T) {
|
||
|
|
suite.Run(t, new(GenesisTestSuite))
|
||
|
|
}
|