132 lines
3.1 KiB
Go
132 lines
3.1 KiB
Go
package validator_test
|
|
|
|
import (
|
|
"context"
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/cerc-io/plugeth-statediff/indexer/ipld"
|
|
helpers "github.com/cerc-io/plugeth-statediff/test_helpers"
|
|
"github.com/cerc-io/plugeth-statediff/test_helpers/chaingen"
|
|
sdtypes "github.com/cerc-io/plugeth-statediff/types"
|
|
"github.com/ethereum/go-ethereum/core/rawdb"
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
"github.com/jmoiron/sqlx"
|
|
// import server helpers for non-canonical chain data
|
|
server_mocks "github.com/cerc-io/ipld-eth-server/v5/pkg/eth/test_helpers"
|
|
|
|
"github.com/cerc-io/ipld-eth-db-validator/v5/pkg/validator"
|
|
)
|
|
|
|
const (
|
|
chainLength = 10
|
|
startBlock = 1
|
|
)
|
|
|
|
var (
|
|
chainConfig = TestChainConfig
|
|
mockTD = big.NewInt(1337)
|
|
testDB = rawdb.NewMemoryDatabase()
|
|
)
|
|
|
|
func init() {
|
|
// The geth sync logs are noisy, silence them
|
|
log.Root().SetHandler(log.DiscardHandler())
|
|
}
|
|
|
|
func setupStateValidator(t *testing.T) *sqlx.DB {
|
|
// Make the test blockchain and state
|
|
gen := chaingen.DefaultGenContext(chainConfig, testDB)
|
|
blocks, receipts, chain := gen.MakeChain(chainLength)
|
|
|
|
t.Cleanup(func() {
|
|
chain.Stop()
|
|
})
|
|
|
|
indexer, err := helpers.TestStateDiffIndexer(context.Background(), chainConfig, gen.Genesis.Hash())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// Insert some non-canonical data into the database so that we test our ability to discern canonicity
|
|
// Use a func here for defer
|
|
func() {
|
|
tx, err := indexer.PushBlock(server_mocks.MockBlock, server_mocks.MockReceipts,
|
|
server_mocks.MockBlock.Difficulty())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer tx.RollbackOnFailure(err)
|
|
|
|
err = tx.Submit()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// The non-canonical header has a child
|
|
tx, err = indexer.PushBlock(server_mocks.MockChild, server_mocks.MockReceipts,
|
|
server_mocks.MockChild.Difficulty())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer tx.RollbackOnFailure(err)
|
|
|
|
ipld := sdtypes.IPLD{
|
|
CID: ipld.Keccak256ToCid(ipld.RawBinary, server_mocks.CodeHash.Bytes()).String(),
|
|
Content: server_mocks.ContractCode,
|
|
}
|
|
err = indexer.PushIPLD(tx, ipld)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = tx.Submit()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}()
|
|
|
|
if err := helpers.IndexChain(indexer, helpers.IndexChainParams{
|
|
StateCache: chain.StateCache(),
|
|
Blocks: blocks,
|
|
Receipts: receipts,
|
|
TotalDifficulty: mockTD,
|
|
}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
db := helpers.SetupDB()
|
|
|
|
t.Cleanup(func() {
|
|
helpers.TearDownDB(db)
|
|
})
|
|
return db
|
|
}
|
|
|
|
func TestStateValidation(t *testing.T) {
|
|
db := setupStateValidator(t)
|
|
|
|
t.Run("Validator", func(t *testing.T) {
|
|
api, err := validator.EthAPI(context.Background(), db, chainConfig)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
for i := uint64(startBlock); i <= chainLength; i++ {
|
|
blockToBeValidated, err := api.B.BlockByNumber(context.Background(), rpc.BlockNumber(i))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if blockToBeValidated == nil {
|
|
t.Fatal("blockToBeValidated is nil")
|
|
}
|
|
|
|
err = validator.ValidateBlock(blockToBeValidated, api.B, i)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
})
|
|
}
|