52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
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))
|
|
}
|