Log checked headers for dent events

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

@@ -28,9 +28,9 @@ type MockDentConverter struct {
 	LogsToConvert         []types.Log
 }
 
-func (c *MockDentConverter) ToModel(ethLog types.Log) (dent.DentModel, error) {
-	c.LogsToConvert = append(c.LogsToConvert, ethLog)
-	return test_data.DentModel, c.converterError
+func (c *MockDentConverter) ToModels(ethLogs []types.Log) ([]dent.DentModel, error) {
+	c.LogsToConvert = append(c.LogsToConvert, ethLogs...)
+	return []dent.DentModel{test_data.DentModel}, c.converterError
 }
 
 func (c *MockDentConverter) SetConverterError(err error) {

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

@@ -15,27 +15,36 @@
 package dent
 
 import (
+	. "github.com/onsi/gomega"
+
 	"github.com/vulcanize/vulcanizedb/pkg/core"
 	"github.com/vulcanize/vulcanizedb/pkg/transformers/dent"
 )
 
 type MockDentRepository struct {
-	PassedStartingBlockNumber int64
-	PassedEndingBlockNumber   int64
-	PassedDentModels          []dent.DentModel
-	PassedHeaderIds           []int64
-	missingHeaders            []core.Header
-	missingHeadersError       error
-	createError               error
+	PassedStartingBlockNumber       int64
+	PassedEndingBlockNumber         int64
+	PassedDentModels                []dent.DentModel
+	PassedHeaderIds                 []int64
+	markHeaderCheckedErr            error
+	markHeaderCheckedPassedHeaderId int64
+	missingHeaders                  []core.Header
+	missingHeadersError             error
+	createError                     error
 }
 
-func (r *MockDentRepository) Create(headerId int64, model dent.DentModel) error {
+func (r *MockDentRepository) Create(headerId int64, models []dent.DentModel) error {
 	r.PassedHeaderIds = append(r.PassedHeaderIds, headerId)
-	r.PassedDentModels = append(r.PassedDentModels, model)
+	r.PassedDentModels = append(r.PassedDentModels, models...)
 
 	return r.createError
 }
 
+func (r *MockDentRepository) MarkHeaderChecked(headerId int64) error {
+	r.markHeaderCheckedPassedHeaderId = headerId
+	return r.markHeaderCheckedErr
+}
+
 func (r *MockDentRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
 	r.PassedStartingBlockNumber = startingBlockNumber
 	r.PassedEndingBlockNumber = endingBlockNumber
@@ -43,6 +52,10 @@ func (r *MockDentRepository) MissingHeaders(startingBlockNumber, endingBlockNumb
 	return r.missingHeaders, r.missingHeadersError
 }
 
+func (r *MockDentRepository) SetMarkHeaderCheckedErr(err error) {
+	r.markHeaderCheckedErr = err
+}
+
 func (r *MockDentRepository) SetMissingHeadersError(err error) {
 	r.missingHeadersError = err
 }
@@ -54,3 +67,7 @@ func (r *MockDentRepository) SetMissingHeaders(headers []core.Header) {
 func (r *MockDentRepository) SetCreateError(err error) {
 	r.createError = err
 }
+
+func (r *MockDentRepository) AssertMarkHeaderCheckedCalledWith(headerId int64) {
+	Expect(r.markHeaderCheckedPassedHeaderId).To(Equal(headerId))
+}