ipld-eth-server/pkg/history/populate_blocks_test.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 (Number 1)", func() {
		blocks := []core.Block{
			{Number: 1},
			{Number: 2},
		}
		blockchain := fakes.NewBlockchainWithBlocks(blocks)
		repository := repositories.NewInMemory()
		repository.CreateOrUpdateBlock(core.Block{Number: 2})

		blocksAdded := history.PopulateMissingBlocks(blockchain, repository, 1)
		_, err := repository.FindBlockByNumber(1)

		Expect(blocksAdded).To(Equal(1))
		Expect(err).ToNot(HaveOccurred())
	})

	It("fills in the three missing blocks (Numbers: 5,8,10)", func() {
		blockchain := fakes.NewBlockchainWithBlocks([]core.Block{
			{Number: 4},
			{Number: 5},
			{Number: 6},
			{Number: 7},
			{Number: 8},
			{Number: 9},
			{Number: 10},
			{Number: 11},
			{Number: 12},
			{Number: 13},
		})
		repository := repositories.NewInMemory()
		repository.CreateOrUpdateBlock(core.Block{Number: 1})
		repository.CreateOrUpdateBlock(core.Block{Number: 2})
		repository.CreateOrUpdateBlock(core.Block{Number: 3})
		repository.CreateOrUpdateBlock(core.Block{Number: 6})
		repository.CreateOrUpdateBlock(core.Block{Number: 7})
		repository.CreateOrUpdateBlock(core.Block{Number: 9})
		repository.CreateOrUpdateBlock(core.Block{Number: 11})
		repository.CreateOrUpdateBlock(core.Block{Number: 12})

		blocksAdded := history.PopulateMissingBlocks(blockchain, repository, 5)

		Expect(blocksAdded).To(Equal(3))
		Expect(repository.BlockCount()).To(Equal(11))
		_, err := repository.FindBlockByNumber(4)
		Expect(err).To(HaveOccurred())
		_, err = repository.FindBlockByNumber(5)
		Expect(err).ToNot(HaveOccurred())
		_, err = repository.FindBlockByNumber(8)
		Expect(err).ToNot(HaveOccurred())
		_, err = repository.FindBlockByNumber(10)
		Expect(err).ToNot(HaveOccurred())
		_, err = repository.FindBlockByNumber(13)
		Expect(err).To(HaveOccurred())
	})

	It("updates the repository with a range of blocks w/in sliding window ", func() {
		blockchain := fakes.NewBlockchainWithBlocks([]core.Block{
			{Number: 1},
			{Number: 2},
			{Number: 3},
			{Number: 4},
			{Number: 5},
		})
		repository := repositories.NewInMemory()
		repository.CreateOrUpdateBlock(blockchain.GetBlockByNumber(5))

		history.UpdateBlocksWindow(blockchain, repository, 2)

		Expect(repository.BlockCount()).To(Equal(3))
		Expect(repository.HandleBlockCallCount).To(Equal(3))
	})

	It("Generates a range of int64", func() {
		numberOfBlocksCreated := history.MakeRange(0, 5)
		expected := []int64{0, 1, 2, 3, 4}

		Expect(numberOfBlocksCreated).To(Equal(expected))
	})

	It("returns the number of blocks created", func() {
		blockchain := fakes.NewBlockchainWithBlocks([]core.Block{
			{Number: 4},
			{Number: 5},
			{Number: 6},
		})
		repository := repositories.NewInMemory()
		repository.CreateOrUpdateBlock(core.Block{Number: 3})
		repository.CreateOrUpdateBlock(core.Block{Number: 6})

		numberOfBlocksCreated := history.PopulateMissingBlocks(blockchain, repository, 3)

		Expect(numberOfBlocksCreated).To(Equal(2))
	})

	It("returns the window size", func() {
		window := history.Window{1, 3, 10}
		Expect(window.Size()).To(Equal(2))
	})

	It("returns the number of largest block", func() {
		blockchain := fakes.NewBlockchainWithBlocks([]core.Block{
			{Number: 1},
			{Number: 2},
			{Number: 3},
		})
		maxBlockNumber := blockchain.LastBlock()

		Expect(maxBlockNumber.Int64()).To(Equal(int64(3)))
	})
})