Skip to content

Commit 166602f

Browse files
committed
Fix non-deterministic test
1 parent eeab7f4 commit 166602f

File tree

1 file changed

+8
-10
lines changed

1 file changed

+8
-10
lines changed

grpc/middleware/canceler_test.go

Lines changed: 8 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,10 @@ import (
44
"context"
55
"testing"
66

7-
grpcm "goa.design/goa/v3/grpc/middleware"
7+
"github.com/stretchr/testify/assert"
88
"google.golang.org/grpc"
9+
10+
grpcm "goa.design/goa/v3/grpc/middleware"
911
)
1012

1113
type (
@@ -46,15 +48,11 @@ func TestStreamCanceler(t *testing.T) {
4648
for _, c := range cases {
4749
t.Run(c.name, func(t *testing.T) {
4850
ctx, cancel := context.WithCancel(context.Background())
49-
50-
go func() {
51-
cancel()
52-
}()
53-
54-
if err := grpcm.StreamCanceler(ctx)(nil, c.stream, stream, c.handler); err != nil {
55-
if err.Error() != "server is stopping" {
56-
t.Errorf("StreamCanceler error: %v", err)
57-
}
51+
interceptor := grpcm.StreamCanceler(ctx)
52+
cancel()
53+
err := interceptor(nil, c.stream, stream, c.handler)
54+
if err != nil {
55+
assert.ErrorContains(t, err, "server is stopping")
5856
}
5957
})
6058
}

0 commit comments

Comments
 (0)