ipld-eth-server/pkg/eth/contract_watcher/full/converter/converter_test.go

// VulcanizeDB
// Copyright © 2019 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 <http://www.gnu.org/licenses/>.

package converter_test

import (
	"github.com/ethereum/go-ethereum/common"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"

	"github.com/vulcanize/vulcanizedb/pkg/eth/contract_watcher/full/converter"
	"github.com/vulcanize/vulcanizedb/pkg/eth/contract_watcher/shared/contract"
	"github.com/vulcanize/vulcanizedb/pkg/eth/contract_watcher/shared/helpers"
	"github.com/vulcanize/vulcanizedb/pkg/eth/contract_watcher/shared/helpers/test_helpers"
	"github.com/vulcanize/vulcanizedb/pkg/eth/contract_watcher/shared/helpers/test_helpers/mocks"
)

var _ = Describe("Converter", func() {
	var con *contract.Contract
	var wantedEvents = []string{"Transfer"}
	var err error

	BeforeEach(func() {
		con = test_helpers.SetupTusdContract(wantedEvents, []string{"balanceOf"})
	})

	Describe("Update", func() {
		It("Updates contract con held by the converter", func() {
			c := converter.Converter{}
			c.Update(con)
			Expect(c.ContractInfo).To(Equal(con))

			con := test_helpers.SetupTusdContract([]string{}, []string{})
			c.Update(con)
			Expect(c.ContractInfo).To(Equal(con))
		})
	})

	Describe("Convert", func() {
		It("Converts a watched event log to mapping of event input names to values", func() {
			_, ok := con.Events["Approval"]
			Expect(ok).To(Equal(false))

			event, ok := con.Events["Transfer"]
			Expect(ok).To(Equal(true))
			err = con.GenerateFilters()
			Expect(err).ToNot(HaveOccurred())

			c := converter.Converter{}
			c.Update(con)
			log, err := c.Convert(mocks.MockTranferEvent, event)
			Expect(err).ToNot(HaveOccurred())

			from := common.HexToAddress("0x000000000000000000000000000000000000000000000000000000000000af21")
			to := common.HexToAddress("0x9dd48110dcc444fdc242510c09bbbbe21a5975cac061d82f7b843bce061ba391")
			value := helpers.BigFromString("1097077688018008265106216665536940668749033598146")

			v := log.Values["value"]

			Expect(log.Values["to"]).To(Equal(to.String()))
			Expect(log.Values["from"]).To(Equal(from.String()))
			Expect(v).To(Equal(value.String()))
		})

		It("Keeps track of addresses it sees to grow a token holder address list for the contract", func() {
			event, ok := con.Events["Transfer"]
			Expect(ok).To(Equal(true))

			c := converter.Converter{}
			c.Update(con)
			_, err := c.Convert(mocks.MockTranferEvent, event)
			Expect(err).ToNot(HaveOccurred())

			b, ok := con.EmittedAddrs[common.HexToAddress("0x000000000000000000000000000000000000Af21")]
			Expect(ok).To(Equal(true))
			Expect(b).To(Equal(true))

			b, ok = con.EmittedAddrs[common.HexToAddress("0x09BbBBE21a5975cAc061D82f7b843bCE061BA391")]
			Expect(ok).To(Equal(true))
			Expect(b).To(Equal(true))

			_, ok = con.EmittedAddrs[common.HexToAddress("0x")]
			Expect(ok).To(Equal(false))

			_, ok = con.EmittedAddrs[""]
			Expect(ok).To(Equal(false))

			_, ok = con.EmittedAddrs[common.HexToAddress("0x09THISE21a5IS5cFAKE1D82fAND43bCE06MADEUP")]
			Expect(ok).To(Equal(false))
		})

		It("Fails with an empty contract", func() {
			event := con.Events["Transfer"]
			c := converter.Converter{}
			c.Update(&contract.Contract{})
			_, err = c.Convert(mocks.MockTranferEvent, event)
			Expect(err).To(HaveOccurred())
		})
	})
})