dkg: fixed TestSyncFlow (#3309)

Possibly fixed the flakey test. category: test ticket: #3308
ObolNetwork · Oct 1, 2024 · bb91ab7 · bb91ab7
1 parent 8876c3c
commit bb91ab7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/dkg/dkg_test.go b/dkg/dkg_test.go
@@ -6,6 +6,7 @@ import (
 	"context"
 	"encoding/hex"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"math/rand"
 	"net/http"
@@ -537,7 +538,9 @@ func TestSyncFlow(t *testing.T) {
 			var disconnectedCount int
 			for err := range dkgErrChan {
 				testutil.SkipIfBindErr(t, err)
-				require.NoError(t, err)
+				if !errors.Is(err, context.Canceled) {
+					require.NoError(t, err)
+				}
 				disconnectedCount++
 				if disconnectedCount == test.nodes {
 					break