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 getBlockNumberCalled bool getBlockNumberPassedHash common.Hash getBlockPassedHash common.Hash getBlockPassedNumber uint64 getBlockReceiptsCalled bool getBlockReceiptsPassedHash common.Hash getBlockReceiptsPassedNumber uint64 getCanonicalHashCalled bool getCanonicalHashPassedNumber uint64 getCanonicalHashReturnHash common.Hash getHeadBlockHashCalled bool getHeadBlockHashReturnHash common.Hash passedHash common.Hash returnBlock *types.Block returnBlockNumber uint64 returnReceipts types.Receipts } func NewMockLevelDatabaseReader() *MockLevelDatabaseReader { return &MockLevelDatabaseReader{ getBlockCalled: false, getBlockNumberCalled: false, getBlockNumberPassedHash: common.Hash{}, getBlockPassedHash: common.Hash{}, getBlockPassedNumber: 0, getBlockReceiptsCalled: false, getBlockReceiptsPassedHash: common.Hash{}, getBlockReceiptsPassedNumber: 0, getCanonicalHashCalled: false, getCanonicalHashPassedNumber: 0, getCanonicalHashReturnHash: common.Hash{}, getHeadBlockHashCalled: false, getHeadBlockHashReturnHash: common.Hash{}, passedHash: common.Hash{}, returnBlock: nil, returnBlockNumber: 0, returnReceipts: nil, } } func (mldr *MockLevelDatabaseReader) SetReturnBlock(block *types.Block) { mldr.returnBlock = block } func (mldr *MockLevelDatabaseReader) SetReturnBlockNumber(n uint64) { mldr.returnBlockNumber = n } func (mldr *MockLevelDatabaseReader) SetGetCanonicalHashReturnHash(hash common.Hash) { mldr.getCanonicalHashReturnHash = hash } func (mldr *MockLevelDatabaseReader) SetHeadBlockHashReturnHash(hash common.Hash) { mldr.getHeadBlockHashReturnHash = hash } func (mldr *MockLevelDatabaseReader) SetReturnReceipts(receipts types.Receipts) { mldr.returnReceipts = receipts } 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) GetBlockNumber(hash common.Hash) *uint64 { mldr.getBlockNumberCalled = true mldr.getBlockNumberPassedHash = hash return &mldr.returnBlockNumber } func (mldr *MockLevelDatabaseReader) GetCanonicalHash(number uint64) common.Hash { mldr.getCanonicalHashCalled = true mldr.getCanonicalHashPassedNumber = number return mldr.getCanonicalHashReturnHash } func (mldr *MockLevelDatabaseReader) GetHeadBlockHash() common.Hash { mldr.getHeadBlockHashCalled = true return mldr.getHeadBlockHashReturnHash } 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) AssertGetBlockNumberCalledWith(hash common.Hash) { Expect(mldr.getBlockNumberCalled).To(BeTrue()) Expect(mldr.getBlockNumberPassedHash).To(Equal(hash)) } 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)) } func (mldr *MockLevelDatabaseReader) AssertGetCanonicalHashCalledWith(number uint64) { Expect(mldr.getCanonicalHashCalled).To(BeTrue()) Expect(mldr.getCanonicalHashPassedNumber).To(Equal(number)) } func (mldr *MockLevelDatabaseReader) AssertGetHeadBlockHashCalled() { Expect(mldr.getHeadBlockHashCalled).To(BeTrue()) }