65 lines
1.9 KiB
Go
65 lines
1.9 KiB
Go
|
package mock
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
"github.com/ethereum/go-ethereum/core/types"
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
|
||
|
|
"github.com/golang/mock/gomock"
|
||
|
|
|
||
|
|
mocks "github.com/vulcanize/eth-pg-ipfs-state-snapshot/mocks/snapshot"
|
||
|
|
snapt "github.com/vulcanize/eth-pg-ipfs-state-snapshot/pkg/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
type MockPublisher struct {
|
||
|
|
*mocks.MockPublisher
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMockPublisher(t *testing.T) *MockPublisher {
|
||
|
|
ctl := gomock.NewController(t)
|
||
|
|
return &MockPublisher{mocks.NewMockPublisher(ctl)}
|
||
|
|
}
|
||
|
|
|
||
|
|
func dump(funcname string, xs ...interface{}) {
|
||
|
|
if true {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
fmt.Printf(">> %s", funcname)
|
||
|
|
fmt.Printf(strings.Repeat(" %+v", len(xs))+"\n", xs...)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *MockPublisher) PublishHeader(header *types.Header) (int64, error) {
|
||
|
|
// fmt.Printf("PublishHeader %+v\n", header)
|
||
|
|
dump("PublishHeader", header)
|
||
|
|
return p.MockPublisher.PublishHeader(header)
|
||
|
|
}
|
||
|
|
func (p *MockPublisher) PublishStateNode(node *snapt.Node, headerID int64, tx *sqlx.Tx) (int64, error) {
|
||
|
|
dump("PublishStateNode", node, headerID)
|
||
|
|
return p.MockPublisher.PublishStateNode(node, headerID, tx)
|
||
|
|
|
||
|
|
}
|
||
|
|
func (p *MockPublisher) PublishStorageNode(node *snapt.Node, stateID int64, tx *sqlx.Tx) error {
|
||
|
|
dump("PublishStorageNode", node, stateID)
|
||
|
|
return p.MockPublisher.PublishStorageNode(node, stateID, tx)
|
||
|
|
}
|
||
|
|
func (p *MockPublisher) PublishCode(codeHash common.Hash, codeBytes []byte, tx *sqlx.Tx) error {
|
||
|
|
dump("PublishCode", codeHash, codeBytes)
|
||
|
|
return p.MockPublisher.PublishCode(codeHash, codeBytes, tx)
|
||
|
|
}
|
||
|
|
func (p *MockPublisher) BeginTx() (*sqlx.Tx, error) {
|
||
|
|
dump("BeginTx")
|
||
|
|
return p.MockPublisher.BeginTx()
|
||
|
|
}
|
||
|
|
func (p *MockPublisher) CommitTx(tx *sqlx.Tx) error {
|
||
|
|
dump("CommitTx", tx)
|
||
|
|
return p.MockPublisher.CommitTx(tx)
|
||
|
|
}
|
||
|
|
func (p *MockPublisher) PrepareTxForBatch(tx *sqlx.Tx, batchSize uint) (*sqlx.Tx, error) {
|
||
|
|
dump("PrepareTxForBatch", tx, batchSize)
|
||
|
|
return p.MockPublisher.PrepareTxForBatch(tx, batchSize)
|
||
|
|
}
|