116 lines
4.5 KiB
Go
116 lines
4.5 KiB
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 eth_test
|
|
|
|
import (
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/shared"
|
|
|
|
"github.com/vulcanize/vulcanizedb/pkg/eth/datastore/postgres"
|
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/eth"
|
|
eth2 "github.com/vulcanize/vulcanizedb/pkg/super_node/eth"
|
|
"github.com/vulcanize/vulcanizedb/pkg/super_node/eth/mocks"
|
|
)
|
|
|
|
var _ = Describe("Indexer", func() {
|
|
var (
|
|
db *postgres.DB
|
|
err error
|
|
repo *eth2.CIDIndexer
|
|
)
|
|
BeforeEach(func() {
|
|
db, err = eth.SetupDB()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
repo = eth2.NewCIDIndexer(db)
|
|
})
|
|
AfterEach(func() {
|
|
eth.TearDownDB(db)
|
|
})
|
|
|
|
Describe("Index", func() {
|
|
It("Indexes CIDs and related metadata into vulcanizedb", func() {
|
|
err = repo.Index(mocks.MockCIDPayload)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
pgStr := `SELECT cid, td FROM eth.header_cids
|
|
WHERE block_number = $1`
|
|
// check header was properly indexed
|
|
type res struct {
|
|
CID string
|
|
TD string
|
|
}
|
|
headers := new(res)
|
|
err = db.QueryRowx(pgStr, 1).StructScan(headers)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(headers.CID).To(Equal("mockHeaderCID"))
|
|
Expect(headers.TD).To(Equal("1337"))
|
|
// check trxs were properly indexed
|
|
trxs := make([]string, 0)
|
|
pgStr = `SELECT transaction_cids.cid FROM eth.transaction_cids INNER JOIN eth.header_cids ON (transaction_cids.header_id = header_cids.id)
|
|
WHERE header_cids.block_number = $1`
|
|
err = db.Select(&trxs, pgStr, 1)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(len(trxs)).To(Equal(2))
|
|
Expect(shared.ListContainsString(trxs, "mockTrxCID1")).To(BeTrue())
|
|
Expect(shared.ListContainsString(trxs, "mockTrxCID2")).To(BeTrue())
|
|
// check receipts were properly indexed
|
|
rcts := make([]string, 0)
|
|
pgStr = `SELECT receipt_cids.cid FROM eth.receipt_cids, eth.transaction_cids, eth.header_cids
|
|
WHERE receipt_cids.tx_id = transaction_cids.id
|
|
AND transaction_cids.header_id = header_cids.id
|
|
AND header_cids.block_number = $1`
|
|
err = db.Select(&rcts, pgStr, 1)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(len(rcts)).To(Equal(2))
|
|
Expect(shared.ListContainsString(rcts, "mockRctCID1")).To(BeTrue())
|
|
Expect(shared.ListContainsString(rcts, "mockRctCID2")).To(BeTrue())
|
|
// check that state nodes were properly indexed
|
|
stateNodes := make([]eth.StateNodeModel, 0)
|
|
pgStr = `SELECT state_cids.cid, state_cids.state_key, state_cids.leaf FROM eth.state_cids INNER JOIN eth.header_cids ON (state_cids.header_id = header_cids.id)
|
|
WHERE header_cids.block_number = $1`
|
|
err = db.Select(&stateNodes, pgStr, 1)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(len(stateNodes)).To(Equal(2))
|
|
for _, stateNode := range stateNodes {
|
|
if stateNode.CID == "mockStateCID1" {
|
|
Expect(stateNode.Leaf).To(Equal(true))
|
|
Expect(stateNode.StateKey).To(Equal(mocks.ContractLeafKey.Hex()))
|
|
}
|
|
if stateNode.CID == "mockStateCID2" {
|
|
Expect(stateNode.Leaf).To(Equal(true))
|
|
Expect(stateNode.StateKey).To(Equal(mocks.AnotherContractLeafKey.Hex()))
|
|
}
|
|
}
|
|
// check that storage nodes were properly indexed
|
|
storageNodes := make([]eth.StorageNodeWithStateKeyModel, 0)
|
|
pgStr = `SELECT storage_cids.cid, state_cids.state_key, storage_cids.storage_key, storage_cids.leaf FROM eth.storage_cids, eth.state_cids, eth.header_cids
|
|
WHERE storage_cids.state_id = state_cids.id
|
|
AND state_cids.header_id = header_cids.id
|
|
AND header_cids.block_number = $1`
|
|
err = db.Select(&storageNodes, pgStr, 1)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(len(storageNodes)).To(Equal(1))
|
|
Expect(storageNodes[0]).To(Equal(eth.StorageNodeWithStateKeyModel{
|
|
CID: "mockStorageCID",
|
|
Leaf: true,
|
|
StorageKey: "0x0000000000000000000000000000000000000000000000000000000000000001",
|
|
StateKey: mocks.ContractLeafKey.Hex(),
|
|
}))
|
|
})
|
|
})
|
|
})
|