Use transformer factory for Vat flux

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

@@ -4,19 +4,18 @@ import (
 	"github.com/ethereum/go-ethereum/core/types"
 
 	"github.com/vulcanize/vulcanizedb/pkg/transformers/test_data"
-	"github.com/vulcanize/vulcanizedb/pkg/transformers/vat_flux"
 )
 
-type MockVatFlux struct {
+type MockVatFluxConverter struct {
 	err        error
 	PassedLogs []types.Log
 }
 
-func (converter *MockVatFlux) ToModels(ethLogs []types.Log) ([]vat_flux.VatFluxModel, error) {
+func (converter *MockVatFluxConverter) ToModels(ethLogs []types.Log) ([]interface{}, error) {
 	converter.PassedLogs = ethLogs
-	return []vat_flux.VatFluxModel{test_data.VatFluxModel}, converter.err
+	return []interface{}{test_data.VatFluxModel}, converter.err
 }
 
-func (converter *MockVatFlux) SetConverterError(e error) {
+func (converter *MockVatFluxConverter) SetConverterError(e error) {
 	converter.err = e
 }

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

@@ -16,7 +16,7 @@ package vat_flux
 
 import (
 	"github.com/vulcanize/vulcanizedb/pkg/core"
-	"github.com/vulcanize/vulcanizedb/pkg/transformers/vat_flux"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
 )
 
 type MockVatFluxRepository struct {
@@ -28,26 +28,31 @@ type MockVatFluxRepository struct {
 	PassedStartingBlockNumber       int64
 	PassedEndingBlockNumber         int64
 	PassedHeaderID                  int64
-	PassedModels                    []vat_flux.VatFluxModel
+	PassedModels                    []interface{}
+	SetDbCalled                     bool
 }
 
-func (repository *MockVatFluxRepository) MarkCheckedHeader(headerId int64) error {
-	repository.MarkHeaderCheckedPassedHeaderID = headerId
-	return repository.markHeaderCheckedErr
-}
-
-func (repository *MockVatFluxRepository) Create(headerID int64, models []vat_flux.VatFluxModel) error {
+func (repository *MockVatFluxRepository) Create(headerID int64, models []interface{}) error {
 	repository.PassedHeaderID = headerID
 	repository.PassedModels = models
 	return repository.createErr
 }
 
+func (repository *MockVatFluxRepository) MarkHeaderChecked(headerID int64) error {
+	repository.MarkHeaderCheckedPassedHeaderID = headerID
+	return repository.markHeaderCheckedErr
+}
+
 func (repository *MockVatFluxRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
 	repository.PassedStartingBlockNumber = startingBlockNumber
 	repository.PassedEndingBlockNumber = endingBlockNumber
 	return repository.missingHeaders, repository.missingHeadersErr
 }
 
+func (repository *MockVatFluxRepository) SetDB(db *postgres.DB) {
+	repository.SetDbCalled = true
+}
+
 func (repository *MockVatFluxRepository) SetMarkHeaderCheckedErr(e error) {
 	repository.markHeaderCheckedErr = e
 }