chore: ignore 'session shutdown' yamux error in tests

spikecurtis · spikecurtis · commit fc400d693e13 · 2025-05-21T11:14:45.000+04:00
diff --git a/testutil/logger.go b/testutil/logger.go
@@ -2,6 +2,7 @@ package testutil
 
 import (
 	"context"
+	"github.com/hashicorp/yamux"
 	"strings"
 	"testing"
 
@@ -24,6 +25,11 @@ func IgnoreLoggedError(entry slog.SinkEntry) bool {
 	if !ok {
 		return false
 	}
+	// Yamux sessions get shut down when we are shutting down tests, so ignoring
+	// them should reduce flakiness.
+	if xerrors.Is(err, yamux.ErrSessionShutdown) {
+		return true
+	}
 	// Canceled queries usually happen when we're shutting down tests, and so
 	// ignoring them should reduce flakiness.  This also includes
 	// context.Canceled and context.DeadlineExceeded errors, even if they are
