Backfill Frob log events

2018-08-14 10:47:43 -05:00
parent 55229cd2eb
commit ac63e43543
40 changed files with 1251 additions and 149 deletions
@@ -0,0 +1,44 @@
+// Copyright 2018 Vulcanize
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package flip_kick
+
+import (
+	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/flip_kick"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/test_data"
+)
+
+type MockFlipKickConverter struct {
+	ConverterContract string
+	ConverterAbi      string
+	LogsToConvert     []types.Log
+	EntitiesToConvert []flip_kick.FlipKickEntity
+	ConverterError    error
+}
+
+func (mfkc *MockFlipKickConverter) ToEntity(contractAddress string, contractAbi string, ethLog types.Log) (*flip_kick.FlipKickEntity, error) {
+	mfkc.ConverterContract = contractAddress
+	mfkc.ConverterAbi = contractAbi
+	mfkc.LogsToConvert = append(mfkc.LogsToConvert, ethLog)
+	return &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) SetConverterError(err error) {
+	mfkc.ConverterError = err
+}
@@ -0,0 +1,55 @@
+// Copyright 2018 Vulcanize
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package flip_kick
+
+import (
+	"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
+}
+
+func (mfkr *MockFlipKickRepository) Create(headerId int64, flipKick flip_kick.FlipKickModel) error {
+	mfkr.HeaderIds = append(mfkr.HeaderIds, headerId)
+	mfkr.FlipKicksCreated = append(mfkr.FlipKicksCreated, flipKick)
+
+	return mfkr.CreateRecordError
+}
+
+func (mfkr *MockFlipKickRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
+	mfkr.StartingBlockNumber = startingBlockNumber
+	mfkr.EndingBlockNumber = endingBlockNumber
+
+	return mfkr.HeadersToReturn, mfkr.MissingHeadersError
+}
+
+func (mfkr *MockFlipKickRepository) SetHeadersToReturn(headers []core.Header) {
+	mfkr.HeadersToReturn = headers
+}
+
+func (mfkr *MockFlipKickRepository) SetCreateRecordError(err error) {
+	mfkr.CreateRecordError = err
+}
+func (mfkr *MockFlipKickRepository) SetMissingHeadersError(err error) {
+	mfkr.MissingHeadersError = err
+}
@@ -0,0 +1,45 @@
+// Copyright 2018 Vulcanize
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package frob
+
+import (
+	"github.com/ethereum/go-ethereum/core/types"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/frob"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/test_data"
+)
+
+type MockFrobConverter struct {
+	PassedContractAddress string
+	PassedContractABI     string
+	PassedLog             types.Log
+	PassedEntity          frob.FrobEntity
+	converterError        error
+}
+
+func (converter *MockFrobConverter) SetConverterError(err error) {
+	converter.converterError = err
+}
+
+func (converter *MockFrobConverter) ToEntity(contractAddress string, contractAbi string, ethLog types.Log) (frob.FrobEntity, error) {
+	converter.PassedContractAddress = contractAddress
+	converter.PassedContractABI = contractAbi
+	converter.PassedLog = ethLog
+	return test_data.FrobEntity, converter.converterError
+}
+
+func (converter *MockFrobConverter) ToModel(frobEntity frob.FrobEntity) frob.FrobModel {
+	converter.PassedEntity = frobEntity
+	return test_data.FrobModel
+}
@@ -0,0 +1,56 @@
+// Copyright 2018 Vulcanize
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package frob
+
+import (
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/frob"
+)
+
+type MockFrobRepository struct {
+	createError               error
+	PassedEndingBlockNumber   int64
+	PassedFrobModel           frob.FrobModel
+	PassedHeaderID            int64
+	PassedStartingBlockNumber int64
+	PassedTransactionIndex    uint
+	missingHeaders            []core.Header
+	missingHeadersErr         error
+}
+
+func (repository *MockFrobRepository) SetCreateError(err error) {
+	repository.createError = err
+}
+
+func (repository *MockFrobRepository) SetMissingHeadersErr(err error) {
+	repository.missingHeadersErr = err
+}
+
+func (repository *MockFrobRepository) SetMissingHeaders(headers []core.Header) {
+	repository.missingHeaders = headers
+}
+
+func (repository *MockFrobRepository) Create(headerID int64, transactionIndex uint, model frob.FrobModel) error {
+	repository.PassedHeaderID = headerID
+	repository.PassedTransactionIndex = transactionIndex
+	repository.PassedFrobModel = model
+	return repository.createError
+}
+
+func (repository *MockFrobRepository) MissingHeaders(startingBlockNumber, endingBlockNumber int64) ([]core.Header, error) {
+	repository.PassedStartingBlockNumber = startingBlockNumber
+	repository.PassedEndingBlockNumber = endingBlockNumber
+	return repository.missingHeaders, repository.missingHeadersErr
+}
@@ -0,0 +1,44 @@
+// Copyright 2018 Vulcanize
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package mocks
+
+import (
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+)
+
+type MockLogFetcher struct {
+	FetchedContractAddress string
+	FetchedTopics          [][]common.Hash
+	FetchedBlocks          []int64
+	fetcherError           error
+	FetchedLogs            []types.Log
+}
+
+func (mlf *MockLogFetcher) FetchLogs(contractAddress string, topics [][]common.Hash, blockNumber int64) ([]types.Log, error) {
+	mlf.FetchedContractAddress = contractAddress
+	mlf.FetchedTopics = topics
+	mlf.FetchedBlocks = append(mlf.FetchedBlocks, blockNumber)
+
+	return mlf.FetchedLogs, mlf.fetcherError
+}
+
+func (mlf *MockLogFetcher) SetFetcherError(err error) {
+	mlf.fetcherError = err
+}
+
+func (mlf *MockLogFetcher) SetFetchedLogs(logs []types.Log) {
+	mlf.FetchedLogs = logs
+}