kaspad/testing/integration/basic_sync_test.go

package integration

import (
	"testing"
	"time"

	"github.com/kaspanet/kaspad/domain/consensus/utils/consensusserialization"

	"github.com/kaspanet/kaspad/app/appmessage"
)

func TestIntegrationBasicSync(t *testing.T) {
	appHarness1, appHarness2, appHarness3, teardown := standardSetup(t)
	defer teardown()

	// Connect nodes in chain: 1 <--> 2 <--> 3
	// So that node 3 doesn't directly get blocks from node 1
	connect(t, appHarness1, appHarness2)
	connect(t, appHarness2, appHarness3)

	app2OnBlockAddedChan := make(chan *appmessage.MsgBlockHeader)
	setOnBlockAddedHandler(t, appHarness2, func(notification *appmessage.BlockAddedNotificationMessage) {
		app2OnBlockAddedChan <- &notification.Block.Header
	})

	app3OnBlockAddedChan := make(chan *appmessage.MsgBlockHeader)
	setOnBlockAddedHandler(t, appHarness3, func(notification *appmessage.BlockAddedNotificationMessage) {
		app3OnBlockAddedChan <- &notification.Block.Header
	})

	block := mineNextBlock(t, appHarness1)

	var header *appmessage.MsgBlockHeader
	select {
	case header = <-app2OnBlockAddedChan:
	case <-time.After(defaultTimeout):
		t.Fatalf("Timeout waiting for block added notification on node directly connected to miner")
	}

	blockHash := consensusserialization.BlockHash(block)
	if *header.BlockHash() != *blockHash {
		t.Errorf("Expected block with hash '%s', but got '%s'", blockHash, header.BlockHash())
	}

	select {
	case header = <-app3OnBlockAddedChan:
	case <-time.After(defaultTimeout):
		t.Fatalf("Timeout waiting for block added notification on node indirectly connected to miner")
	}

	blockHash = consensusserialization.BlockHash(block)
	if *header.BlockHash() != *blockHash {
		t.Errorf("Expected block with hash '%s', but got '%s'", blockHash, header.BlockHash())
	}
}