kaspad/domain/consensus/processes/dagtopologymanager/dagtopologymanager_external_test.go

package dagtopologymanager_test

import (
	"testing"

	"github.com/kaspanet/kaspad/domain/consensus"
	"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
	"github.com/kaspanet/kaspad/domain/consensus/utils/testutils"
	"github.com/kaspanet/kaspad/domain/dagconfig"
)

func TestIsAncestorOf(t *testing.T) {
	testutils.ForAllNets(t, true, func(t *testing.T, params *dagconfig.Params) {
		factory := consensus.NewFactory()
		tc, tearDown, err := factory.NewTestConsensus(params, "TestIsAncestorOf")
		if err != nil {
			t.Fatalf("NewTestConsensus: %s", err)
		}
		defer tearDown()

		// Add a chain of two blocks above the genesis. This will be the
		// selected parent chain.
		blockA, err := tc.AddBlock([]*externalapi.DomainHash{params.GenesisHash}, nil, nil)
		if err != nil {
			t.Fatalf("AddBlock: %+v", err)
		}

		blockB, err := tc.AddBlock([]*externalapi.DomainHash{blockA}, nil, nil)
		if err != nil {
			t.Fatalf("AddBlock: %s", err)
		}

		// Add another block above the genesis
		blockC, err := tc.AddBlock([]*externalapi.DomainHash{params.GenesisHash}, nil, nil)
		if err != nil {
			t.Fatalf("AddBlock: %s", err)
		}

		// Add a block whose parents are the two tips
		blockD, err := tc.AddBlock([]*externalapi.DomainHash{blockB, blockC}, nil, nil)
		if err != nil {
			t.Fatalf("AddBlock: %s", err)
		}

		isAncestorOf, err := tc.DAGTopologyManager().IsAncestorOf(blockC, blockD)
		if err != nil {
			t.Fatalf("IsAncestorOf: %s", err)
		}
		if !isAncestorOf {
			t.Fatalf("TestIsInPast: node C is unexpectedly not the past of node D")
		}
	})
}