Roy Crihfield
6b379c563b
- Remove unused bits - simplify and normalize test data - Renames for clarity
67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package test_helpers
|
|
|
|
import (
|
|
"bytes"
|
|
"sort"
|
|
"testing"
|
|
|
|
statediff "github.com/cerc-io/plugeth-statediff"
|
|
sdtypes "github.com/cerc-io/plugeth-statediff/types"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
"github.com/ethereum/go-ethereum/crypto"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type TestCase struct {
|
|
Name string
|
|
Args statediff.Args
|
|
Expected *sdtypes.StateObject
|
|
}
|
|
|
|
type CheckedRoots = map[*types.Block][]byte
|
|
|
|
func RunBuilderTests(
|
|
t *testing.T,
|
|
builder statediff.Builder,
|
|
tests []TestCase,
|
|
params statediff.Params,
|
|
roots CheckedRoots,
|
|
) {
|
|
for _, test := range tests {
|
|
diff, err := builder.BuildStateDiffObject(test.Args, params)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
normalize(test.Expected)
|
|
normalize(&diff)
|
|
require.Equal(t, *test.Expected, diff, test.Name)
|
|
}
|
|
// Let's also confirm that our root state nodes form the state root hash in the headers
|
|
for block, node := range roots {
|
|
require.Equal(t, block.Root(), crypto.Keccak256Hash(node),
|
|
"expected root does not match actual root", block.Number())
|
|
}
|
|
}
|
|
|
|
// Sorts contained state nodes, storage nodes, and IPLDs
|
|
func normalize(diff *sdtypes.StateObject) {
|
|
sort.Slice(diff.IPLDs, func(i, j int) bool {
|
|
return diff.IPLDs[i].CID < diff.IPLDs[j].CID
|
|
})
|
|
sort.Slice(diff.Nodes, func(i, j int) bool {
|
|
return bytes.Compare(
|
|
diff.Nodes[i].AccountWrapper.LeafKey,
|
|
diff.Nodes[j].AccountWrapper.LeafKey,
|
|
) < 0
|
|
})
|
|
for _, node := range diff.Nodes {
|
|
sort.Slice(node.StorageDiff, func(i, j int) bool {
|
|
return bytes.Compare(
|
|
node.StorageDiff[i].LeafKey,
|
|
node.StorageDiff[j].LeafKey,
|
|
) < 0
|
|
})
|
|
}
|
|
}
|