diff --git a/pkg/transformers/test_data/mocks/vat_move/repository.go b/pkg/transformers/test_data/mocks/vat_move/repository.go index 3b40753f..46c3eb02 100644 --- a/pkg/transformers/test_data/mocks/vat_move/repository.go +++ b/pkg/transformers/test_data/mocks/vat_move/repository.go @@ -64,6 +64,4 @@ func (repository *MockVatMoveRepository) SetCheckedHeaderError(e error) { repository.CheckedHeaderError = e } -func (repository *MockVatMoveRepository) SetDB(db *postgres.DB) { - // TODO Implement passed DB -} \ No newline at end of file +func (repository *MockVatMoveRepository) SetDB(db *postgres.DB) {} diff --git a/pkg/transformers/vat_move/config.go b/pkg/transformers/vat_move/config.go index 32077783..a98bc52a 100644 --- a/pkg/transformers/vat_move/config.go +++ b/pkg/transformers/vat_move/config.go @@ -19,6 +19,7 @@ import ( ) var VatMoveConfig = shared.TransformerConfig{ + TransformerName: "vat move", ContractAddresses: []string{shared.VatContractAddress}, ContractAbi: shared.VatABI, Topics: []string{shared.VatMoveSignature}, diff --git a/pkg/transformers/vat_move/repository_test.go b/pkg/transformers/vat_move/repository_test.go index 978de862..986cc4d1 100644 --- a/pkg/transformers/vat_move/repository_test.go +++ b/pkg/transformers/vat_move/repository_test.go @@ -190,4 +190,13 @@ var _ = Describe("Vat Move", func() { Expect(headerChecked).To(BeTrue()) }) }) + + Describe("SetDB", func() { + It("sets the repository db", func() { + vatMoveRepository.DB = nil + Expect(vatMoveRepository.DB).To(BeNil()) + vatMoveRepository.SetDB(db) + Expect(vatMoveRepository.DB).To(Equal(db)) + }) + }) }) diff --git a/pkg/transformers/vat_move/transformer_test.go b/pkg/transformers/vat_move/transformer_test.go index 9b63cc06..e49dacc0 100644 --- a/pkg/transformers/vat_move/transformer_test.go +++ b/pkg/transformers/vat_move/transformer_test.go @@ -47,10 +47,10 @@ var _ = Describe("Vat move transformer", func() { It("gets missing headers for block numbers specified in config", func() { transformer := factories.Transformer{ - Config: vat_move.VatMoveConfig, + Config: config, Converter: &converter, Repository: &repository, - Fetcher: &fetcher, + Fetcher: &fetcher, }.NewTransformer(nil, nil) err := transformer.Execute()