ipld-eth-server/pkg/eth/fakes/mock_block_repository.go

// VulcanizeDB
// Copyright © 2019 Vulcanize

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.

// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

package fakes

import (
	. "github.com/onsi/gomega"

	"github.com/vulcanize/vulcanizedb/pkg/eth/core"
)

type MockBlockRepository struct {
	createOrUpdateBlockCallCount                 int
	createOrUpdateBlockCalled                    bool
	createOrUpdateBlockPassedBlock               core.Block
	createOrUpdateBlockPassedBlockNumbers        []int64
	createOrUpdateBlockReturnErr                 error
	createOrUpdateBlockReturnInt                 int64
	missingBlockNumbersCalled                    bool
	missingBlockNumbersPassedEndingBlockNumber   int64
	missingBlockNumbersPassedNodeID              string
	missingBlockNumbersPassedStartingBlockNumber int64
	missingBlockNumbersReturnArray               []int64
	setBlockStatusCalled                         bool
	setBlockStatusPassedChainHead                int64
}

func NewMockBlockRepository() *MockBlockRepository {
	return &MockBlockRepository{
		createOrUpdateBlockCallCount:                 0,
		createOrUpdateBlockCalled:                    false,
		createOrUpdateBlockPassedBlock:               core.Block{},
		createOrUpdateBlockPassedBlockNumbers:        nil,
		createOrUpdateBlockReturnErr:                 nil,
		createOrUpdateBlockReturnInt:                 0,
		missingBlockNumbersCalled:                    false,
		missingBlockNumbersPassedEndingBlockNumber:   0,
		missingBlockNumbersPassedNodeID:              "",
		missingBlockNumbersPassedStartingBlockNumber: 0,
		missingBlockNumbersReturnArray:               nil,
		setBlockStatusCalled:                         false,
		setBlockStatusPassedChainHead:                0,
	}
}

func (repository *MockBlockRepository) SetCreateOrUpdateBlockReturnVals(i int64, err error) {
	repository.createOrUpdateBlockReturnInt = i
	repository.createOrUpdateBlockReturnErr = err
}

func (repository *MockBlockRepository) SetMissingBlockNumbersReturnArray(returnArray []int64) {
	repository.missingBlockNumbersReturnArray = returnArray
}

func (repository *MockBlockRepository) CreateOrUpdateBlock(block core.Block) (int64, error) {
	repository.createOrUpdateBlockCallCount++
	repository.createOrUpdateBlockCalled = true
	repository.createOrUpdateBlockPassedBlock = block
	repository.createOrUpdateBlockPassedBlockNumbers = append(repository.createOrUpdateBlockPassedBlockNumbers, block.Number)
	return repository.createOrUpdateBlockReturnInt, repository.createOrUpdateBlockReturnErr
}

func (repository *MockBlockRepository) GetBlock(blockNumber int64) (core.Block, error) {
	return core.Block{Number: blockNumber}, nil
}

func (repository *MockBlockRepository) MissingBlockNumbers(startingBlockNumber int64, endingBlockNumber int64, nodeID string) []int64 {
	repository.missingBlockNumbersCalled = true
	repository.missingBlockNumbersPassedStartingBlockNumber = startingBlockNumber
	repository.missingBlockNumbersPassedEndingBlockNumber = endingBlockNumber
	repository.missingBlockNumbersPassedNodeID = nodeID
	return repository.missingBlockNumbersReturnArray
}

func (repository *MockBlockRepository) SetBlocksStatus(chainHead int64) error {
	repository.setBlockStatusCalled = true
	repository.setBlockStatusPassedChainHead = chainHead
	return nil
}

func (repository *MockBlockRepository) AssertCreateOrUpdateBlockCallCountEquals(times int) {
	Expect(repository.createOrUpdateBlockCallCount).To(Equal(times))
}

func (repository *MockBlockRepository) AssertCreateOrUpdateBlocksCallCountAndBlockNumbersEquals(times int, blockNumbers []int64) {
	Expect(repository.createOrUpdateBlockCallCount).To(Equal(times))
	Expect(repository.createOrUpdateBlockPassedBlockNumbers).To(Equal(blockNumbers))
}

func (repository *MockBlockRepository) AssertCreateOrUpdateBlockCalledWith(block core.Block) {
	Expect(repository.createOrUpdateBlockCalled).To(BeTrue())
	Expect(repository.createOrUpdateBlockPassedBlock).To(Equal(block))
}

func (repository *MockBlockRepository) AssertMissingBlockNumbersCalledWith(startingBlockNumber int64, endingBlockNumber int64, nodeID string) {
	Expect(repository.missingBlockNumbersCalled).To(BeTrue())
	Expect(repository.missingBlockNumbersPassedStartingBlockNumber).To(Equal(startingBlockNumber))
	Expect(repository.missingBlockNumbersPassedEndingBlockNumber).To(Equal(endingBlockNumber))
	Expect(repository.missingBlockNumbersPassedNodeID).To(Equal(nodeID))
}

func (repository *MockBlockRepository) AssertSetBlockStatusCalledWith(chainHead int64) {
	Expect(repository.setBlockStatusCalled).To(BeTrue())
	Expect(repository.setBlockStatusPassedChainHead).To(Equal(chainHead))
}