130 lines
3.1 KiB
Go
130 lines
3.1 KiB
Go
package evidence_test
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/crypto/ed25519"
|
|
|
|
codecstd "github.com/cosmos/cosmos-sdk/codec/std"
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/evidence"
|
|
"github.com/cosmos/cosmos-sdk/x/evidence/exported"
|
|
"github.com/cosmos/cosmos-sdk/x/evidence/types"
|
|
)
|
|
|
|
type HandlerTestSuite struct {
|
|
suite.Suite
|
|
|
|
handler sdk.Handler
|
|
app *simapp.SimApp
|
|
}
|
|
|
|
func testMsgSubmitEvidence(r *require.Assertions, e exported.Evidence, s sdk.AccAddress) codecstd.MsgSubmitEvidence {
|
|
msg, err := codecstd.NewMsgSubmitEvidence(e, s)
|
|
r.NoError(err)
|
|
return msg
|
|
}
|
|
|
|
func testEquivocationHandler(k interface{}) types.Handler {
|
|
return func(ctx sdk.Context, e exported.Evidence) error {
|
|
if err := e.ValidateBasic(); err != nil {
|
|
return err
|
|
}
|
|
|
|
ee, ok := e.(*types.Equivocation)
|
|
if !ok {
|
|
return fmt.Errorf("unexpected evidence type: %T", e)
|
|
}
|
|
if ee.Height%2 == 0 {
|
|
return fmt.Errorf("unexpected even evidence height: %d", ee.Height)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) SetupTest() {
|
|
checkTx := false
|
|
app := simapp.Setup(checkTx)
|
|
|
|
// recreate keeper in order to use custom testing types
|
|
evidenceKeeper := evidence.NewKeeper(
|
|
codecstd.NewAppCodec(app.Codec()), app.GetKey(evidence.StoreKey),
|
|
app.GetSubspace(evidence.ModuleName), app.StakingKeeper, app.SlashingKeeper,
|
|
)
|
|
router := evidence.NewRouter()
|
|
router = router.AddRoute(types.RouteEquivocation, testEquivocationHandler(*evidenceKeeper))
|
|
evidenceKeeper.SetRouter(router)
|
|
|
|
app.EvidenceKeeper = *evidenceKeeper
|
|
|
|
suite.handler = evidence.NewHandler(*evidenceKeeper)
|
|
suite.app = app
|
|
}
|
|
|
|
func (suite *HandlerTestSuite) TestMsgSubmitEvidence() {
|
|
pk := ed25519.GenPrivKey()
|
|
s := sdk.AccAddress("test")
|
|
|
|
testCases := []struct {
|
|
msg sdk.Msg
|
|
expectErr bool
|
|
}{
|
|
{
|
|
testMsgSubmitEvidence(
|
|
suite.Require(),
|
|
&types.Equivocation{
|
|
Height: 11,
|
|
Time: time.Now().UTC(),
|
|
Power: 100,
|
|
ConsensusAddress: pk.PubKey().Address().Bytes(),
|
|
},
|
|
s,
|
|
),
|
|
false,
|
|
},
|
|
{
|
|
testMsgSubmitEvidence(
|
|
suite.Require(),
|
|
&types.Equivocation{
|
|
Height: 10,
|
|
Time: time.Now().UTC(),
|
|
Power: 100,
|
|
ConsensusAddress: pk.PubKey().Address().Bytes(),
|
|
},
|
|
s,
|
|
),
|
|
true,
|
|
},
|
|
{
|
|
types.NewMsgSubmitEvidenceBase(s),
|
|
true,
|
|
},
|
|
}
|
|
|
|
for i, tc := range testCases {
|
|
ctx := suite.app.BaseApp.NewContext(false, abci.Header{Height: suite.app.LastBlockHeight() + 1})
|
|
|
|
res, err := suite.handler(ctx, tc.msg)
|
|
if tc.expectErr {
|
|
suite.Require().Error(err, "expected error; tc #%d", i)
|
|
} else {
|
|
suite.Require().NoError(err, "unexpected error; tc #%d", i)
|
|
suite.Require().NotNil(res, "expected non-nil result; tc #%d", i)
|
|
|
|
msg := tc.msg.(codecstd.MsgSubmitEvidence)
|
|
suite.Require().Equal(msg.GetEvidence().Hash().Bytes(), res.Data, "invalid hash; tc #%d", i)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHandlerTestSuite(t *testing.T) {
|
|
suite.Run(t, new(HandlerTestSuite))
|
|
}
|