Fix race condition in test

raphael · raphael · commit d021963cddb8 · 2024-05-18T15:05:02.000-07:00
diff --git a/grpc/middleware/canceler_test.go b/grpc/middleware/canceler_test.go
@@ -52,7 +52,9 @@ func TestStreamCanceler(t *testing.T) {
 			}()
 
 			if err := grpcm.StreamCanceler(ctx)(nil, c.stream, stream, c.handler); err != nil {
-				t.Errorf("StreamCanceler error: %v", err)
+				if err.Error() != "server is stopping" {
+					t.Errorf("StreamCanceler error: %v", err)
+				}
 			}
 		})
 	}

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,9 @@ func TestStreamCanceler(t *testing.T) {`
`52`	`52`	`}()`
`53`	`53`
`54`	`54`	`if err := grpcm.StreamCanceler(ctx)(nil, c.stream, stream, c.handler); err != nil {`
`55`		`- t.Errorf("StreamCanceler error: %v", err)`
	`55`	`+ if err.Error() != "server is stopping" {`
	`56`	`+ t.Errorf("StreamCanceler error: %v", err)`
	`57`	`+ }`
`56`	`58`	`}`
`57`	`59`	`})`
`58`	`60`	`}`