kaspad/domain/consensus/processes/consensusstatemanager/pick_virtual_parents_test.go

package consensusstatemanager_test

import (
	"fmt"
	"github.com/kaspanet/kaspad/domain/consensus"
	"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
	"github.com/kaspanet/kaspad/domain/consensus/utils/consensushashing"
	"github.com/kaspanet/kaspad/domain/dagconfig"
	"github.com/kaspanet/kaspad/infrastructure/logger"
	"testing"
	"time"
)

func TestPickVirtualParents(t *testing.T) {
	params := dagconfig.DevnetParams
	params.SkipProofOfWork = true

	factory := consensus.NewFactory()
	testConsensus, teardown, err := factory.NewTestConsensus(&params, false, "TestPickVirtualParents")
	if err != nil {
		t.Fatalf("Error setting up consensus: %+v", err)
	}
	defer teardown(false)

	found := false

	// Build three chains over the genesis
	for chainIndex := 0; chainIndex < 3; chainIndex++ {
		const chainSize = 1000
		accumulatedValidationTime := time.Duration(0)

		tipHash := params.GenesisHash
		for blockIndex := 0; blockIndex < chainSize; blockIndex++ {
			if found {
				fmt.Printf("\n\n\nBUILD BLOCK WITH PARENTS\n\n\n")
			}
			block, _, err := testConsensus.BuildBlockWithParents([]*externalapi.DomainHash{tipHash}, nil, nil)
			if err != nil {
				t.Fatalf("Could not build block: %s", err)
			}
			blockHash := consensushashing.BlockHash(block)
			start := time.Now()
			if found {
				fmt.Printf("\n\n\nVALIDATE AND INSERT BLOCK\n\n\n")
			}
			_, err = testConsensus.ValidateAndInsertBlock(block)
			if err != nil {
				t.Fatalf("Failed to validate block %s: %s", blockHash, err)
			}
			validationTime := time.Since(start)

			accumulatedValidationTime += validationTime
			fmt.Printf("Validated block #%d in chain #%d, took %s\n", blockIndex, chainIndex, validationTime)
			tipHash = blockHash

			if found {
				t.Fatalf("DONE")
			}
			if validationTime > 100*time.Millisecond {
				found = true
				logger.SetLogLevels("debug")
			}
		}

		averageValidationTime := accumulatedValidationTime / 1000
		fmt.Printf("Average validation time for chain #%d: %s\n", chainIndex, averageValidationTime)
	}
}