deep-stack/ipld-eth-server
6
0
Fork 0
forked from cerc-io/ipld-eth-server
Code Pull Requests Activity

Remove unused functions on mock event repository

Browse Source
This commit is contained in:
Rob Mulholand 2019-09-10 14:20:47 -05:00
parent 3d6c973f6e
commit c568fedd89
1 changed files with 4 additions and 51 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
libraries/shared/mocks/event_repository.go
View File

@ -17,24 +17,14 @@
package mocks package mocks
import ( import (
. "github.com/onsi/gomega"
"github.com/vulcanize/vulcanizedb/pkg/core"
"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres" "github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
) )
type MockEventRepository struct { type MockEventRepository struct {
createError error createError error
markHeaderCheckedError error PassedModels []interface{}
MarkHeaderCheckedPassedHeaderIDs []int64 SetDbCalled bool
missingHeaders []core.Header CreateCalledCounter int
allHeaders []core.Header
missingHeadersError error
PassedStartingBlockNumber int64
PassedEndingBlockNumber int64
PassedModels []interface{}
SetDbCalled bool
CreateCalledCounter int
} }
func (repository *MockEventRepository) Create(models []interface{}) error { func (repository *MockEventRepository) Create(models []interface{}) error {
@ -44,47 +34,10 @@ func (repository *MockEventRepository) Create(models []interface{}) error {
return repository.createError return repository.createError
} }
func (repository *MockEventRepository) MarkHeaderChecked(headerID int64) error {
repository.MarkHeaderCheckedPassedHeaderIDs = append(repository.MarkHeaderCheckedPassedHeaderIDs, headerID)
return repository.markHeaderCheckedError
}
func (repository *MockEventRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
repository.PassedStartingBlockNumber = startingBlockNumber
repository.PassedEndingBlockNumber = endingBlockNumber
return repository.missingHeaders, repository.missingHeadersError
}
func (repository *MockEventRepository) RecheckHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
repository.PassedStartingBlockNumber = startingBlockNumber
repository.PassedEndingBlockNumber = endingBlockNumber
return repository.allHeaders, nil
}
func (repository *MockEventRepository) SetDB(db *postgres.DB) { func (repository *MockEventRepository) SetDB(db *postgres.DB) {
repository.SetDbCalled = true repository.SetDbCalled = true
} }
func (repository *MockEventRepository) SetMissingHeadersError(e error) {
repository.missingHeadersError = e
}
func (repository *MockEventRepository) SetAllHeaders(headers []core.Header) {
repository.allHeaders = headers
}
func (repository *MockEventRepository) SetMissingHeaders(headers []core.Header) {
repository.missingHeaders = headers
}
func (repository *MockEventRepository) SetMarkHeaderCheckedError(e error) {
repository.markHeaderCheckedError = e
}
func (repository *MockEventRepository) SetCreateError(e error) { func (repository *MockEventRepository) SetCreateError(e error) {
repository.createError = e repository.createError = e
} }
func (repository *MockEventRepository) AssertMarkHeaderCheckedCalledWith(i int64) {
Expect(repository.MarkHeaderCheckedPassedHeaderIDs).To(ContainElement(i))
}