File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff line change @@ -113,25 +113,24 @@ func (s) TestGracefulStop(t *testing.T) {
113
113
ss := & stubserver.StubServer {
114
114
Listener : dlis ,
115
115
FullDuplexCallF : func (stream testgrpc.TestService_FullDuplexCallServer ) error {
116
- _ , err := stream .Recv ()
117
- if err != nil {
116
+ if _ , err := stream .Recv (); err != nil {
118
117
return err
119
118
}
120
119
return stream .Send (& testpb.StreamingOutputCallResponse {})
121
120
},
122
121
S : grpc .NewServer (),
123
122
}
123
+ // 1. Start Server
124
124
stubserver .StartTestService (t , ss )
125
125
126
- // 1. Start Server
126
+ // 2. GracefulStop() Server after listener's Accept is called, but don't
127
+ // allow Accept() to exit when Close() is called on it.
127
128
gracefulStopDone := make (chan struct {})
128
129
<- dlis .acceptCalled
129
130
go func () {
130
131
ss .S .GracefulStop ()
131
132
close (gracefulStopDone )
132
133
}()
133
- // 2. GracefulStop() Server after listener's Accept is called, but don't
134
- // allow Accept() to exit when Close() is called on it.
135
134
136
135
// 3. Create a new connection to the server after listener.Close() is called.
137
136
// Server should close this connection immediately, before handshaking.
You can’t perform that action at this time.
0 commit comments