// 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/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/constants" "github.com/vulcanize/vulcanizedb/pkg/omni/helpers/test_helpers" "github.com/vulcanize/vulcanizedb/pkg/omni/transformer" ) var block1 = core.Block{ Hash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad123ert", Number: 6194633, Transactions: []core.Transaction{{ Hash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad654aaa", Receipt: core.Receipt{ TxHash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad654aaa", ContractAddress: "", Logs: []core.Log{{ BlockNumber: 6194633, TxHash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad654aaa", Address: constants.TusdContractAddress, Topics: core.Topics{ constants.TransferEvent.Signature(), "0x000000000000000000000000000000000000000000000000000000000000af21", "0x9dd48110dcc444fdc242510c09bbbbe21a5975cac061d82f7b843bce061ba391", "", }, Index: 1, Data: "0x000000000000000000000000c02aaa39b223fe8d0a0e5c4f27ead9083c756cc200000000000000000000000089d24a6b4ccb1b6faa2625fe562bdd9a23260359000000000000000000000000000000000000000000000000392d2e2bda9c00000000000000000000000000000000000000000000000000927f41fa0a4a418000000000000000000000000000000000000000000000000000000000005adcfebe", }}, }, }}, } var block2 = core.Block{ Hash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad123ooo", Number: 6194634, Transactions: []core.Transaction{{ Hash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad654eee", Receipt: core.Receipt{ TxHash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad654eee", ContractAddress: "", Logs: []core.Log{{ BlockNumber: 6194634, TxHash: "0x135391a0962a63944e5908e6fedfff90fb4be3e3290a21017861099bad654eee", Address: constants.TusdContractAddress, Topics: core.Topics{ constants.TransferEvent.Signature(), "0x000000000000000000000000000000000000000000000000000000000000af21", "0x9dd48110dcc444fdc242510c09bbbbe21a5975cac061d82f7b843bce061ba391", "", }, Index: 1, Data: "0x000000000000000000000000c02aaa39b223fe8d0a0e5c4f27ead9083c756cc200000000000000000000000089d24a6b4ccb1b6faa2625fe562bdd9a23260359000000000000000000000000000000000000000000000000392d2e2bda9c00000000000000000000000000000000000000000000000000927f41fa0a4a418000000000000000000000000000000000000000000000000000000000005adcfebe", }}, }, }}, } 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.SetEventAddrs(constants.TusdContractAddress, eventAddrs) Expect(t.EventAddrs[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.SetMethodAddrs(constants.TusdContractAddress, methodAddrs) Expect(t.MethodAddrs[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(block1) blockRepository.CreateOrUpdateBlock(block2) 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 nil", func() { blockRepository.CreateOrUpdateBlock(block1) blockRepository.CreateOrUpdateBlock(block2) 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(block1) blockRepository.CreateOrUpdateBlock(block2) }) 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 c%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", func() { t := transformer.NewTransformer("", blockChain, db) t.SetEvents(constants.TusdContractAddress, []string{"Transfer"}) t.SetMethods(constants.TusdContractAddress, nil) 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.TknHolderAddrs["0x000000000000000000000000000000000000Af21"] Expect(ok).To(Equal(true)) Expect(b).To(Equal(true)) b, ok = c.TknHolderAddrs["0x09BbBBE21a5975cAc061D82f7b843bCE061BA391"] Expect(ok).To(Equal(true)) Expect(b).To(Equal(true)) _, ok = c.TknHolderAddrs["0x09BbBBE21a5975cAc061D82f7b843b1234567890"] Expect(ok).To(Equal(false)) _, ok = c.TknHolderAddrs["0x"] 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 c%s.balanceof_method WHERE who_ = '0x000000000000000000000000000000000000Af21' 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 c%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()) }) }) })