ipld-eth-beacon-indexer/internal/boot/boot_test.go

// VulcanizeDB
// Copyright © 2022 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 boot_test

import (
	"context"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
	"github.com/vulcanize/ipld-eth-beacon-indexer/internal/boot"
)

var _ = Describe("Boot", func() {
	var (
		dbAddress             string = "localhost"
		dbPort                int    = 8076
		dbName                string = "vulcanize_testing"
		dbUsername            string = "vdbm"
		dbPassword            string = "password"
		dbDriver              string = "PGX"
		bcAddress             string = "localhost"
		bcPort                int    = 5052
		bcConnectionProtocol  string = "http"
		bcType                string = "lighthouse"
		bcBootRetryInterval   int    = 1
		bcBootMaxRetry        int    = 5
		bcKgTableIncrement    int    = 10
		bcUniqueIdentifier    int    = 100
		bcCheckDb             bool   = false
		bcProcessBeaconBlocks bool   = true
		bcProcessBeaconState  bool   = true
	)
	Describe("Booting the application", Label("integration"), func() {
		Context("When the DB and BC are both up and running, we skip checking for a synced head, and we are processing head", func() {
			It("Should connect successfully", func() {
				_, db, err := boot.BootApplicationWithRetry(context.Background(), dbAddress, dbPort, dbName, dbUsername, dbPassword, dbDriver, bcAddress, bcPort, bcConnectionProtocol, bcType, bcBootRetryInterval, bcBootMaxRetry, bcKgTableIncrement, "head", true, bcUniqueIdentifier, bcCheckDb, bcProcessBeaconBlocks, bcProcessBeaconState)
				defer db.Close()
				Expect(err).ToNot(HaveOccurred())
			})
		})
		Context("When the DB and BC are both up and running, we skip checking for a synced head, and we are processing historic ", func() {
			It("Should connect successfully", func() {
				_, db, err := boot.BootApplicationWithRetry(context.Background(), dbAddress, dbPort, dbName, dbUsername, dbPassword, dbDriver, bcAddress, bcPort, bcConnectionProtocol, bcType, bcBootRetryInterval, bcBootMaxRetry, bcKgTableIncrement, "historic", true, bcUniqueIdentifier, bcCheckDb, bcProcessBeaconBlocks, bcProcessBeaconState)
				defer db.Close()
				Expect(err).ToNot(HaveOccurred())
			})
		})
		Context("When the DB and BC are both up and running, and we check for a synced head", func() {
			It("Should not connect successfully", func() {
				_, db, err := boot.BootApplicationWithRetry(context.Background(), dbAddress, dbPort, dbName, dbUsername, dbPassword, dbDriver, bcAddress, bcPort, bcConnectionProtocol, bcType, bcBootRetryInterval, bcBootMaxRetry, bcKgTableIncrement, "head", false, bcUniqueIdentifier, bcCheckDb, bcProcessBeaconBlocks, bcProcessBeaconState)
				defer db.Close()
				Expect(err).To(HaveOccurred())
			})
		})
		Context("When the DB and BC are both up and running, we skip checking for a synced head, but the unique identifier is 0", func() {
			It("Should not connect successfully", func() {
				_, db, err := boot.BootApplicationWithRetry(context.Background(), dbAddress, dbPort, dbName, dbUsername, dbPassword, dbDriver, bcAddress, bcPort, bcConnectionProtocol, bcType, bcBootRetryInterval, bcBootMaxRetry, bcKgTableIncrement, "head", false, 0, bcCheckDb, bcProcessBeaconBlocks, bcProcessBeaconState)
				defer db.Close()
				Expect(err).To(HaveOccurred())
			})
		})
		Context("When the DB is running but not the BC", func() {
			It("Should not connect successfully", func() {
				_, _, err := boot.BootApplication(context.Background(), dbAddress, dbPort, dbName, dbUsername, dbPassword, dbDriver, "hi", 100, bcConnectionProtocol, bcKgTableIncrement, true, bcUniqueIdentifier, bcCheckDb, bcProcessBeaconBlocks, bcProcessBeaconState)
				Expect(err).To(HaveOccurred())
			})
		})
		Context("When the BC is running but not the DB", func() {
			It("Should not connect successfully", func() {
				_, _, err := boot.BootApplication(context.Background(), "hi", 10, dbName, dbUsername, dbPassword, dbDriver, bcAddress, bcPort, bcConnectionProtocol, bcKgTableIncrement, true, bcUniqueIdentifier, bcCheckDb, bcProcessBeaconBlocks, bcProcessBeaconState)
				Expect(err).To(HaveOccurred())
			})
		})
		Context("When neither the BC or DB are running", func() {
			It("Should not connect successfully", func() {
				_, _, err := boot.BootApplication(context.Background(), "hi", 10, dbName, dbUsername, dbPassword, dbDriver, "hi", 100, bcConnectionProtocol, bcKgTableIncrement, true, bcUniqueIdentifier, bcCheckDb, bcProcessBeaconBlocks, bcProcessBeaconState)
				Expect(err).To(HaveOccurred())
			})
		})
	})

})