Implement TestPickVirtualParents.

domain/consensus/processes/consensusstatemanager/pick_virtual_parents_test.go

@@ -0,0 +1,49 @@
+package consensusstatemanager_test
+
+import (
+	"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"
+	"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)
+
+	// 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++ {
+			block, _, err := testConsensus.BuildBlockWithParents([]*externalapi.DomainHash{tipHash}, nil, nil)
+			if err != nil {
+				t.Fatalf("Could not build block with the genesis block as its only parent: %s", err)
+			}
+			blockHash := consensushashing.BlockHash(block)
+			start := time.Now()
+			_, err = testConsensus.ValidateAndInsertBlock(block)
+			if err != nil {
+				t.Fatalf("Failed to validate block %s: %s", blockHash, err)
+			}
+			validationTime := time.Since(start)
+			accumulatedValidationTime += validationTime
+			t.Logf("Validated block #%d in chain #%d, took %s", blockIndex, chainIndex, validationTime)
+			tipHash = blockHash
+		}
+
+		averageValidationTime := accumulatedValidationTime / 1000
+		t.Logf("Average validation time for chain #%d: %s", chainIndex, averageValidationTime)
+	}
+}