70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
|
package history_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"github.com/8thlight/vulcanizedb/pkg/core"
|
||
|
|
"github.com/8thlight/vulcanizedb/pkg/fakes"
|
||
|
|
"github.com/8thlight/vulcanizedb/pkg/history"
|
||
|
|
"github.com/8thlight/vulcanizedb/pkg/repositories"
|
||
|
|
. "github.com/onsi/ginkgo"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ = Describe("Populating blocks", func() {
|
||
|
|
|
||
|
|
It("fills in the only missing block", func() {
|
||
|
|
blocks := []core.Block{{Number: 1, Hash: "x012343"}}
|
||
|
|
blockchain := fakes.NewBlockchainWithBlocks(blocks)
|
||
|
|
repository := repositories.NewInMemory()
|
||
|
|
repository.CreateBlock(core.Block{Number: 2})
|
||
|
|
|
||
|
|
history.PopulateBlocks(blockchain, repository, 1)
|
||
|
|
|
||
|
|
block := repository.FindBlockByNumber(1)
|
||
|
|
Expect(block).NotTo(BeNil())
|
||
|
|
Expect(block.Hash).To(Equal("x012343"))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("fills in two missing blocks", func() {
|
||
|
|
blockchain := fakes.NewBlockchainWithBlocks([]core.Block{
|
||
|
|
{Number: 4},
|
||
|
|
{Number: 5},
|
||
|
|
{Number: 8},
|
||
|
|
{Number: 10},
|
||
|
|
{Number: 13},
|
||
|
|
})
|
||
|
|
repository := repositories.NewInMemory()
|
||
|
|
repository.CreateBlock(core.Block{Number: 1})
|
||
|
|
repository.CreateBlock(core.Block{Number: 2})
|
||
|
|
repository.CreateBlock(core.Block{Number: 3})
|
||
|
|
repository.CreateBlock(core.Block{Number: 6})
|
||
|
|
repository.CreateBlock(core.Block{Number: 7})
|
||
|
|
repository.CreateBlock(core.Block{Number: 9})
|
||
|
|
repository.CreateBlock(core.Block{Number: 11})
|
||
|
|
repository.CreateBlock(core.Block{Number: 12})
|
||
|
|
|
||
|
|
history.PopulateBlocks(blockchain, repository, 5)
|
||
|
|
|
||
|
|
Expect(repository.BlockCount()).To(Equal(11))
|
||
|
|
Expect(repository.FindBlockByNumber(4)).To(BeNil())
|
||
|
|
Expect(repository.FindBlockByNumber(5)).NotTo(BeNil())
|
||
|
|
Expect(repository.FindBlockByNumber(8)).NotTo(BeNil())
|
||
|
|
Expect(repository.FindBlockByNumber(10)).NotTo(BeNil())
|
||
|
|
Expect(repository.FindBlockByNumber(13)).To(BeNil())
|
||
|
|
})
|
||
|
|
|
||
|
|
It("returns the number of blocks created", func() {
|
||
|
|
blockchain := fakes.NewBlockchainWithBlocks([]core.Block{
|
||
|
|
{Number: 4},
|
||
|
|
{Number: 5},
|
||
|
|
})
|
||
|
|
repository := repositories.NewInMemory()
|
||
|
|
repository.CreateBlock(core.Block{Number: 3})
|
||
|
|
repository.CreateBlock(core.Block{Number: 6})
|
||
|
|
|
||
|
|
numberOfBlocksCreated := history.PopulateBlocks(blockchain, repository, 3)
|
||
|
|
|
||
|
|
Expect(numberOfBlocksCreated).To(Equal(2))
|
||
|
|
})
|
||
|
|
|
||
|
|
})
|