45 lines
1.5 KiB
Go
45 lines
1.5 KiB
Go
|
package pep_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
. "github.com/onsi/ginkgo"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/core/types"
|
||
|
|
"github.com/vulcanize/vulcanizedb/pkg/core"
|
||
|
|
"github.com/vulcanize/vulcanizedb/pkg/datastore/postgres/repositories"
|
||
|
|
"github.com/vulcanize/vulcanizedb/pkg/fakes"
|
||
|
|
"github.com/vulcanize/vulcanizedb/pkg/transformers/pep"
|
||
|
|
"github.com/vulcanize/vulcanizedb/test_config"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ = Describe("Pep transformer", func() {
|
||
|
|
It("returns nil if no logs found", func() {
|
||
|
|
chain := fakes.NewMockBlockChain()
|
||
|
|
db := test_config.NewTestDB(core.Node{})
|
||
|
|
transformer := pep.NewPepTransformer(chain, db)
|
||
|
|
|
||
|
|
err := transformer.Execute(core.Header{}, 123)
|
||
|
|
|
||
|
|
Expect(err).NotTo(HaveOccurred())
|
||
|
|
})
|
||
|
|
|
||
|
|
It("creates pep row for found log", func() {
|
||
|
|
chain := fakes.NewMockBlockChain()
|
||
|
|
chain.SetGetEthLogsWithCustomQueryReturnLogs([]types.Log{{Data: []byte{1, 2, 3, 4, 5}}})
|
||
|
|
db := test_config.NewTestDB(core.Node{})
|
||
|
|
headerRepository := repositories.NewHeaderRepository(db)
|
||
|
|
header := core.Header{BlockNumber: 12345}
|
||
|
|
headerID, err := headerRepository.CreateOrUpdateHeader(header)
|
||
|
|
Expect(err).NotTo(HaveOccurred())
|
||
|
|
transformer := pep.NewPepTransformer(chain, db)
|
||
|
|
|
||
|
|
err = transformer.Execute(header, headerID)
|
||
|
|
|
||
|
|
Expect(err).NotTo(HaveOccurred())
|
||
|
|
var dbPep pep.Pep
|
||
|
|
err = db.Get(&dbPep, `SELECT block_number, header_id, usd_value FROM maker.peps WHERE header_id = $1`, headerID)
|
||
|
|
Expect(err).NotTo(HaveOccurred())
|
||
|
|
Expect(dbPep.BlockNumber).To(Equal(header.BlockNumber))
|
||
|
|
})
|
||
|
|
})
|