vat.fold: switch to batched interface

pkg/transformers/test_data/mocks/vat_fold/converter.go

@@ -23,12 +23,12 @@ import (
 
 type MockVatFoldConverter struct {
 	converterErr error
-	PassedLog    types.Log
+	PassedLogs   []types.Log
 }
 
-func (converter *MockVatFoldConverter) ToModel(ethLog types.Log) (vat_fold.VatFoldModel, error) {
-	converter.PassedLog = ethLog
-	return test_data.VatFoldModel, converter.converterErr
+func (converter *MockVatFoldConverter) ToModels(ethLogs []types.Log) ([]vat_fold.VatFoldModel, error) {
+	converter.PassedLogs = ethLogs
+	return []vat_fold.VatFoldModel{test_data.VatFoldModel}, converter.converterErr
 }
 
 func (converter *MockVatFoldConverter) SetConverterError(e error) {

pkg/transformers/test_data/mocks/vat_fold/repository.go

@@ -15,32 +15,45 @@
 package vat_fold
 
 import (
+	. "github.com/onsi/gomega"
+
 	"github.com/vulcanize/vulcanizedb/pkg/core"
 	"github.com/vulcanize/vulcanizedb/pkg/transformers/vat_fold"
 )
 
 type MockVatFoldRepository struct {
-	createErr                 error
-	missingHeaders            []core.Header
-	missingHeadersErr         error
-	PassedStartingBlockNumber int64
-	PassedEndingBlockNumber   int64
-	PassedHeaderID            int64
-	PassedModel               vat_fold.VatFoldModel
+	createErr                       error
+	markHeaderCheckedErr            error
+	markHeaderCheckedPassedHeaderID int64
+	missingHeaders                  []core.Header
+	missingHeadersErr               error
+	PassedStartingBlockNumber       int64
+	PassedEndingBlockNumber         int64
+	PassedHeaderID                  int64
+	PassedModels                    []vat_fold.VatFoldModel
 }
 
-func (repository *MockVatFoldRepository) Create(headerID int64, model vat_fold.VatFoldModel) error {
+func (repository *MockVatFoldRepository) Create(headerID int64, models []vat_fold.VatFoldModel) error {
 	repository.PassedHeaderID = headerID
-	repository.PassedModel = model
+	repository.PassedModels = models
 	return repository.createErr
 }
 
+func (repository *MockVatFoldRepository) MarkHeaderChecked(headerID int64) error {
+	repository.markHeaderCheckedPassedHeaderID = headerID
+	return repository.markHeaderCheckedErr
+}
+
 func (repository *MockVatFoldRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
 	repository.PassedStartingBlockNumber = startingBlockNumber
 	repository.PassedEndingBlockNumber = endingBlockNumber
 	return repository.missingHeaders, repository.missingHeadersErr
 }
 
+func (repository *MockVatFoldRepository) SetMarkHeaderCheckedErr(e error) {
+	repository.markHeaderCheckedErr = e
+}
+
 func (repository *MockVatFoldRepository) SetMissingHeadersErr(e error) {
 	repository.missingHeadersErr = e
 }
@@ -52,3 +65,7 @@ func (repository *MockVatFoldRepository) SetMissingHeaders(headers []core.Header
 func (repository *MockVatFoldRepository) SetCreateError(e error) {
 	repository.createErr = e
 }
+
+func (repository *MockVatFoldRepository) AssertMarkHeaderCheckedCalledWith(i int64) {
+	Expect(repository.markHeaderCheckedPassedHeaderID).To(Equal(i))
+}