diff --git a/server_test.go b/server_test.go index 4c5430a4a..595213e99 100644 --- a/server_test.go +++ b/server_test.go @@ -363,6 +363,8 @@ func TestClientEOF(t *testing.T) { t.Fatal(err) } + client.UserOnCloseWait(ctx) + // server shutdown, but we still make a call. if err := client.Call(ctx, serviceName, "Test", tp, tp); err == nil { t.Fatalf("expected error when calling against shutdown server")