op-e2e: Add timeouts to multi-peer sync test (#13801)

op-node/p2p/sync_test.go

@@ -236,7 +236,7 @@ func TestMultiPeerSync(t *testing.T) {
 	hostA, hostB, hostC := hosts[0], hosts[1], hosts[2]
 	require.Equal(t, hostA.Network().Connectedness(hostB.ID()), network.Connected)
 
-	ctx, cancel := context.WithCancel(context.Background())
+	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Minute)
 	defer cancel()
 
 	clA, recvA := setupPeer(ctx, hostA)
@@ -264,11 +264,15 @@ func TestMultiPeerSync(t *testing.T) {
 	// With such large range to request we are going to hit the rate-limits of B and C,
 	// but that means we'll balance the work between the peers.
 	for i := uint64(89); i > 10; i-- { // wait for all payloads
-		e := <-recvA
+		select {
+		case e := <-recvA:
 			p := e.ExecutionPayload
 			exp, ok := payloads.getPayload(uint64(p.BlockNumber))
 			require.True(t, ok, "expecting known payload")
 			require.Equal(t, exp.ExecutionPayload.BlockHash, p.BlockHash, "expecting the correct payload")
+		case <-ctx.Done():
+			t.Fatal("timed out waiting for payload")
+		}
 	}
 
 	// now see if B can sync a range, and fill the gap with a re-request
@@ -280,15 +284,19 @@ func TestMultiPeerSync(t *testing.T) {
 	require.True(t, clB.activeRangeRequests.get(rangeReqId), "expecting range request to be active")
 
 	for i := uint64(29); i > 25; i-- {
-		p := <-recvB
+		select {
+		case p := <-recvB:
 			exp, ok := payloads.getPayload(uint64(p.ExecutionPayload.BlockNumber))
 			require.True(t, ok, "expecting known payload")
 			require.Equal(t, exp.ExecutionPayload.BlockHash, p.ExecutionPayload.BlockHash, "expecting the correct payload")
+		case <-ctx.Done():
+			t.Fatal("timed out waiting for payload")
+		}
 	}
 
 	// Wait for the request for block 25 to be made
-	ctx, cancelFunc := context.WithTimeout(context.Background(), 30*time.Second)
-	defer cancelFunc()
+	childCtx, cancelChild := context.WithTimeout(ctx, 30*time.Second)
+	defer cancelChild()
 	requestMade := false
 	for requestMade != true {
 		select {
@@ -296,7 +304,7 @@ func TestMultiPeerSync(t *testing.T) {
 			if blockNum == 25 {
 				requestMade = true
 			}
-		case <-ctx.Done():
+		case <-childCtx.Done():
 			t.Fatal("Did not request block 25 in a reasonable time")
 		}
 	}
@@ -310,10 +318,10 @@ func TestMultiPeerSync(t *testing.T) {
 	// But the re-request checks the status in the main loop, and it may thus look like it's still in-flight,
 	// and thus not run the new request.
 	// Wait till the failed request is recognized as marked as done, so the re-request actually runs.
-	ctx, cancelFunc = context.WithTimeout(context.Background(), 30*time.Second)
-	defer cancelFunc()
+	childCtx, cancelChild = context.WithTimeout(ctx, 30*time.Second)
+	defer cancelChild()
 	for {
-		isInFlight, err := clB.isInFlight(ctx, 25)
+		isInFlight, err := clB.isInFlight(childCtx, 25)
 		require.NoError(t, err)
 		time.Sleep(time.Second)
 		if !isInFlight {
@@ -328,7 +336,8 @@ func TestMultiPeerSync(t *testing.T) {
 	_, err = clB.RequestL2Range(ctx, payloads.getBlockRef(20), payloads.getBlockRef(26))
 	require.NoError(t, err)
 	for i := uint64(25); i > 20; i-- {
-		p := <-recvB
+		select {
+		case p := <-recvB:
 			exp, ok := payloads.getPayload(uint64(p.ExecutionPayload.BlockNumber))
 			require.True(t, ok, "expecting known payload")
 			require.Equal(t, exp.ExecutionPayload.BlockHash, p.ExecutionPayload.BlockHash, "expecting the correct payload")
@@ -338,6 +347,9 @@ func TestMultiPeerSync(t *testing.T) {
 			} else {
 				require.Nil(t, p.ParentBeaconBlockRoot)
 			}
+		case <-ctx.Done():
+			t.Fatal("timed out waiting for payload")
+		}
 	}
 }