Add cold import script

2018-05-04 10:54:00 -05:00
parent 462f94d84a
commit 5f6bf32ec1
411 changed files with 81507 additions and 1566 deletions
@@ -0,0 +1,51 @@
+package fakes
+
+import (
+	. "github.com/onsi/gomega"
+
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+)
+
+type MockBlockRepository struct {
+	createOrUpdateBlockCalled      bool
+	createOrUpdateBlockPassedBlock core.Block
+	createOrUpdateBlockReturnInt   int64
+	createOrUpdateBlockReturnErr   error
+}
+
+func NewMockBlockRepository() *MockBlockRepository {
+	return &MockBlockRepository{
+		createOrUpdateBlockCalled:      false,
+		createOrUpdateBlockPassedBlock: core.Block{},
+		createOrUpdateBlockReturnInt:   0,
+		createOrUpdateBlockReturnErr:   nil,
+	}
+}
+
+func (mbr *MockBlockRepository) SetCreateOrUpdateBlockReturnVals(i int64, err error) {
+	mbr.createOrUpdateBlockReturnInt = i
+	mbr.createOrUpdateBlockReturnErr = err
+}
+
+func (mbr *MockBlockRepository) CreateOrUpdateBlock(block core.Block) (int64, error) {
+	mbr.createOrUpdateBlockCalled = true
+	mbr.createOrUpdateBlockPassedBlock = block
+	return mbr.createOrUpdateBlockReturnInt, mbr.createOrUpdateBlockReturnErr
+}
+
+func (mbr *MockBlockRepository) GetBlock(blockNumber int64) (core.Block, error) {
+	panic("implement me")
+}
+
+func (mbr *MockBlockRepository) MissingBlockNumbers(startingBlockNumber int64, endingBlockNumber int64) []int64 {
+	panic("implement me")
+}
+
+func (mbr *MockBlockRepository) SetBlocksStatus(chainHead int64) {
+	panic("implement me")
+}
+
+func (mbr *MockBlockRepository) AssertCreateOrUpdateBlockCalledWith(block core.Block) {
+	Expect(mbr.createOrUpdateBlockCalled).To(BeTrue())
+	Expect(mbr.createOrUpdateBlockPassedBlock).To(Equal(block))
+}
@@ -0,0 +1,86 @@
+package fakes
+
+import (
+	. "github.com/onsi/gomega"
+
+	"github.com/ethereum/go-ethereum/core/types"
+)
+
+type MockEthereumDatabase struct {
+	getBlockCalled                 bool
+	getBlockPassedHash             []byte
+	getBlockPassedNumber           int64
+	getBlockReturnBlock            *types.Block
+	getBlockHashCalled             bool
+	getBlockHashPassedNumber       int64
+	getBlockHashReturnHash         []byte
+	getBlockReceiptsCalled         bool
+	getBlockReceiptsPassedHash     []byte
+	getBlockReceiptsPassedNumber   int64
+	getBlockReceiptsReturnReceipts types.Receipts
+}
+
+func NewMockEthereumDatabase() *MockEthereumDatabase {
+	return &MockEthereumDatabase{
+		getBlockCalled:                 false,
+		getBlockPassedHash:             nil,
+		getBlockPassedNumber:           0,
+		getBlockReturnBlock:            nil,
+		getBlockHashCalled:             false,
+		getBlockHashPassedNumber:       0,
+		getBlockHashReturnHash:         nil,
+		getBlockReceiptsCalled:         false,
+		getBlockReceiptsPassedHash:     nil,
+		getBlockReceiptsPassedNumber:   0,
+		getBlockReceiptsReturnReceipts: nil,
+	}
+}
+
+func (med *MockEthereumDatabase) SetReturnBlock(block *types.Block) {
+	med.getBlockReturnBlock = block
+}
+
+func (med *MockEthereumDatabase) SetReturnHash(hash []byte) {
+	med.getBlockHashReturnHash = hash
+}
+
+func (med *MockEthereumDatabase) SetReturnReceipts(receipts types.Receipts) {
+	med.getBlockReceiptsReturnReceipts = receipts
+}
+
+func (med *MockEthereumDatabase) GetBlock(hash []byte, blockNumber int64) *types.Block {
+	med.getBlockCalled = true
+	med.getBlockPassedHash = hash
+	med.getBlockPassedNumber = blockNumber
+	return med.getBlockReturnBlock
+}
+
+func (med *MockEthereumDatabase) GetBlockHash(blockNumber int64) []byte {
+	med.getBlockHashCalled = true
+	med.getBlockHashPassedNumber = blockNumber
+	return med.getBlockHashReturnHash
+}
+
+func (med *MockEthereumDatabase) GetBlockReceipts(blockHash []byte, blockNumber int64) types.Receipts {
+	med.getBlockReceiptsCalled = true
+	med.getBlockReceiptsPassedHash = blockHash
+	med.getBlockReceiptsPassedNumber = blockNumber
+	return med.getBlockReceiptsReturnReceipts
+}
+
+func (med *MockEthereumDatabase) AssertGetBlockCalledWith(hash []byte, blockNumber int64) {
+	Expect(med.getBlockCalled).To(BeTrue())
+	Expect(med.getBlockPassedHash).To(Equal(hash))
+	Expect(med.getBlockPassedNumber).To(Equal(blockNumber))
+}
+
+func (med *MockEthereumDatabase) AssertGetBlockHashCalledWith(blockNumber int64) {
+	Expect(med.getBlockHashCalled).To(BeTrue())
+	Expect(med.getBlockHashPassedNumber).To(Equal(blockNumber))
+}
+
+func (med *MockEthereumDatabase) AssertGetBlockReceiptsCalledWith(blockHash []byte, blockNumber int64) {
+	Expect(med.getBlockReceiptsCalled).To(BeTrue())
+	Expect(med.getBlockReceiptsPassedHash).To(Equal(blockHash))
+	Expect(med.getBlockReceiptsPassedNumber).To(Equal(blockNumber))
+}
@@ -0,0 +1,88 @@
+package fakes
+
+import (
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+	. "github.com/onsi/gomega"
+)
+
+type MockLevelDatabaseReader struct {
+	getBlockCalled               bool
+	getBlockReceiptsCalled       bool
+	getCanonicalHashCalled       bool
+	passedHash                   common.Hash
+	getCanonicalHashPassedNumber uint64
+	getBlockPassedHash           common.Hash
+	getBlockPassedNumber         uint64
+	getBlockReceiptsPassedHash   common.Hash
+	getBlockReceiptsPassedNumber uint64
+	returnBlock                  *types.Block
+	returnHash                   common.Hash
+	returnReceipts               types.Receipts
+}
+
+func NewMockLevelDatabaseReader() *MockLevelDatabaseReader {
+	return &MockLevelDatabaseReader{
+		getBlockCalled:               false,
+		getBlockReceiptsCalled:       false,
+		getCanonicalHashCalled:       false,
+		passedHash:                   common.Hash{},
+		getCanonicalHashPassedNumber: 0,
+		getBlockPassedHash:           common.Hash{},
+		getBlockPassedNumber:         0,
+		getBlockReceiptsPassedHash:   common.Hash{},
+		getBlockReceiptsPassedNumber: 0,
+		returnBlock:                  nil,
+		returnHash:                   common.Hash{},
+		returnReceipts:               nil,
+	}
+}
+
+func (mldr *MockLevelDatabaseReader) SetReturnBlock(block *types.Block) {
+	mldr.returnBlock = block
+}
+
+func (mldr *MockLevelDatabaseReader) SetReturnHash(hash common.Hash) {
+	mldr.returnHash = hash
+}
+
+func (mldr *MockLevelDatabaseReader) SetReturnReceipts(receipts types.Receipts) {
+	mldr.returnReceipts = receipts
+}
+
+func (mldr *MockLevelDatabaseReader) GetCanonicalHash(number uint64) common.Hash {
+	mldr.getCanonicalHashCalled = true
+	mldr.getCanonicalHashPassedNumber = number
+	return mldr.returnHash
+}
+
+func (mldr *MockLevelDatabaseReader) GetBlock(hash common.Hash, number uint64) *types.Block {
+	mldr.getBlockCalled = true
+	mldr.getBlockPassedHash = hash
+	mldr.getBlockPassedNumber = number
+	return mldr.returnBlock
+}
+
+func (mldr *MockLevelDatabaseReader) GetBlockReceipts(hash common.Hash, number uint64) types.Receipts {
+	mldr.getBlockReceiptsCalled = true
+	mldr.getBlockReceiptsPassedHash = hash
+	mldr.getBlockReceiptsPassedNumber = number
+	return mldr.returnReceipts
+}
+
+func (mldr *MockLevelDatabaseReader) AssertGetCanonicalHashCalledWith(number uint64) {
+	Expect(mldr.getCanonicalHashCalled).To(BeTrue())
+	Expect(mldr.getCanonicalHashPassedNumber).To(Equal(number))
+}
+
+func (mldr *MockLevelDatabaseReader) AssertGetBlockCalledWith(hash common.Hash, number uint64) {
+	Expect(mldr.getBlockCalled).To(BeTrue())
+	Expect(mldr.getBlockPassedHash).To(Equal(hash))
+	Expect(mldr.getBlockPassedNumber).To(Equal(number))
+}
+
+func (mldr *MockLevelDatabaseReader) AssertGetBlockReceiptsCalledWith(hash common.Hash, number uint64) {
+	Expect(mldr.getBlockReceiptsCalled).To(BeTrue())
+	Expect(mldr.getBlockReceiptsPassedHash).To(Equal(hash))
+	Expect(mldr.getBlockReceiptsPassedNumber).To(Equal(number))
+}
@@ -0,0 +1,52 @@
+package fakes
+
+import (
+	. "github.com/onsi/gomega"
+
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+)
+
+type MockReceiptRepository struct {
+	createReceiptsAndLogsCalled         bool
+	createReceiptsAndLogsPassedBlockId  int64
+	createReceiptsAndLogsPassedReceipts []core.Receipt
+	createReceiptsAndLogsReturnErr      error
+}
+
+func NewMockReceiptRepository() *MockReceiptRepository {
+	return &MockReceiptRepository{
+		createReceiptsAndLogsCalled:         false,
+		createReceiptsAndLogsPassedBlockId:  0,
+		createReceiptsAndLogsPassedReceipts: nil,
+		createReceiptsAndLogsReturnErr:      nil,
+	}
+}
+
+func (mrr *MockReceiptRepository) SetCreateReceiptsAndLogsReturnErr(err error) {
+	mrr.createReceiptsAndLogsReturnErr = err
+}
+
+func (mrr *MockReceiptRepository) CreateReceiptsAndLogs(blockId int64, receipts []core.Receipt) error {
+	mrr.createReceiptsAndLogsCalled = true
+	mrr.createReceiptsAndLogsPassedBlockId = blockId
+	mrr.createReceiptsAndLogsPassedReceipts = receipts
+	return mrr.createReceiptsAndLogsReturnErr
+}
+
+func (mrr *MockReceiptRepository) CreateReceipt(blockId int64, receipt core.Receipt) (int64, error) {
+	panic("implement me")
+}
+
+func (mrr *MockReceiptRepository) GetReceipt(txHash string) (core.Receipt, error) {
+	panic("implement me")
+}
+
+func (mrr *MockReceiptRepository) AssertCreateReceiptsAndLogsCalledWith(blockId int64, receipts []core.Receipt) {
+	Expect(mrr.createReceiptsAndLogsCalled).To(BeTrue())
+	Expect(mrr.createReceiptsAndLogsPassedBlockId).To(Equal(blockId))
+	Expect(mrr.createReceiptsAndLogsPassedReceipts).To(Equal(receipts))
+}
+
+func (mrr *MockReceiptRepository) AssertCreateReceiptsAndLogsNotCalled() {
+	Expect(mrr.createReceiptsAndLogsCalled).To(BeFalse())
+}
@@ -0,0 +1,40 @@
+package fakes
+
+import (
+	"github.com/ethereum/go-ethereum/core/types"
+	. "github.com/onsi/gomega"
+
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+)
+
+type MockTransactionConverter struct {
+	convertTransactionsToCoreCalled             bool
+	convertTransactionsToCorePassedBlock        *types.Block
+	convertTransactionsToCoreReturnTransactions []core.Transaction
+	convertTransactionsToCoreReturnError        error
+}
+
+func NewMockTransactionConverter() *MockTransactionConverter {
+	return &MockTransactionConverter{
+		convertTransactionsToCoreCalled:             false,
+		convertTransactionsToCorePassedBlock:        nil,
+		convertTransactionsToCoreReturnTransactions: nil,
+		convertTransactionsToCoreReturnError:        nil,
+	}
+}
+
+func (mtc *MockTransactionConverter) SetConvertTransactionsToCoreReturnVals(transactions []core.Transaction, err error) {
+	mtc.convertTransactionsToCoreReturnTransactions = transactions
+	mtc.convertTransactionsToCoreReturnError = err
+}
+
+func (mtc *MockTransactionConverter) ConvertTransactionsToCore(gethBlock *types.Block) ([]core.Transaction, error) {
+	mtc.convertTransactionsToCoreCalled = true
+	mtc.convertTransactionsToCorePassedBlock = gethBlock
+	return mtc.convertTransactionsToCoreReturnTransactions, mtc.convertTransactionsToCoreReturnError
+}
+
+func (mtc *MockTransactionConverter) AssertConvertTransactionsToCoreCalledWith(gethBlock *types.Block) {
+	Expect(mtc.convertTransactionsToCoreCalled).To(BeTrue())
+	Expect(mtc.convertTransactionsToCorePassedBlock).To(Equal(gethBlock))
+}