client_test: Fix another case in racy TestTimeoutResponse.
this is hopefully last last one.
Change-Id: I6dd2fd8a8198950fd2211451ab6d383cfeea22e3
diff --git a/runtime/internal/rpc/client.go b/runtime/internal/rpc/client.go
index 5b9860e..64312d0 100644
--- a/runtime/internal/rpc/client.go
+++ b/runtime/internal/rpc/client.go
@@ -648,6 +648,9 @@
onlyErrNetwork = false
case stream.ErrAborted.ID, stream.ErrNetwork.ID:
// do nothing
+ case verror.ErrTimeout.ID:
+ topLevelError = verror.ErrTimeout
+ onlyErrNetwork = false
default:
onlyErrNetwork = false
}
diff --git a/runtime/internal/rpc/stream/manager/manager.go b/runtime/internal/rpc/stream/manager/manager.go
index 0898ad4..eb9af6d 100644
--- a/runtime/internal/rpc/stream/manager/manager.go
+++ b/runtime/internal/rpc/stream/manager/manager.go
@@ -169,7 +169,7 @@
result.conn.Close()
}
}()
- return nil, verror.New(stream.ErrDialFailed, ctx, err)
+ return nil, verror.New(verror.ErrTimeout, ctx)
}
if err != nil {
return nil, err
