48 lines
1.2 KiB
Go
48 lines
1.2 KiB
Go
package state
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/tendermint/tendermint/crypto/ed25519"
|
|
cmn "github.com/tendermint/tendermint/libs/common"
|
|
dbm "github.com/tendermint/tendermint/libs/db"
|
|
"github.com/tendermint/tendermint/types"
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
|
)
|
|
|
|
func TestTxFilter(t *testing.T) {
|
|
genDoc := randomGenesisDoc()
|
|
genDoc.ConsensusParams.BlockSize.MaxBytes = 3000
|
|
|
|
testCases := []struct {
|
|
tx types.Tx
|
|
isTxValid bool
|
|
}{
|
|
{types.Tx(cmn.RandBytes(250)), true},
|
|
{types.Tx(cmn.RandBytes(3001)), false},
|
|
}
|
|
|
|
for i, tc := range testCases {
|
|
stateDB := dbm.NewDB("state", "memdb", os.TempDir())
|
|
state, err := LoadStateFromDBOrGenesisDoc(stateDB, genDoc)
|
|
require.NoError(t, err)
|
|
|
|
f := TxFilter(state)
|
|
assert.Equal(t, tc.isTxValid, f(tc.tx), "#%v", i)
|
|
}
|
|
}
|
|
|
|
func randomGenesisDoc() *types.GenesisDoc {
|
|
pubkey := ed25519.GenPrivKey().PubKey()
|
|
return &types.GenesisDoc{
|
|
GenesisTime: tmtime.Now(),
|
|
ChainID: "abc",
|
|
Validators: []types.GenesisValidator{{pubkey.Address(), pubkey, 10, "myval"}},
|
|
ConsensusParams: types.DefaultConsensusParams(),
|
|
}
|
|
}
|