// 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 . package transformer_test import ( "fmt" "math/rand" "time" "github.com/ethereum/go-ethereum/common" . "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/omni/full/transformer" "github.com/vulcanize/vulcanizedb/pkg/omni/shared/constants" "github.com/vulcanize/vulcanizedb/pkg/omni/shared/helpers/test_helpers" "github.com/vulcanize/vulcanizedb/pkg/omni/shared/helpers/test_helpers/mocks" ) var _ = Describe("Transformer", func() { var db *postgres.DB var err error var blockChain core.BlockChain var blockRepository repositories.BlockRepository rand.Seed(time.Now().UnixNano()) BeforeEach(func() { db, blockChain = test_helpers.SetupDBandBC() blockRepository = *repositories.NewBlockRepository(db) }) AfterEach(func() { test_helpers.TearDown(db) }) Describe("SetEvents", func() { It("Sets which events to watch from the given contract address", func() { watchedEvents := []string{"Transfer", "Mint"} t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, watchedEvents) Expect(t.WatchedEvents[constants.TusdContractAddress]).To(Equal(watchedEvents)) }) }) Describe("SetEventAddrs", func() { It("Sets which account addresses to watch events for", func() { eventAddrs := []string{"test1", "test2"} t := transformer.NewTransformer("", blockChain, db) t.SetEventArgs(constants.TusdContractAddress, eventAddrs) Expect(t.EventArgs[constants.TusdContractAddress]).To(Equal(eventAddrs)) }) }) Describe("SetMethods", func() { It("Sets which methods to poll at the given contract address", func() { watchedMethods := []string{"balanceOf", "totalSupply"} t := transformer.NewTransformer("", blockChain, db) t.SetMethods(constants.TusdContractAddress, watchedMethods) Expect(t.WantedMethods[constants.TusdContractAddress]).To(Equal(watchedMethods)) }) }) Describe("SetMethodAddrs", func() { It("Sets which account addresses to poll methods against", func() { methodAddrs := []string{"test1", "test2"} t := transformer.NewTransformer("", blockChain, db) t.SetMethodArgs(constants.TusdContractAddress, methodAddrs) Expect(t.MethodArgs[constants.TusdContractAddress]).To(Equal(methodAddrs)) }) }) Describe("SetRange", func() { It("Sets the block range that the contract should be watched within", func() { rng := [2]int64{1, 100000} t := transformer.NewTransformer("", blockChain, db) t.SetRange(constants.TusdContractAddress, rng) Expect(t.ContractRanges[constants.TusdContractAddress]).To(Equal(rng)) }) }) Describe("Init", func() { It("Initializes transformer's contract objects", func() { blockRepository.CreateOrUpdateBlock(mocks.TransferBlock1) blockRepository.CreateOrUpdateBlock(mocks.TransferBlock2) t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, []string{"Transfer"}) err = t.Init() Expect(err).ToNot(HaveOccurred()) c, ok := t.Contracts[constants.TusdContractAddress] Expect(ok).To(Equal(true)) Expect(c.StartingBlock).To(Equal(int64(6194633))) Expect(c.LastBlock).To(Equal(int64(6194634))) Expect(c.Abi).To(Equal(constants.TusdAbiString)) Expect(c.Name).To(Equal("TrueUSD")) Expect(c.Address).To(Equal(constants.TusdContractAddress)) }) It("Fails to initialize if first and most recent blocks cannot be fetched from vDB", func() { t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, []string{"Transfer"}) err = t.Init() Expect(err).To(HaveOccurred()) }) It("Does nothing if watched events are unset", func() { blockRepository.CreateOrUpdateBlock(mocks.TransferBlock1) blockRepository.CreateOrUpdateBlock(mocks.TransferBlock2) t := transformer.NewTransformer("", blockChain, db) err = t.Init() Expect(err).ToNot(HaveOccurred()) _, ok := t.Contracts[constants.TusdContractAddress] Expect(ok).To(Equal(false)) }) }) Describe("Execute", func() { BeforeEach(func() { blockRepository.CreateOrUpdateBlock(mocks.TransferBlock1) blockRepository.CreateOrUpdateBlock(mocks.TransferBlock2) }) It("Transforms watched contract data into custom repositories", func() { t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, []string{"Transfer"}) t.SetMethods(constants.TusdContractAddress, nil) err = t.Init() Expect(err).ToNot(HaveOccurred()) err = t.Execute() Expect(err).ToNot(HaveOccurred()) log := test_helpers.TransferLog{} err = db.QueryRowx(fmt.Sprintf("SELECT * FROM full_%s.transfer_event WHERE block = 6194634", constants.TusdContractAddress)).StructScan(&log) // We don't know vulcID, so compare individual fields instead of complete structures Expect(log.Tx).To(Equal("0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad654eee")) Expect(log.Block).To(Equal(int64(6194634))) Expect(log.From).To(Equal("0x000000000000000000000000000000000000Af21")) Expect(log.To).To(Equal("0x09BbBBE21a5975cAc061D82f7b843bCE061BA391")) Expect(log.Value).To(Equal("1097077688018008265106216665536940668749033598146")) }) It("Keeps track of contract-related addresses while transforming event data if they need to be used for later method polling", func() { t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, []string{"Transfer"}) t.SetMethods(constants.TusdContractAddress, []string{"balanceOf"}) err = t.Init() Expect(err).ToNot(HaveOccurred()) c, ok := t.Contracts[constants.TusdContractAddress] Expect(ok).To(Equal(true)) err = t.Execute() Expect(err).ToNot(HaveOccurred()) b, ok := c.EmittedAddrs[common.HexToAddress("0x000000000000000000000000000000000000Af21")] Expect(ok).To(Equal(true)) Expect(b).To(Equal(true)) b, ok = c.EmittedAddrs[common.HexToAddress("0x09BbBBE21a5975cAc061D82f7b843bCE061BA391")] Expect(ok).To(Equal(true)) Expect(b).To(Equal(true)) _, ok = c.EmittedAddrs[common.HexToAddress("0x09BbBBE21a5975cAc061D82f7b843b1234567890")] Expect(ok).To(Equal(false)) _, ok = c.EmittedAddrs[common.HexToAddress("0x")] Expect(ok).To(Equal(false)) _, ok = c.EmittedAddrs[""] Expect(ok).To(Equal(false)) _, ok = c.EmittedAddrs[common.HexToAddress("0x09THISE21a5IS5cFAKE1D82fAND43bCE06MADEUP")] Expect(ok).To(Equal(false)) }) It("Polls given methods using generated token holder address", func() { t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, []string{"Transfer"}) t.SetMethods(constants.TusdContractAddress, []string{"balanceOf"}) err = t.Init() Expect(err).ToNot(HaveOccurred()) err = t.Execute() Expect(err).ToNot(HaveOccurred()) res := test_helpers.BalanceOf{} err = db.QueryRowx(fmt.Sprintf("SELECT * FROM full_%s.balanceof_method WHERE who_ = '0x09BbBBE21a5975cAc061D82f7b843bCE061BA391' AND block = '6194634'", constants.TusdContractAddress)).StructScan(&res) Expect(err).ToNot(HaveOccurred()) Expect(res.Balance).To(Equal("0")) Expect(res.TokenName).To(Equal("TrueUSD")) err = db.QueryRowx(fmt.Sprintf("SELECT * FROM full_%s.balanceof_method WHERE who_ = '0x09BbBBE21a5975cAc061D82f7b843bCE061BA391' AND block = '6194634'", constants.TusdContractAddress)).StructScan(&res) Expect(err).ToNot(HaveOccurred()) Expect(res.Balance).To(Equal("0")) Expect(res.TokenName).To(Equal("TrueUSD")) err = db.QueryRowx(fmt.Sprintf("SELECT * FROM full_%s.balanceof_method WHERE who_ = '0xfE9e8709d3215310075d67E3ed32A380CCf451C8' AND block = '6194634'", constants.TusdContractAddress)).StructScan(&res) Expect(err).To(HaveOccurred()) }) It("Fails if initialization has not been done", func() { t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, []string{"Transfer"}) t.SetMethods(constants.TusdContractAddress, nil) err = t.Execute() Expect(err).To(HaveOccurred()) }) }) })