@@ -3085,13 +3085,15 @@ public void LongPollingTransportAccessTokenProviderThrowsAfterHandshakeClosesCon
3085
3085
public void LongPollingTransportAccessTokenProviderThrowsDuringStop () {
3086
3086
AtomicInteger requestCount = new AtomicInteger (0 );
3087
3087
CompletableSubject blockGet = CompletableSubject .create ();
3088
+ CompletableSubject blockStop = CompletableSubject .create ();
3088
3089
TestHttpClient client = new TestHttpClient ()
3089
3090
.on ("POST" , "http://example.com/negotiate?negotiateVersion=1" ,
3090
3091
(req ) -> Single .just (new HttpResponse (200 , "" ,
3091
3092
TestUtils .stringToByteBuffer ("{\" connectionId\" :\" bVOiRPG8-6YiJ6d7ZcTOVQ\" ,\" "
3092
3093
+ "availableTransports\" :[{\" transport\" :\" LongPolling\" ,\" transferFormats\" :[\" Text\" ,\" Binary\" ]}]}" ))))
3093
3094
.on ("GET" , (req ) -> {
3094
3095
if (requestCount .getAndIncrement () > 1 ) {
3096
+ blockStop .onComplete ();
3095
3097
blockGet .blockingAwait ();
3096
3098
}
3097
3099
return Single .just (new HttpResponse (200 , "" , TestUtils .stringToByteBuffer ("{}" + RECORD_SEPARATOR )));
@@ -3120,6 +3122,7 @@ public void LongPollingTransportAccessTokenProviderThrowsDuringStop() {
3120
3122
3121
3123
hubConnection .start ().timeout (30 , TimeUnit .SECONDS ).blockingAwait ();
3122
3124
3125
+ blockStop .timeout (30 , TimeUnit .SECONDS ).blockingAwait ();
3123
3126
try {
3124
3127
hubConnection .stop ().timeout (30 , TimeUnit .SECONDS ).blockingAwait ();
3125
3128
assertTrue (false );
0 commit comments