From 963975135cde875c52df7ac324601ccf7ca6e825 Mon Sep 17 00:00:00 2001
From: Elle Mouton <elle.mouton@gmail.com>
Date: Wed, 25 Feb 2026 08:58:12 +0200
Subject: [PATCH] graph/tests: make randEdgePolicy assert internally

Update randEdgePolicy to take a testing handle, mark itself as a
helper, and assert internal packing errors directly with require.

Call sites now receive only the policy value without plumbing an
error through immediate require.NoError checks.
---
 graph/notifications_test.go | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/graph/notifications_test.go b/graph/notifications_test.go
index 340cdaf68..bef6ebdce 100644
--- a/graph/notifications_test.go
+++ b/graph/notifications_test.go
@@ -98,8 +98,10 @@ func createTestNode(t *testing.T) *models.Node {
 	return n
 }
 
-func randEdgePolicy(chanID *lnwire.ShortChannelID,
-	node *models.Node) (*models.ChannelEdgePolicy, error) {
+func randEdgePolicy(t testing.TB, chanID *lnwire.ShortChannelID,
+	node *models.Node) *models.ChannelEdgePolicy {
+
+	t.Helper()
 
 	InboundFee := models.InboundFee{
 		Base: prand.Int31() * -1,
@@ -108,9 +110,7 @@ func randEdgePolicy(chanID *lnwire.ShortChannelID,
 	inboundFee := InboundFee.ToWire()
 
 	var extraOpaqueData lnwire.ExtraOpaqueData
-	if err := extraOpaqueData.PackRecords(&inboundFee); err != nil {
-		return nil, err
-	}
+	require.NoError(t, extraOpaqueData.PackRecords(&inboundFee))
 
 	return &models.ChannelEdgePolicy{
 		Version:                   lnwire.GossipVersion1,
@@ -125,7 +125,7 @@ func randEdgePolicy(chanID *lnwire.ShortChannelID,
 		ToNode:                    node.PubKeyBytes,
 		InboundFee:                fn.Some(inboundFee),
 		ExtraOpaqueData:           extraOpaqueData,
-	}, nil
+	}
 }
 
 func createChannelEdge(bitcoinKey1, bitcoinKey2 []byte,
@@ -480,12 +480,10 @@ func TestEdgeUpdateNotification(t *testing.T) {
 
 	// Create random policy edges that are stemmed to the channel id
 	// created above.
-	edge1, err := randEdgePolicy(chanID, node1)
-	require.NoError(t, err, "unable to create a random chan policy")
+	edge1 := randEdgePolicy(t, chanID, node1)
 	edge1.ChannelFlags = 0
 
-	edge2, err := randEdgePolicy(chanID, node2)
-	require.NoError(t, err, "unable to create a random chan policy")
+	edge2 := randEdgePolicy(t, chanID, node2)
 	edge2.ChannelFlags = 1
 
 	require.NoError(t, ctx.builder.UpdateEdge(ctxb, edge1))