erigon-pulse/cl/sentinel/sentinel_gossip_test.go

package sentinel

import (
	"context"
	"math"
	"testing"
	"time"

	"github.com/ledgerwatch/erigon/cl/clparams"
	"github.com/ledgerwatch/erigon/cl/persistence"
	"github.com/ledgerwatch/log/v3"
	"github.com/libp2p/go-libp2p/core/peer"
	"github.com/stretchr/testify/require"
)

func TestSentinelGossipAverage(t *testing.T) {
	t.Skip("TODO: fix me")
	listenAddrHost := "127.0.0.1"

	ctx := context.Background()
	db, _, f, _, _ := loadChain(t)
	raw := persistence.NewAferoRawBlockSaver(f, &clparams.MainnetBeaconConfig)
	genesisConfig, networkConfig, beaconConfig := clparams.GetConfigsByNetwork(clparams.MainnetNetwork)
	sentinel1, err := New(ctx, &SentinelConfig{
		NetworkConfig: networkConfig,
		BeaconConfig:  beaconConfig,
		GenesisConfig: genesisConfig,
		IpAddr:        listenAddrHost,
		Port:          7070,
		EnableBlocks:  true,
	}, raw, db, log.New())
	require.NoError(t, err)
	defer sentinel1.Stop()

	require.NoError(t, sentinel1.Start())
	h := sentinel1.host

	sentinel2, err := New(ctx, &SentinelConfig{
		NetworkConfig: networkConfig,
		BeaconConfig:  beaconConfig,
		GenesisConfig: genesisConfig,
		IpAddr:        listenAddrHost,
		Port:          7077,
		EnableBlocks:  true,
		TCPPort:       9123,
	}, raw, db, log.New())
	require.NoError(t, err)
	defer sentinel2.Stop()

	require.NoError(t, sentinel2.Start())
	h2 := sentinel2.host

	sub1, err := sentinel1.SubscribeGossip(BeaconBlockSsz)
	require.NoError(t, err)
	defer sub1.Close()

	require.NoError(t, sub1.Listen())

	sub2, err := sentinel2.SubscribeGossip(BeaconBlockSsz)
	require.NoError(t, err)
	defer sub2.Close()
	require.NoError(t, sub2.Listen())

	err = h.Connect(ctx, peer.AddrInfo{
		ID:    h2.ID(),
		Addrs: h2.Addrs(),
	})
	require.NoError(t, err)

	ch := sentinel2.RecvGossip()
	msg := []byte("hello")
	go func() {
		// delay to make sure that the connection is established
		time.Sleep(5 * time.Second)
		sub1.Publish(msg)
	}()

	select {
	case ans := <-ch:
		require.Equal(t, len(msg), len(ans.Message.Data))
		require.Equal(t, msg, ans.Data)
	case <-ctx.Done():
		t.Fatal("timeout")
	}
}

func TestSentinelGossipOnHardFork(t *testing.T) {
	listenAddrHost := "127.0.0.1"

	ctx := context.Background()
	db, _, f, _, _ := loadChain(t)
	raw := persistence.NewAferoRawBlockSaver(f, &clparams.MainnetBeaconConfig)
	genesisConfig, networkConfig, beaconConfig := clparams.GetConfigsByNetwork(clparams.MainnetNetwork)
	bcfg := *beaconConfig

	bcfg.AltairForkEpoch = math.MaxUint64
	bcfg.BellatrixForkEpoch = math.MaxUint64
	bcfg.CapellaForkEpoch = math.MaxUint64
	bcfg.DenebForkEpoch = math.MaxUint64
	bcfg.InitializeForkSchedule()

	sentinel1, err := New(ctx, &SentinelConfig{
		NetworkConfig: networkConfig,
		BeaconConfig:  &bcfg,
		GenesisConfig: genesisConfig,
		IpAddr:        listenAddrHost,
		Port:          7070,
		EnableBlocks:  true,
	}, raw, db, log.New())
	require.NoError(t, err)
	defer sentinel1.Stop()

	require.NoError(t, sentinel1.Start())
	h := sentinel1.host

	sentinel2, err := New(ctx, &SentinelConfig{
		NetworkConfig: networkConfig,
		BeaconConfig:  &bcfg,
		GenesisConfig: genesisConfig,
		IpAddr:        listenAddrHost,
		Port:          7077,
		EnableBlocks:  true,
		TCPPort:       9123,
	}, raw, db, log.New())
	require.NoError(t, err)
	defer sentinel2.Stop()

	require.NoError(t, sentinel2.Start())
	h2 := sentinel2.host

	sub1, err := sentinel1.SubscribeGossip(BeaconBlockSsz)
	require.NoError(t, err)
	defer sub1.Close()

	require.NoError(t, sub1.Listen())

	sub2, err := sentinel2.SubscribeGossip(BeaconBlockSsz)
	require.NoError(t, err)
	defer sub2.Close()
	require.NoError(t, sub2.Listen())

	err = h.Connect(ctx, peer.AddrInfo{
		ID:    h2.ID(),
		Addrs: h2.Addrs(),
	})
	require.NoError(t, err)

	ch := sentinel2.RecvGossip()
	msg := []byte("hello")
	go func() {
		// delay to make sure that the connection is established
		time.Sleep(time.Second)
		sub1.Publish(msg)
	}()
	previousTopic := ""

	select {
	case ans := <-ch:
		require.Equal(t, len(msg), len(ans.Message.Data))
		previousTopic = *ans.Topic
	case <-ctx.Done():
		t.Fatal("timeout")
	}

	bcfg.AltairForkEpoch = clparams.MainnetBeaconConfig.AltairForkEpoch
	bcfg.InitializeForkSchedule()
	msg = []byte("hello1")
	go func() {
		// delay to make sure that the connection is established
		time.Sleep(time.Second)
		sub1 = sentinel1.subManager.GetMatchingSubscription(string(BeaconBlockSsz.Name))
		sub1.Publish(msg)
	}()

	select {
	case ans := <-ch:
		require.NotEqual(t, previousTopic, *ans.Topic)
	case <-ctx.Done():
		t.Fatal("timeout")
	}
}