Log checked headers for flip kick events

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

@@ -29,17 +29,18 @@ type MockFlipKickConverter struct {
 	ConverterError    error
 }
 
-func (mfkc *MockFlipKickConverter) ToEntity(contractAddress string, contractAbi string, ethLog types.Log) (*flip_kick.FlipKickEntity, error) {
+func (mfkc *MockFlipKickConverter) ToEntities(contractAddress string, contractAbi string, ethLogs []types.Log) ([]flip_kick.FlipKickEntity, error) {
 	mfkc.ConverterContract = contractAddress
 	mfkc.ConverterAbi = contractAbi
-	mfkc.LogsToConvert = append(mfkc.LogsToConvert, ethLog)
-	return &test_data.FlipKickEntity, mfkc.ConverterError
+	mfkc.LogsToConvert = append(mfkc.LogsToConvert, ethLogs...)
+	return []flip_kick.FlipKickEntity{test_data.FlipKickEntity}, mfkc.ConverterError
 }
 
-func (mfkc *MockFlipKickConverter) ToModel(flipKickEntity flip_kick.FlipKickEntity) (flip_kick.FlipKickModel, error) {
-	mfkc.EntitiesToConvert = append(mfkc.EntitiesToConvert, flipKickEntity)
-	return test_data.FlipKickModel, nil
+func (mfkc *MockFlipKickConverter) ToModels(flipKickEntities []flip_kick.FlipKickEntity) ([]flip_kick.FlipKickModel, error) {
+	mfkc.EntitiesToConvert = append(mfkc.EntitiesToConvert, flipKickEntities...)
+	return []flip_kick.FlipKickModel{test_data.FlipKickModel}, nil
 }
+
 func (mfkc *MockFlipKickConverter) SetConverterError(err error) {
 	mfkc.ConverterError = err
 }

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

@@ -15,27 +15,36 @@
 package flip_kick
 
 import (
+	. "github.com/onsi/gomega"
+
 	"github.com/vulcanize/vulcanizedb/pkg/core"
 	"github.com/vulcanize/vulcanizedb/pkg/transformers/flip_kick"
 )
 
 type MockFlipKickRepository struct {
-	HeaderIds           []int64
-	HeadersToReturn     []core.Header
-	StartingBlockNumber int64
-	EndingBlockNumber   int64
-	FlipKicksCreated    []flip_kick.FlipKickModel
-	CreateRecordError   error
-	MissingHeadersError error
+	CreateRecordError               error
+	EndingBlockNumber               int64
+	FlipKicksCreated                []flip_kick.FlipKickModel
+	HeaderIds                       []int64
+	HeadersToReturn                 []core.Header
+	MissingHeadersError             error
+	StartingBlockNumber             int64
+	markHeaderCheckedErr            error
+	markHeaderCheckedPassedHeaderId int64
 }
 
-func (mfkr *MockFlipKickRepository) Create(headerId int64, flipKick flip_kick.FlipKickModel) error {
+func (mfkr *MockFlipKickRepository) Create(headerId int64, flipKick []flip_kick.FlipKickModel) error {
 	mfkr.HeaderIds = append(mfkr.HeaderIds, headerId)
-	mfkr.FlipKicksCreated = append(mfkr.FlipKicksCreated, flipKick)
+	mfkr.FlipKicksCreated = append(mfkr.FlipKicksCreated, flipKick...)
 
 	return mfkr.CreateRecordError
 }
 
+func (mfkr *MockFlipKickRepository) MarkHeaderChecked(headerId int64) error {
+	mfkr.markHeaderCheckedPassedHeaderId = headerId
+	return mfkr.markHeaderCheckedErr
+}
+
 func (mfkr *MockFlipKickRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
 	mfkr.StartingBlockNumber = startingBlockNumber
 	mfkr.EndingBlockNumber = endingBlockNumber
@@ -50,6 +59,15 @@ func (mfkr *MockFlipKickRepository) SetHeadersToReturn(headers []core.Header) {
 func (mfkr *MockFlipKickRepository) SetCreateRecordError(err error) {
 	mfkr.CreateRecordError = err
 }
+
+func (mfkr *MockFlipKickRepository) SetMarkHeaderCheckedErr(err error) {
+	mfkr.markHeaderCheckedErr = err
+}
+
 func (mfkr *MockFlipKickRepository) SetMissingHeadersError(err error) {
 	mfkr.MissingHeadersError = err
 }
+
+func (mfkr *MockFlipKickRepository) AssertMarkHeaderCheckedCalledWith(headerId int64) {
+	Expect(mfkr.markHeaderCheckedPassedHeaderId).To(Equal(headerId))
+}