@@ -344,7 +344,7 @@ public void clientStartStop() throws Exception {
344344 client .shutdown (shutdownReason );
345345 verify (mockClientTransportListener , timeout (TIMEOUT_MS )).transportTerminated ();
346346 inOrder .verify (mockClientTransportListener ).transportShutdown (same (shutdownReason ),
347- eq ( SimpleDisconnectError . SUBCHANNEL_SHUTDOWN ));
347+ any ( DisconnectError . class ));
348348 inOrder .verify (mockClientTransportListener ).transportTerminated ();
349349 verify (mockClientTransportListener , never ()).transportInUse (anyBoolean ());
350350 }
@@ -361,7 +361,7 @@ public void clientStartAndStopOnceConnected() throws Exception {
361361 client .shutdown (Status .UNAVAILABLE );
362362 verify (mockClientTransportListener , timeout (TIMEOUT_MS )).transportTerminated ();
363363 inOrder .verify (mockClientTransportListener ).transportShutdown (any (Status .class ),
364- eq ( SimpleDisconnectError . SUBCHANNEL_SHUTDOWN ));
364+ any ( DisconnectError . class ));
365365 inOrder .verify (mockClientTransportListener ).transportTerminated ();
366366 assertTrue (serverTransportListener .waitForTermination (TIMEOUT_MS , TimeUnit .MILLISECONDS ));
367367 server .shutdown ();
@@ -598,7 +598,7 @@ public void ping_duringShutdown() throws Exception {
598598 stream .start (clientStreamListener );
599599 client .shutdown (Status .UNAVAILABLE );
600600 verify (mockClientTransportListener , timeout (TIMEOUT_MS )).transportShutdown (any (Status .class ),
601- eq ( SimpleDisconnectError . SUBCHANNEL_SHUTDOWN ));
601+ any ( DisconnectError . class ));
602602 ClientTransport .PingCallback mockPingCallback = mock (ClientTransport .PingCallback .class );
603603 try {
604604 client .ping (mockPingCallback , MoreExecutors .directExecutor ());
@@ -643,7 +643,7 @@ public void newStream_duringShutdown() throws Exception {
643643 stream .start (clientStreamListener );
644644 client .shutdown (Status .UNAVAILABLE );
645645 verify (mockClientTransportListener , timeout (TIMEOUT_MS )).transportShutdown (any (Status .class ),
646- eq ( SimpleDisconnectError . SUBCHANNEL_SHUTDOWN ));
646+ any ( DisconnectError . class ));
647647
648648 ClientStream stream2 = client .newStream (
649649 methodDescriptor , new Metadata (), callOptions , tracers );
0 commit comments