Log checked headers for tend events

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

@@ -22,17 +22,13 @@ import (
 )
 
 type MockTendConverter struct {
-	ConverterContract string
-	ConverterAbi      string
-	LogsToConvert     []types.Log
-	ConverterError    error
+	LogsToConvert  []types.Log
+	ConverterError error
 }
 
-func (c *MockTendConverter) Convert(contractAddress string, contractAbi string, ethLog types.Log) (tend.TendModel, error) {
-	c.ConverterContract = contractAddress
-	c.ConverterAbi = contractAbi
-	c.LogsToConvert = append(c.LogsToConvert, ethLog)
-	return test_data.TendModel, c.ConverterError
+func (c *MockTendConverter) Convert(ethLogs []types.Log) ([]tend.TendModel, error) {
+	c.LogsToConvert = append(c.LogsToConvert, ethLogs...)
+	return []tend.TendModel{test_data.TendModel}, c.ConverterError
 }
 
 func (c *MockTendConverter) SetConverterError(err error) {

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

@@ -15,23 +15,27 @@
 package tend
 
 import (
+	. "github.com/onsi/gomega"
+
 	"github.com/vulcanize/vulcanizedb/pkg/core"
 	"github.com/vulcanize/vulcanizedb/pkg/transformers/tend"
 )
 
 type MockTendRepository struct {
-	createError               error
-	PassedEndingBlockNumber   int64
-	PassedHeaderID            int64
-	PassedStartingBlockNumber int64
-	PassedTendModel           tend.TendModel
-	missingHeaders            []core.Header
-	missingHeadersErr         error
+	createError                     error
+	PassedEndingBlockNumber         int64
+	PassedHeaderID                  int64
+	PassedStartingBlockNumber       int64
+	PassedTendModel                 tend.TendModel
+	markHeaderCheckedErr            error
+	markHeaderCheckedPassedHeaderId int64
+	missingHeaders                  []core.Header
+	missingHeadersErr               error
 }
 
-func (repository *MockTendRepository) Create(headerId int64, tend tend.TendModel) error {
+func (repository *MockTendRepository) Create(headerId int64, tend []tend.TendModel) error {
 	repository.PassedHeaderID = headerId
-	repository.PassedTendModel = tend
+	repository.PassedTendModel = tend[0]
 	return repository.createError
 }
 
@@ -39,6 +43,10 @@ func (repository *MockTendRepository) SetCreateError(err error) {
 	repository.createError = err
 }
 
+func (repository *MockTendRepository) SetMarkHeaderCheckedErr(err error) {
+	repository.markHeaderCheckedErr = err
+}
+
 func (repository *MockTendRepository) SetMissingHeadersErr(err error) {
 	repository.missingHeadersErr = err
 }
@@ -47,8 +55,17 @@ func (repository *MockTendRepository) SetMissingHeaders(headers []core.Header) {
 	repository.missingHeaders = headers
 }
 
+func (repository *MockTendRepository) MarkHeaderChecked(headerId int64) error {
+	repository.markHeaderCheckedPassedHeaderId = headerId
+	return repository.markHeaderCheckedErr
+}
+
 func (repository *MockTendRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
 	repository.PassedStartingBlockNumber = startingBlockNumber
 	repository.PassedEndingBlockNumber = endingBlockNumber
 	return repository.missingHeaders, repository.missingHeadersErr
 }
+
+func (repository *MockTendRepository) AssertMarkHeaderCheckedCalledWith(headerId int64) {
+	Expect(repository.markHeaderCheckedPassedHeaderId).To(Equal(headerId))
+}