forked from cerc-io/ipld-eth-server
Convert raw header to JSON
- Enables parsing values off of header via Postgres migration
This commit is contained in:
Rob Mulholand
@@ -15,8 +15,9 @@
|
||||
package dent_test
|
||||
|
||||
import (
|
||||
"math/rand"
|
||||
"encoding/json"
|
||||
|
||||
"github.com/ethereum/go-ethereum/core/types"
|
||||
. "github.com/onsi/ginkgo"
|
||||
. "github.com/onsi/gomega"
|
||||
|
||||
@@ -29,24 +30,28 @@ import (
|
||||
)
|
||||
|
||||
var _ = Describe("Dent Repository", func() {
|
||||
var node core.Node
|
||||
var db *postgres.DB
|
||||
var dentRepository dent.DentRepository
|
||||
var headerRepository repositories.HeaderRepository
|
||||
var headerId int64
|
||||
var err error
|
||||
var (
|
||||
db *postgres.DB
|
||||
dentRepository dent.DentRepository
|
||||
headerRepository repositories.HeaderRepository
|
||||
err error
|
||||
rawHeader []byte
|
||||
)
|
||||
|
||||
BeforeEach(func() {
|
||||
node = test_config.NewTestNode()
|
||||
db = test_config.NewTestDB(node)
|
||||
db = test_config.NewTestDB(test_config.NewTestNode())
|
||||
test_config.CleanTestDB(db)
|
||||
dentRepository = dent.NewDentRepository(db)
|
||||
headerRepository = repositories.NewHeaderRepository(db)
|
||||
rawHeader, err = json.Marshal(types.Header{})
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
})
|
||||
|
||||
Describe("Create", func() {
|
||||
var headerId int64
|
||||
|
||||
BeforeEach(func() {
|
||||
headerId, err = headerRepository.CreateOrUpdateHeader(core.Header{})
|
||||
headerId, err = headerRepository.CreateOrUpdateHeader(core.Header{Raw: rawHeader})
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
|
||||
err := dentRepository.Create(headerId, []dent.DentModel{test_data.DentModel})
|
||||
@@ -99,8 +104,10 @@ var _ = Describe("Dent Repository", func() {
|
||||
})
|
||||
|
||||
Describe("MarkHeaderChecked", func() {
|
||||
var headerId int64
|
||||
|
||||
BeforeEach(func() {
|
||||
headerId, err = headerRepository.CreateOrUpdateHeader(core.Header{})
|
||||
headerId, err = headerRepository.CreateOrUpdateHeader(core.Header{Raw: rawHeader})
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
})
|
||||
|
||||
@@ -129,85 +136,75 @@ var _ = Describe("Dent Repository", func() {
|
||||
})
|
||||
|
||||
Describe("MissingHeaders", func() {
|
||||
var dentBlockNumber int64
|
||||
var startingBlockNumber int64
|
||||
var endingBlockNumber int64
|
||||
var blockNumbers []int64
|
||||
var (
|
||||
dentBlock, startingBlock, endingBlock int64
|
||||
blockNumbers, headerIds []int64
|
||||
)
|
||||
|
||||
BeforeEach(func() {
|
||||
dentBlockNumber = rand.Int63()
|
||||
startingBlockNumber = dentBlockNumber - 1
|
||||
endingBlockNumber = dentBlockNumber + 1
|
||||
outOfRangeBlockNumber := dentBlockNumber + 2
|
||||
dentBlock = GinkgoRandomSeed()
|
||||
startingBlock = dentBlock - 1
|
||||
endingBlock = dentBlock + 1
|
||||
outOfRangeBlockNumber := dentBlock + 2
|
||||
|
||||
blockNumbers = []int64{startingBlockNumber, dentBlockNumber, endingBlockNumber, outOfRangeBlockNumber}
|
||||
blockNumbers = []int64{startingBlock, dentBlock, endingBlock, outOfRangeBlockNumber}
|
||||
|
||||
var headerIds []int64
|
||||
headerIds = []int64{}
|
||||
for _, number := range blockNumbers {
|
||||
headerId, err := headerRepository.CreateOrUpdateHeader(core.Header{BlockNumber: number})
|
||||
headerId, err := headerRepository.CreateOrUpdateHeader(core.Header{BlockNumber: number, Raw: rawHeader})
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
headerIds = append(headerIds, headerId)
|
||||
}
|
||||
|
||||
dentRepository.MarkHeaderChecked(headerIds[1])
|
||||
})
|
||||
|
||||
It("returns header records that haven't been checked", func() {
|
||||
missingHeaders, err := dentRepository.MissingHeaders(startingBlockNumber, endingBlockNumber)
|
||||
dentRepository.MarkHeaderChecked(headerIds[1])
|
||||
missingHeaders, err := dentRepository.MissingHeaders(startingBlock, endingBlock)
|
||||
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(len(missingHeaders)).To(Equal(2))
|
||||
Expect(missingHeaders[0].BlockNumber).To(Equal(startingBlockNumber))
|
||||
Expect(missingHeaders[1].BlockNumber).To(Equal(endingBlockNumber))
|
||||
Expect(missingHeaders[0].BlockNumber).To(Equal(startingBlock))
|
||||
Expect(missingHeaders[1].BlockNumber).To(Equal(endingBlock))
|
||||
})
|
||||
|
||||
It("only treats headers as checked if deal have been checked", func() {
|
||||
startingBlockNumber := int64(1)
|
||||
dentBlockNumber := int64(2)
|
||||
endingBlockNumber := int64(3)
|
||||
blockNumbers := []int64{startingBlockNumber, dentBlockNumber, endingBlockNumber, endingBlockNumber + 1}
|
||||
var headerIDs []int64
|
||||
for _, n := range blockNumbers {
|
||||
headerID, err := headerRepository.CreateOrUpdateHeader(core.Header{BlockNumber: n})
|
||||
headerIDs = append(headerIDs, headerID)
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
}
|
||||
_, err := db.Exec(`INSERT INTO public.checked_headers (header_id, price_feeds_checked) VALUES ($1, $2)`, headerIDs[1], true)
|
||||
It("only treats headers as checked if dent has been checked", func() {
|
||||
_, err := db.Exec(`INSERT INTO public.checked_headers (header_id) VALUES ($1)`, headerIds[1])
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
|
||||
headers, err := dentRepository.MissingHeaders(startingBlockNumber, endingBlockNumber)
|
||||
headers, err := dentRepository.MissingHeaders(startingBlock, endingBlock)
|
||||
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(len(headers)).To(Equal(3))
|
||||
Expect(headers[0].BlockNumber).To(Or(Equal(startingBlockNumber), Equal(endingBlockNumber), Equal(dentBlockNumber)))
|
||||
Expect(headers[1].BlockNumber).To(Or(Equal(startingBlockNumber), Equal(endingBlockNumber), Equal(dentBlockNumber)))
|
||||
Expect(headers[2].BlockNumber).To(Or(Equal(startingBlockNumber), Equal(endingBlockNumber), Equal(dentBlockNumber)))
|
||||
Expect(headers[0].BlockNumber).To(Or(Equal(startingBlock), Equal(endingBlock), Equal(dentBlock)))
|
||||
Expect(headers[1].BlockNumber).To(Or(Equal(startingBlock), Equal(endingBlock), Equal(dentBlock)))
|
||||
Expect(headers[2].BlockNumber).To(Or(Equal(startingBlock), Equal(endingBlock), Equal(dentBlock)))
|
||||
})
|
||||
|
||||
It("only returns missing headers for the given node", func() {
|
||||
dentRepository.MarkHeaderChecked(headerIds[1])
|
||||
node2 := core.Node{}
|
||||
db2 := test_config.NewTestDB(node2)
|
||||
dentRepository2 := dent.NewDentRepository(db2)
|
||||
headerRepository2 := repositories.NewHeaderRepository(db2)
|
||||
var node2HeaderIds []int64
|
||||
for _, number := range blockNumbers {
|
||||
id, err := headerRepository2.CreateOrUpdateHeader(core.Header{BlockNumber: number})
|
||||
id, err := headerRepository2.CreateOrUpdateHeader(core.Header{BlockNumber: number, Raw: rawHeader})
|
||||
node2HeaderIds = append(node2HeaderIds, id)
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
}
|
||||
|
||||
missingHeadersNode1, err := dentRepository.MissingHeaders(startingBlockNumber, endingBlockNumber)
|
||||
missingHeadersNode1, err := dentRepository.MissingHeaders(startingBlock, endingBlock)
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(len(missingHeadersNode1)).To(Equal(2))
|
||||
Expect(missingHeadersNode1[0].BlockNumber).To(Equal(startingBlockNumber))
|
||||
Expect(missingHeadersNode1[1].BlockNumber).To(Equal(endingBlockNumber))
|
||||
Expect(missingHeadersNode1[0].BlockNumber).To(Equal(startingBlock))
|
||||
Expect(missingHeadersNode1[1].BlockNumber).To(Equal(endingBlock))
|
||||
|
||||
missingHeadersNode2, err := dentRepository2.MissingHeaders(startingBlockNumber, endingBlockNumber)
|
||||
missingHeadersNode2, err := dentRepository2.MissingHeaders(startingBlock, endingBlock)
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(len(missingHeadersNode2)).To(Equal(3))
|
||||
Expect(missingHeadersNode2[0].BlockNumber).To(Equal(startingBlockNumber))
|
||||
Expect(missingHeadersNode2[1].BlockNumber).To(Equal(dentBlockNumber))
|
||||
Expect(missingHeadersNode2[2].BlockNumber).To(Equal(endingBlockNumber))
|
||||
Expect(missingHeadersNode2[0].BlockNumber).To(Or(Equal(startingBlock), Equal(dentBlock), Equal(endingBlock)))
|
||||
Expect(missingHeadersNode2[1].BlockNumber).To(Or(Equal(startingBlock), Equal(dentBlock), Equal(endingBlock)))
|
||||
Expect(missingHeadersNode2[2].BlockNumber).To(Or(Equal(startingBlock), Equal(dentBlock), Equal(endingBlock)))
|
||||
})
|
||||
})
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user