rebase; extract factories and the mocks they are dependent on to

libraries/shared; adjust omni test_helpers to drop and recreate
checked_headers table to avoid reaching 1600 column limit after repeated
tests (dropping columns doesn't actually remove them from contributing
to that limit)

libraries/shared/mocks/converter.go

@@ -0,0 +1,58 @@
+// VulcanizeDB
+// Copyright © 2018 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package mocks
+
+import (
+	"github.com/ethereum/go-ethereum/core/types"
+)
+
+type MockConverter struct {
+	ToEntitiesError         error
+	PassedContractAddresses []string
+	ToModelsError           error
+	entityConverterError    error
+	modelConverterError     error
+	ContractAbi             string
+	LogsToConvert           []types.Log
+	EntitiesToConvert       []interface{}
+	EntitiesToReturn        []interface{}
+	ModelsToReturn          []interface{}
+	ToEntitiesCalledCounter int
+	ToModelsCalledCounter   int
+}
+
+func (converter *MockConverter) ToEntities(contractAbi string, ethLogs []types.Log) ([]interface{}, error) {
+	for _, log := range ethLogs {
+		converter.PassedContractAddresses = append(converter.PassedContractAddresses, log.Address.Hex())
+	}
+	converter.ContractAbi = contractAbi
+	converter.LogsToConvert = ethLogs
+	return converter.EntitiesToReturn, converter.ToEntitiesError
+}
+
+func (converter *MockConverter) ToModels(entities []interface{}) ([]interface{}, error) {
+	converter.EntitiesToConvert = entities
+	return converter.ModelsToReturn, converter.ToModelsError
+}
+
+func (converter *MockConverter) SetToEntityConverterError(err error) {
+	converter.entityConverterError = err
+}
+
+func (c *MockConverter) SetToModelConverterError(err error) {
+	c.modelConverterError = err
+}

libraries/shared/mocks/log_note_converter.go

@@ -0,0 +1,42 @@
+// VulcanizeDB
+// Copyright © 2018 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package mocks
+
+import (
+	"github.com/ethereum/go-ethereum/core/types"
+)
+
+type MockLogNoteConverter struct {
+	err                   error
+	returnModels          []interface{}
+	PassedLogs            []types.Log
+	ToModelsCalledCounter int
+}
+
+func (converter *MockLogNoteConverter) ToModels(ethLogs []types.Log) ([]interface{}, error) {
+	converter.PassedLogs = ethLogs
+	converter.ToModelsCalledCounter++
+	return converter.returnModels, converter.err
+}
+
+func (converter *MockLogNoteConverter) SetConverterError(e error) {
+	converter.err = e
+}
+
+func (converter *MockLogNoteConverter) SetReturnModels(models []interface{}) {
+	converter.returnModels = models
+}

libraries/shared/mocks/mappings.go

@@ -0,0 +1,39 @@
+// VulcanizeDB
+// Copyright © 2018 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package mocks
+
+import (
+	"github.com/ethereum/go-ethereum/common"
+
+	"github.com/vulcanize/vulcanizedb/libraries/shared/storage/utils"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+)
+
+type MockMappings struct {
+	Metadata     utils.StorageValueMetadata
+	LookupCalled bool
+	LookupErr    error
+}
+
+func (mappings *MockMappings) Lookup(key common.Hash) (utils.StorageValueMetadata, error) {
+	mappings.LookupCalled = true
+	return mappings.Metadata, mappings.LookupErr
+}
+
+func (*MockMappings) SetDB(db *postgres.DB) {
+	panic("implement me")
+}

libraries/shared/mocks/repository.go

@@ -0,0 +1,98 @@
+// VulcanizeDB
+// Copyright © 2018 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package mocks
+
+import (
+	. "github.com/onsi/gomega"
+
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+)
+
+type MockRepository struct {
+	createError                      error
+	markHeaderCheckedError           error
+	MarkHeaderCheckedPassedHeaderIDs []int64
+	CreatedHeaderIds                 []int64
+	missingHeaders                   []core.Header
+	allHeaders                       []core.Header
+	missingHeadersError              error
+	PassedStartingBlockNumber        int64
+	PassedEndingBlockNumber          int64
+	PassedHeaderID                   int64
+	PassedModels                     []interface{}
+	SetDbCalled                      bool
+	CreateCalledCounter              int
+}
+
+func (repository *MockRepository) Create(headerID int64, models []interface{}) error {
+	repository.PassedHeaderID = headerID
+	repository.PassedModels = models
+	repository.CreatedHeaderIds = append(repository.CreatedHeaderIds, headerID)
+	repository.CreateCalledCounter++
+
+	return repository.createError
+}
+
+func (repository *MockRepository) MarkHeaderChecked(headerID int64) error {
+	repository.MarkHeaderCheckedPassedHeaderIDs = append(repository.MarkHeaderCheckedPassedHeaderIDs, headerID)
+	return repository.markHeaderCheckedError
+}
+
+func (repository *MockRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
+	repository.PassedStartingBlockNumber = startingBlockNumber
+	repository.PassedEndingBlockNumber = endingBlockNumber
+	return repository.missingHeaders, repository.missingHeadersError
+}
+
+func (repository *MockRepository) RecheckHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
+	repository.PassedStartingBlockNumber = startingBlockNumber
+	repository.PassedEndingBlockNumber = endingBlockNumber
+	return repository.allHeaders, nil
+}
+
+func (repository *MockRepository) SetDB(db *postgres.DB) {
+	repository.SetDbCalled = true
+}
+
+func (repository *MockRepository) SetMissingHeadersError(e error) {
+	repository.missingHeadersError = e
+}
+
+func (repository *MockRepository) SetAllHeaders(headers []core.Header) {
+	repository.allHeaders = headers
+}
+
+func (repository *MockRepository) SetMissingHeaders(headers []core.Header) {
+	repository.missingHeaders = headers
+}
+
+func (repository *MockRepository) SetMarkHeaderCheckedError(e error) {
+	repository.markHeaderCheckedError = e
+}
+
+func (repository *MockRepository) SetCreateError(e error) {
+	repository.createError = e
+}
+
+func (repository *MockRepository) AssertMarkHeaderCheckedCalledWith(i int64) {
+	Expect(repository.MarkHeaderCheckedPassedHeaderIDs).To(ContainElement(i))
+}
+
+func (repository *MockRepository) AssertMarkHeaderCheckedNotCalled() {
+	Expect(len(repository.MarkHeaderCheckedPassedHeaderIDs)).To(Equal(0))
+}

libraries/shared/mocks/storage_queue.go

@@ -0,0 +1,31 @@
+// VulcanizeDB
+// Copyright © 2018 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package mocks
+
+import (
+	"github.com/vulcanize/vulcanizedb/libraries/shared/storage/utils"
+)
+
+type MockStorageQueue struct {
+	AddCalled bool
+	AddError  error
+}
+
+func (queue *MockStorageQueue) Add(row utils.StorageDiffRow) error {
+	queue.AddCalled = true
+	return queue.AddError
+}

libraries/shared/mocks/storage_repository.go

@@ -0,0 +1,42 @@
+// VulcanizeDB
+// Copyright © 2018 Vulcanize
+
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+package mocks
+
+import (
+	"github.com/vulcanize/vulcanizedb/libraries/shared/storage/utils"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+)
+
+type MockStorageRepository struct {
+	CreateErr         error
+	PassedBlockNumber int
+	PassedBlockHash   string
+	PassedMetadata    utils.StorageValueMetadata
+	PassedValue       interface{}
+}
+
+func (repository *MockStorageRepository) Create(blockNumber int, blockHash string, metadata utils.StorageValueMetadata, value interface{}) error {
+	repository.PassedBlockNumber = blockNumber
+	repository.PassedBlockHash = blockHash
+	repository.PassedMetadata = metadata
+	repository.PassedValue = value
+	return repository.CreateErr
+}
+
+func (*MockStorageRepository) SetDB(db *postgres.DB) {
+	panic("implement me")
+}