Vat heal (#56)

* VatHeal Converter

* Add VatHeal repository

* Add VatHeal transformer

* Add VatHeal to continuousLogSync command

* Mark vat_init_checked as true when creating vat init records

* Update urn and v converting

* Return error if Repository.MarkCheckedHeader fails

* Add deleting vat heal from test cleanup method

pkg/transformers/vat_heal/config.go

@@ -0,0 +1,25 @@
+// 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 vat_heal
+
+import "github.com/vulcanize/vulcanizedb/pkg/transformers/shared"
+
+var VatHealConfig = shared.TransformerConfig{
+	ContractAddresses:   []string{shared.VatContractAddress},
+	ContractAbi:         shared.VatABI,
+	Topics:              []string{shared.VatHealSignature},
+	StartingBlockNumber: 0,
+	EndingBlockNumber:   10000000,
+}

pkg/transformers/vat_heal/converter.go

@@ -0,0 +1,70 @@
+// 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 vat_heal
+
+import (
+	"encoding/json"
+	"errors"
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+	"strconv"
+)
+
+type Converter interface {
+	ToModels(ethLogs []types.Log) ([]VatHealModel, error)
+}
+
+type VatHealConverter struct{}
+
+func (VatHealConverter) ToModels(ethLogs []types.Log) ([]VatHealModel, error) {
+	var models []VatHealModel
+	for _, ethLog := range ethLogs {
+		err := verifyLog(ethLog)
+		if err != nil {
+			return nil, err
+		}
+
+		urn := common.BytesToAddress(ethLog.Topics[1].Bytes()[:common.AddressLength])
+		v := common.BytesToAddress(ethLog.Topics[2].Bytes()[:common.AddressLength])
+		radInt, err := strconv.ParseInt(ethLog.Topics[3].Hex(), 0, 64)
+		if err != nil {
+			return nil, err
+		}
+
+		rawLogJson, err := json.Marshal(ethLog)
+		if err != nil {
+			return nil, err
+		}
+
+		model := VatHealModel{
+			Urn:              urn.String(),
+			V:                v.String(),
+			Rad:              int(radInt),
+			TransactionIndex: ethLog.TxIndex,
+			Raw:              rawLogJson,
+		}
+
+		models = append(models, model)
+	}
+
+	return models, nil
+}
+
+func verifyLog(log types.Log) error {
+	if len(log.Topics) < 4 {
+		return errors.New("log missing topics")
+	}
+	return nil
+}

pkg/transformers/vat_heal/converter_test.go

@@ -0,0 +1,53 @@
+// 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 vat_heal_test
+
+import (
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/test_data"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/vat_heal"
+)
+
+var _ = Describe("VatHeal converter", func() {
+	It("Converts logs to models", func() {
+		converter := vat_heal.VatHealConverter{}
+		models, err := converter.ToModels([]types.Log{test_data.VatHealLog})
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(models[0].Urn).To(Equal(test_data.VatHealModel.Urn))
+		Expect(models[0].V).To(Equal(test_data.VatHealModel.V))
+		Expect(models[0].Rad).To(Equal(test_data.VatHealModel.Rad))
+		Expect(models[0].TransactionIndex).To(Equal(test_data.VatHealModel.TransactionIndex))
+		Expect(models[0].Raw).To(Equal(test_data.VatHealModel.Raw))
+	})
+
+	It("Returns an error there are missing topics", func() {
+		converter := vat_heal.VatHealConverter{}
+		badLog := types.Log{
+			Topics: []common.Hash{
+				common.HexToHash("0x"),
+				common.HexToHash("0x"),
+				common.HexToHash("0x"),
+			},
+		}
+		_, err := converter.ToModels([]types.Log{badLog})
+
+		Expect(err).To(HaveOccurred())
+	})
+})

pkg/transformers/vat_heal/model.go

@@ -0,0 +1,23 @@
+// 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 vat_heal
+
+type VatHealModel struct {
+	Urn              string
+	V                string
+	Rad              int
+	TransactionIndex uint   `db:"tx_idx"`
+	Raw              []byte `db:"raw_log"`
+}

pkg/transformers/vat_heal/repository.go

@@ -0,0 +1,84 @@
+// 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 vat_heal
+
+import (
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+)
+
+type Repository interface {
+	Create(headerId int64, models []VatHealModel) error
+	MissingHeaders(startingBlock, endingBlock int64) ([]core.Header, error)
+	MarkCheckedHeader(headerId int64) error
+}
+
+type VatHealRepository struct {
+	DB *postgres.DB
+}
+
+func NewVatHealRepository(db *postgres.DB) VatHealRepository {
+	return VatHealRepository{DB: db}
+}
+
+func (repository VatHealRepository) Create(headerId int64, models []VatHealModel) error {
+	tx, err := repository.DB.Begin()
+	if err != nil {
+		return err
+	}
+
+	for _, model := range models {
+		_, err := tx.Exec(`INSERT INTO maker.vat_heal (header_id, urn, v, rad, tx_idx, raw_log)
+		VALUES($1, $2, $3, $4, $5, $6)`,
+			headerId, model.Urn, model.V, model.Rad, model.TransactionIndex, model.Raw)
+		if err != nil {
+			tx.Rollback()
+			return err
+		}
+	}
+
+	_, err = tx.Exec(`INSERT INTO public.checked_headers (header_id, vat_heal_checked)
+			VALUES($1, $2)
+		ON CONFLICT (header_id) DO
+			UPDATE SET vat_heal_checked = $2`, headerId, true)
+	if err != nil {
+		tx.Rollback()
+		return err
+	}
+	return tx.Commit()
+}
+
+func (repository VatHealRepository) MissingHeaders(startingBlock, endingBlock int64) ([]core.Header, error) {
+	var headers []core.Header
+	err := repository.DB.Select(&headers,
+		`SELECT headers.id, block_number from headers
+               LEFT JOIN checked_headers on headers.id = header_id
+               WHERE (header_id ISNULL OR vat_heal_checked IS FALSE)
+               AND headers.block_number >= $1
+               AND headers.block_number <= $2
+               AND headers.eth_node_fingerprint = $3`,
+		startingBlock, endingBlock, repository.DB.Node.ID)
+
+	return headers, err
+}
+
+func (repository VatHealRepository) MarkCheckedHeader(headerId int64) error {
+	_, err := repository.DB.Exec(`INSERT INTO public.checked_headers (header_id, vat_heal_checked)
+			VALUES($1, $2)
+		ON CONFLICT (header_id) DO
+			UPDATE SET vat_heal_checked = $2`, headerId, true)
+
+	return err
+}

pkg/transformers/vat_heal/repository_test.go

@@ -0,0 +1,225 @@
+// 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 vat_heal_test
+
+import (
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres/repositories"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/test_data"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/vat_heal"
+	"github.com/vulcanize/vulcanizedb/test_config"
+)
+
+var _ = Describe("VatHeal Repository", func() {
+	var db *postgres.DB
+	var repository vat_heal.VatHealRepository
+	var headerRepository repositories.HeaderRepository
+	var headerId int64
+	var err error
+
+	BeforeEach(func() {
+		node := test_config.NewTestNode()
+		db = test_config.NewTestDB(node)
+		test_config.CleanTestDB(db)
+		repository = vat_heal.VatHealRepository{DB: db}
+		headerRepository = repositories.NewHeaderRepository(db)
+		headerId, err = headerRepository.CreateOrUpdateHeader(core.Header{})
+		Expect(err).NotTo(HaveOccurred())
+	})
+
+	type VatHealDBResult struct {
+		vat_heal.VatHealModel
+		Id       int
+		HeaderId int64 `db:"header_id"`
+	}
+
+	type CheckedHeaderResult struct {
+		VatHealChecked bool `db:"vat_heal_checked"`
+	}
+
+	Describe("Create", func() {
+		It("persists vat heal records", func() {
+			anotherVatHeal := test_data.VatHealModel
+			anotherVatHeal.TransactionIndex = test_data.VatHealModel.TransactionIndex + 1
+			err = repository.Create(headerId, []vat_heal.VatHealModel{test_data.VatHealModel, anotherVatHeal})
+
+			var dbResult []VatHealDBResult
+			err = db.Select(&dbResult, `SELECT * from maker.vat_heal where header_id = $1`, headerId)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(len(dbResult)).To(Equal(2))
+			Expect(dbResult[0].Urn).To(Equal(test_data.VatHealModel.Urn))
+			Expect(dbResult[0].V).To(Equal(test_data.VatHealModel.V))
+			Expect(dbResult[0].Rad).To(Equal(test_data.VatHealModel.Rad))
+			Expect(dbResult[0].TransactionIndex).To(Equal(test_data.VatHealModel.TransactionIndex))
+			Expect(dbResult[1].TransactionIndex).To(Equal(test_data.VatHealModel.TransactionIndex + 1))
+			Expect(dbResult[0].Raw).To(MatchJSON(test_data.VatHealModel.Raw))
+			Expect(dbResult[0].HeaderId).To(Equal(headerId))
+		})
+
+		It("returns an error if the insertion fails", func() {
+			err = repository.Create(headerId, []vat_heal.VatHealModel{test_data.VatHealModel})
+			Expect(err).NotTo(HaveOccurred())
+			err = repository.Create(headerId, []vat_heal.VatHealModel{test_data.VatHealModel})
+			Expect(err).To(HaveOccurred())
+			Expect(err.Error()).To(ContainSubstring("pq: duplicate key value violates unique constraint"))
+		})
+
+		It("marks the header as checked for vat heal logs", func() {
+			err = repository.Create(headerId, []vat_heal.VatHealModel{test_data.VatHealModel})
+			Expect(err).NotTo(HaveOccurred())
+
+			var headerChecked bool
+			err = db.Get(&headerChecked, `SELECT vat_heal_checked FROM public.checked_headers WHERE header_id = $1`, headerId)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(headerChecked).To(BeTrue())
+		})
+
+		It("removes vat heal if corresponding header is deleted", func() {
+			err = repository.Create(headerId, []vat_heal.VatHealModel{test_data.VatHealModel})
+			Expect(err).NotTo(HaveOccurred())
+
+			_, err = db.Exec(`DELETE FROM headers WHERE id = $1`, headerId)
+
+			Expect(err).NotTo(HaveOccurred())
+			var count int
+			err = db.QueryRow(`SELECT count(*) from maker.vat_heal`).Scan(&count)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(count).To(Equal(0))
+		})
+
+		It("wraps create in a transaction", func() {
+			err = repository.Create(headerId, []vat_heal.VatHealModel{test_data.VatHealModel, test_data.VatHealModel})
+			Expect(err).To(HaveOccurred())
+			var count int
+			err = repository.DB.QueryRowx(`SELECT count(*) FROM maker.vat_heal`).Scan(&count)
+			Expect(count).To(Equal(0))
+		})
+	})
+
+	Describe("MissingHeaders", func() {
+		It("returns headers that haven't been checked", func() {
+			startingBlock := GinkgoRandomSeed()
+			vatHealBlock := startingBlock + 1
+			endingBlock := startingBlock + 2
+			outsideRangeBlock := startingBlock + 3
+
+			var headerIds []int64
+			blockNumbers := []int64{startingBlock, vatHealBlock, endingBlock, outsideRangeBlock}
+			for _, n := range blockNumbers {
+				headerId, err := headerRepository.CreateOrUpdateHeader(core.Header{BlockNumber: n})
+				Expect(err).NotTo(HaveOccurred())
+				headerIds = append(headerIds, headerId)
+			}
+
+			err = repository.MarkCheckedHeader(headerIds[0])
+			Expect(err).NotTo(HaveOccurred())
+
+			headers, err := repository.MissingHeaders(startingBlock, endingBlock)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(headers[0].Id).To(Or(Equal(headerIds[1]), Equal(headerIds[2])))
+			Expect(headers[1].Id).To(Or(Equal(headerIds[1]), Equal(headerIds[2])))
+			Expect(len(headers)).To(Equal(2))
+		})
+
+		It("returns header ids when checked_headers.vat_heal is false", func() {
+			startingBlock := GinkgoRandomSeed()
+			vatHealBlock := startingBlock + 1
+			endingBlock := startingBlock + 2
+			outsideRangeBlock := startingBlock + 3
+
+			var headerIds []int64
+			blockNumbers := []int64{startingBlock, vatHealBlock, endingBlock, outsideRangeBlock}
+			for _, n := range blockNumbers {
+				headerId, err := headerRepository.CreateOrUpdateHeader(core.Header{BlockNumber: n})
+				Expect(err).NotTo(HaveOccurred())
+				headerIds = append(headerIds, headerId)
+			}
+
+			err = repository.MarkCheckedHeader(headerIds[0])
+			_, err = repository.DB.Exec(`INSERT INTO checked_headers (header_id) VALUES ($1)`, headerIds[1])
+			Expect(err).NotTo(HaveOccurred())
+
+			headers, err := repository.MissingHeaders(startingBlock, endingBlock)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(headers[0].Id).To(Or(Equal(headerIds[1]), Equal(headerIds[2])))
+			Expect(headers[1].Id).To(Or(Equal(headerIds[1]), Equal(headerIds[2])))
+			Expect(len(headers)).To(Equal(2))
+		})
+
+		It("only returns header ids for the current node", func() {
+			startingBlock := GinkgoRandomSeed()
+			vatHealBlock := startingBlock + 1
+			endingBlock := startingBlock + 2
+			outsideRangeBlock := startingBlock + 3
+			db2 := test_config.NewTestDB(core.Node{ID: "second node"})
+			headerRepository2 := repositories.NewHeaderRepository(db2)
+			repository2 := vat_heal.NewVatHealRepository(db2)
+
+			var headerIds []int64
+			blockNumbers := []int64{startingBlock, vatHealBlock, endingBlock, outsideRangeBlock}
+			for _, n := range blockNumbers {
+				headerId, err := headerRepository.CreateOrUpdateHeader(core.Header{BlockNumber: n})
+				Expect(err).NotTo(HaveOccurred())
+				headerIds = append(headerIds, headerId)
+
+				_, err = headerRepository2.CreateOrUpdateHeader(core.Header{BlockNumber: n})
+				Expect(err).NotTo(HaveOccurred())
+			}
+
+			err = repository.MarkCheckedHeader(headerIds[0])
+			Expect(err).NotTo(HaveOccurred())
+
+			nodeOneMissingHeaders, err := repository.MissingHeaders(startingBlock, endingBlock)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(len(nodeOneMissingHeaders)).To(Equal(2))
+
+			nodeTwoMissingHeaders, err := repository2.MissingHeaders(startingBlock, endingBlock)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(len(nodeTwoMissingHeaders)).To(Equal(3))
+		})
+	})
+
+	Describe("MarkCheckedHeader", func() {
+		It("creates a new checked_header record", func() {
+			err := repository.MarkCheckedHeader(headerId)
+			Expect(err).NotTo(HaveOccurred())
+
+			var checkedHeaderResult = CheckedHeaderResult{}
+			err = db.Get(&checkedHeaderResult, `SELECT vat_heal_checked FROM checked_headers WHERE header_id = $1`, headerId)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(checkedHeaderResult.VatHealChecked).To(BeTrue())
+		})
+
+		It("updates an existing checked header", func() {
+			_, err := repository.DB.Exec(`INSERT INTO checked_headers (header_id) VALUES($1)`, headerId)
+			Expect(err).NotTo(HaveOccurred())
+
+			var checkedHeaderResult CheckedHeaderResult
+			err = db.Get(&checkedHeaderResult, `SELECT vat_heal_checked FROM checked_headers WHERE header_id = $1`, headerId)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(checkedHeaderResult.VatHealChecked).To(BeFalse())
+
+			err = repository.MarkCheckedHeader(headerId)
+			Expect(err).NotTo(HaveOccurred())
+
+			err = db.Get(&checkedHeaderResult, `SELECT vat_heal_checked FROM checked_headers WHERE header_id = $1`, headerId)
+			Expect(err).NotTo(HaveOccurred())
+			Expect(checkedHeaderResult.VatHealChecked).To(BeTrue())
+		})
+	})
+})

pkg/transformers/vat_heal/transformer.go

@@ -0,0 +1,80 @@
+// 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 vat_heal
+
+import (
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/shared"
+)
+
+type VatHealTransformer struct {
+	Config     shared.TransformerConfig
+	Converter  Converter
+	Fetcher    shared.LogFetcher
+	Repository Repository
+}
+
+type VatHealTransformerInitializer struct {
+	Config shared.TransformerConfig
+}
+
+func (i VatHealTransformerInitializer) NewVatHealTransformer(db *postgres.DB, blockChain core.BlockChain) shared.Transformer {
+	fetcher := shared.NewFetcher(blockChain)
+	repository := NewVatHealRepository(db)
+	transformer := VatHealTransformer{
+		Fetcher:    fetcher,
+		Repository: repository,
+		Converter:  VatHealConverter{},
+		Config:     i.Config,
+	}
+
+	return transformer
+}
+
+func (transformer VatHealTransformer) Execute() error {
+	config := transformer.Config
+	topics := [][]common.Hash{{common.HexToHash(config.Topics[0])}}
+	headers, err := transformer.Repository.MissingHeaders(config.StartingBlockNumber, config.EndingBlockNumber)
+	if err != nil {
+		return err
+	}
+
+	for _, header := range headers {
+		logs, err := transformer.Fetcher.FetchLogs(config.ContractAddresses, topics, header.BlockNumber)
+		if err != nil {
+			return err
+		}
+
+		if len(logs) < 1 {
+			err = transformer.Repository.MarkCheckedHeader(header.Id)
+		}
+		if err != nil {
+			return err
+		}
+
+		models, err := transformer.Converter.ToModels(logs)
+		if err != nil {
+			return err
+		}
+
+		err = transformer.Repository.Create(header.Id, models)
+		if err != nil {
+			return err
+		}
+	}
+	return nil
+}

pkg/transformers/vat_heal/transformer_test.go

@@ -0,0 +1,163 @@
+// 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 vat_heal_test
+
+import (
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/core/types"
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+	"github.com/vulcanize/vulcanizedb/pkg/core"
+	"github.com/vulcanize/vulcanizedb/pkg/fakes"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/shared"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/test_data"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/test_data/mocks"
+	mock_vat_heal "github.com/vulcanize/vulcanizedb/pkg/transformers/test_data/mocks/vat_heal"
+	"github.com/vulcanize/vulcanizedb/pkg/transformers/vat_heal"
+)
+
+var _ = Describe("VatHeal Transformer", func() {
+	var mockRepository mock_vat_heal.MockVatHealRepository
+	var transformer vat_heal.VatHealTransformer
+	var fetcher mocks.MockLogFetcher
+	var converter mock_vat_heal.MockVatHealConverter
+
+	BeforeEach(func() {
+		mockRepository = mock_vat_heal.MockVatHealRepository{}
+		fetcher = mocks.MockLogFetcher{}
+		converter = mock_vat_heal.MockVatHealConverter{}
+		transformer = vat_heal.VatHealTransformer{
+			Repository: &mockRepository,
+			Config:     vat_heal.VatHealConfig,
+			Fetcher:    &fetcher,
+			Converter:  &converter,
+		}
+	})
+
+	It("gets all of the missing header ids", func() {
+		err := transformer.Execute()
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(mockRepository.PassedStartingBlockNumber).To(Equal(vat_heal.VatHealConfig.StartingBlockNumber))
+		Expect(mockRepository.PassedEndingBlockNumber).To(Equal(vat_heal.VatHealConfig.EndingBlockNumber))
+	})
+
+	It("returns and error if getting the missing headers fails", func() {
+		mockRepository.SetMissingHeadersErr(fakes.FakeError)
+		err := transformer.Execute()
+
+		Expect(err).To(HaveOccurred())
+		Expect(err).To(MatchError(fakes.FakeError))
+	})
+
+	It("fetches vat heal logs for the headers", func() {
+		header := core.Header{BlockNumber: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		err := transformer.Execute()
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(fetcher.FetchedContractAddresses[0]).To(Equal(vat_heal.VatHealConfig.ContractAddresses))
+		Expect(fetcher.FetchedTopics).To(Equal([][]common.Hash{{common.HexToHash(shared.VatHealSignature)}}))
+		Expect(fetcher.FetchedBlocks).To(Equal([]int64{header.BlockNumber}))
+	})
+
+	It("returns and error if fetching the logs fails", func() {
+		header := core.Header{BlockNumber: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		fetcher.SetFetcherError(fakes.FakeError)
+		err := transformer.Execute()
+
+		Expect(err).To(HaveOccurred())
+		Expect(err).To(MatchError(fakes.FakeError))
+	})
+
+	It("converts the logs to models", func() {
+		header := core.Header{BlockNumber: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		fetcher.SetFetchedLogs([]types.Log{test_data.VatHealLog})
+
+		err := transformer.Execute()
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(converter.PassedLogs).To(Equal([]types.Log{test_data.VatHealLog}))
+	})
+
+	It("returns an error if converting fails", func() {
+		header := core.Header{BlockNumber: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		fetcher.SetFetchedLogs([]types.Log{test_data.VatHealLog})
+		converter.SetConverterError(fakes.FakeError)
+
+		err := transformer.Execute()
+
+		Expect(err).To(HaveOccurred())
+		Expect(err).To(MatchError(fakes.FakeError))
+	})
+
+	It("persists the vat heal models", func() {
+		header := core.Header{Id: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		fetcher.SetFetchedLogs([]types.Log{test_data.VatHealLog})
+
+		err := transformer.Execute()
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(mockRepository.PassedModels).To(ContainElement(test_data.VatHealModel))
+		Expect(mockRepository.PassedHeaderID).To(Equal(header.Id))
+	})
+
+	It("returns an error if persisting the vat heal models fails", func() {
+		header := core.Header{Id: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		fetcher.SetFetchedLogs([]types.Log{test_data.VatHealLog})
+		mockRepository.SetCreateError(fakes.FakeError)
+
+		err := transformer.Execute()
+
+		Expect(err).To(HaveOccurred())
+		Expect(err).To(MatchError(fakes.FakeError))
+	})
+
+	It("marks the header as checked when there are no logs", func() {
+		header := core.Header{Id: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+
+		err := transformer.Execute()
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(mockRepository.MarkHeaderCheckedPassedHeaderID).To(Equal(header.Id))
+	})
+
+	It("doesn't call MarkCheckedHeader when there are logs", func() {
+		header := core.Header{Id: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		fetcher.SetFetchedLogs([]types.Log{test_data.VatHealLog})
+		err := transformer.Execute()
+
+		Expect(err).NotTo(HaveOccurred())
+		Expect(mockRepository.MarkHeaderCheckedPassedHeaderID).To(Equal(int64(0)))
+	})
+
+	It("returns an error if MarkCheckedHeader fails", func() {
+		header := core.Header{Id: GinkgoRandomSeed()}
+		mockRepository.SetMissingHeaders([]core.Header{header})
+		mockRepository.SetMissingHeadersErr(fakes.FakeError)
+
+		err := transformer.Execute()
+
+		Expect(err).To(HaveOccurred())
+		Expect(err).To(MatchError(fakes.FakeError))
+	})
+})

pkg/transformers/vat_heal/vat_heal_suite_test.go

@@ -0,0 +1,13 @@
+package vat_heal_test
+
+import (
+	"testing"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+func TestVatHeal(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "VatHeal Suite")
+}