a5a07b0ced
* port statediff from 9b7fd9af80/statediff/statediff.go; minor fixes
* integrating state diff extracting, building, and persisting into geth processes
* work towards persisting created statediffs in ipfs; based off github.com/vulcanize/eth-block-extractor
* Add a state diff service
* Remove diff extractor from blockchain
* Update imports
* Move statediff on/off check to geth cmd config
* Update starting state diff service
* Add debugging logs for creating diff
* Add statediff extractor and builder tests and small refactoring
* Start to write statediff to a CSV
* Restructure statediff directory
* Pull CSV publishing methods into their own file
* Reformatting due to go fmt
* Add gomega to vendor dir
* Remove testing focuses
* Update statediff tests to use golang test pkg
instead of ginkgo
- builder_test
- extractor_test
- publisher_test
* Use hexutil.Encode instead of deprecated common.ToHex
* Remove OldValue from DiffBigInt and DiffUint64 fields
* Update builder test
* Remove old storage value from updated accounts
* Remove old values from created/deleted accounts
* Update publisher to account for only storing current account values
* Update service loop and fetching previous block
* Update testing
- remove statediff ginkgo test suite file
- move mocks to their own dir
* Updates per go fmt
* Updates to tests
* Pass statediff mode and path in through cli
* Return filename from publisher
* Remove some duplication in builder
* Remove code field from state diff output
this is the contract byte code, and it can still be obtained by querying
the db by the codeHash
* Consolidate acct diff structs for updated & updated/deleted accts
* Include block number in csv filename
* Clean up error logging
* Cleanup formatting, spelling, etc
* Address PR comments
* Add contract address and storage value to csv
* Refactor accumulating account row in csv publisher
* Add DiffStorage struct
* Add storage key to csv
* Address PR comments
* Fix publisher to include rows for accounts that don't have store updates
* Update builder test after merging in release/1.8
* Update test contract to include storage on contract intialization
- so that we're able to test that storage diffing works for created and
deleted accounts (not just updated accounts).
* Factor out a common trie iterator method in builder
123 lines
2.9 KiB
Go
123 lines
2.9 KiB
Go
package extractor_test
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/ethereum/go-ethereum/core/types"
|
|
b "github.com/ethereum/go-ethereum/statediff/builder"
|
|
e "github.com/ethereum/go-ethereum/statediff/extractor"
|
|
"github.com/ethereum/go-ethereum/statediff/testhelpers/mocks"
|
|
"math/big"
|
|
"math/rand"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
var publisher mocks.Publisher
|
|
var builder mocks.Builder
|
|
var currentBlockNumber *big.Int
|
|
var parentBlock, currentBlock *types.Block
|
|
var expectedStateDiff b.StateDiff
|
|
var extractor e.Extractor
|
|
var err error
|
|
|
|
func TestExtractor(t *testing.T) {
|
|
publisher = mocks.Publisher{}
|
|
builder = mocks.Builder{}
|
|
extractor = e.NewExtractor(&builder, &publisher)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
blockNumber := rand.Int63()
|
|
parentBlockNumber := big.NewInt(blockNumber - int64(1))
|
|
currentBlockNumber = big.NewInt(blockNumber)
|
|
parentBlock = types.NewBlock(&types.Header{Number: parentBlockNumber}, nil, nil, nil)
|
|
currentBlock = types.NewBlock(&types.Header{Number: currentBlockNumber}, nil, nil, nil)
|
|
|
|
expectedStateDiff = b.StateDiff{
|
|
BlockNumber: blockNumber,
|
|
BlockHash: currentBlock.Hash(),
|
|
CreatedAccounts: nil,
|
|
DeletedAccounts: nil,
|
|
UpdatedAccounts: nil,
|
|
}
|
|
|
|
testBuildStateDiffStruct(t)
|
|
testBuildStateDiffErrorHandling(t)
|
|
testPublishingStateDiff(t)
|
|
testPublisherErrorHandling(t)
|
|
}
|
|
|
|
func testBuildStateDiffStruct(t *testing.T) {
|
|
builder.SetStateDiffToBuild(&expectedStateDiff)
|
|
|
|
_, err = extractor.ExtractStateDiff(*parentBlock, *currentBlock)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !equals(builder.OldStateRoot, parentBlock.Root()) {
|
|
t.Error()
|
|
}
|
|
if !equals(builder.NewStateRoot, currentBlock.Root()) {
|
|
t.Error()
|
|
}
|
|
if !equals(builder.BlockNumber, currentBlockNumber.Int64()) {
|
|
t.Error()
|
|
}
|
|
if !equals(builder.BlockHash, currentBlock.Hash()) {
|
|
t.Error()
|
|
}
|
|
}
|
|
|
|
func testBuildStateDiffErrorHandling(t *testing.T) {
|
|
builder.SetBuilderError(mocks.Error)
|
|
|
|
_, err = extractor.ExtractStateDiff(*parentBlock, *currentBlock)
|
|
if err == nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !equals(err, mocks.Error) {
|
|
t.Error()
|
|
}
|
|
builder.SetBuilderError(nil)
|
|
}
|
|
|
|
func testPublishingStateDiff(t *testing.T) {
|
|
builder.SetStateDiffToBuild(&expectedStateDiff)
|
|
|
|
_, err = extractor.ExtractStateDiff(*parentBlock, *currentBlock)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if !equals(publisher.StateDiff, &expectedStateDiff) {
|
|
t.Error()
|
|
}
|
|
}
|
|
|
|
func testPublisherErrorHandling(t *testing.T) {
|
|
publisher.SetPublisherError(mocks.Error)
|
|
|
|
_, err = extractor.ExtractStateDiff(*parentBlock, *currentBlock)
|
|
if err == nil {
|
|
t.Error("Expected an error, but it didn't occur.")
|
|
}
|
|
if !equals(err, mocks.Error) {
|
|
t.Error()
|
|
}
|
|
|
|
publisher.SetPublisherError(nil)
|
|
}
|
|
|
|
func equals(actual, expected interface{}) (success bool) {
|
|
if actualByteSlice, ok := actual.([]byte); ok {
|
|
if expectedByteSlice, ok := expected.([]byte); ok {
|
|
return bytes.Equal(actualByteSlice, expectedByteSlice)
|
|
}
|
|
}
|
|
|
|
return reflect.DeepEqual(actual, expected)
|
|
}
|