unit test

Author: aarshkshah1992

beacon-chain/p2p/service_test.go

@@ -20,6 +20,7 @@ import (
 	"github.com/OffchainLabs/prysm/v7/testing/assert"
 	"github.com/OffchainLabs/prysm/v7/testing/require"
 	prysmTime "github.com/OffchainLabs/prysm/v7/time"
+	"github.com/ethereum/go-ethereum/p2p/enode"
 	"github.com/libp2p/go-libp2p"
 	"github.com/libp2p/go-libp2p/core/event"
 	"github.com/libp2p/go-libp2p/core/host"
@@ -406,18 +407,24 @@ func TestService_SubscribeReachabilityEvents(t *testing.T) {
 	hook := logTest.NewGlobal()
 	ctx := t.Context()
 
-	h, _, _ := createHost(t, 0)
+	h, pkey, _ := createHost(t, 0)
 	defer func() {
 		if err := h.Close(); err != nil {
 			t.Fatal(err)
 		}
 	}()
 
-	// Create service with the host
+	// Create a LocalNode for ENR testing (without static IP set)
+	db, err := enode.OpenDB("")
+	require.NoError(t, err)
+	t.Cleanup(func() { db.Close() })
+	localNode := enode.NewLocalNode(db, pkey)
+
 	s := &Service{
-		ctx:  ctx,
-		host: h,
-		cfg:  &Config{EnableAutoNAT: true},
+		ctx:         ctx,
+		host:        h,
+		cfg:         &Config{EnableAutoNAT: true},
+		dv5Listener: testp2p.NewMockListener(localNode, nil),
 	}
 
 	// Get an emitter for the reachability event
@@ -455,4 +462,9 @@ func TestService_SubscribeReachabilityEvents(t *testing.T) {
 	require.LogsContain(t, hook, "/ip4/192.168.1.1/tcp/9000")
 	require.LogsContain(t, hook, "/ip4/10.0.0.1/tcp/9001")
 	require.LogsContain(t, hook, "/ip4/172.16.0.1/tcp/9002")
+
+	// Verify the ENR was updated with the reachable address
+	require.LogsContain(t, hook, "Updated ENR IP from AutoNAT reachability")
+	nodeRecord := localNode.Node()
+	assert.Equal(t, "192.168.1.1", nodeRecord.IP().String(), "ENR should contain the reachable IP address")
 }