mirror of https://github.com/grpc/grpc-go.git
xds/resolver/test: use a non-blocking send instead of closing the channel (#6082)
This commit is contained in:
parent
b46bdef165
commit
e83e34be0b
|
|
@ -463,12 +463,15 @@ func (s) TestResolverWatchCallbackAfterClose(t *testing.T) {
|
||||||
// it receives a discovery request for a route configuration resource. And
|
// it receives a discovery request for a route configuration resource. And
|
||||||
// the test goroutine signals the management server when the resolver is
|
// the test goroutine signals the management server when the resolver is
|
||||||
// closed.
|
// closed.
|
||||||
waitForRouteConfigDiscoveryReqCh := make(chan struct{})
|
waitForRouteConfigDiscoveryReqCh := make(chan struct{}, 1)
|
||||||
waitForResolverCloseCh := make(chan struct{})
|
waitForResolverCloseCh := make(chan struct{})
|
||||||
mgmtServer, err := e2e.StartManagementServer(e2e.ManagementServerOptions{
|
mgmtServer, err := e2e.StartManagementServer(e2e.ManagementServerOptions{
|
||||||
OnStreamRequest: func(_ int64, req *v3discoverypb.DiscoveryRequest) error {
|
OnStreamRequest: func(_ int64, req *v3discoverypb.DiscoveryRequest) error {
|
||||||
if req.GetTypeUrl() == version.V3RouteConfigURL {
|
if req.GetTypeUrl() == version.V3RouteConfigURL {
|
||||||
close(waitForRouteConfigDiscoveryReqCh)
|
select {
|
||||||
|
case waitForRouteConfigDiscoveryReqCh <- struct{}{}:
|
||||||
|
default:
|
||||||
|
}
|
||||||
<-waitForResolverCloseCh
|
<-waitForResolverCloseCh
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue