Filter out missed racy close on CancellationTest

apple · Apr 15, 2021 · 8541fda · 8541fda
1 parent 6df4265
commit 8541fda
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/...uter-jersey/src/testFixtures/java/io/servicetalk/http/router/jersey/CancellationTest.java b/...uter-jersey/src/testFixtures/java/io/servicetalk/http/router/jersey/CancellationTest.java
@@ -226,7 +226,12 @@ private void testCancelResponseSingle(final StreamingHttpRequest req,
             Single<StreamingHttpResponse> respSingle = execRule.executor().submit(() ->
                     jerseyRouter.handle(ctx, req, HTTP_REQ_RES_FACTORY)
             ).flatMap(identity())
-                    .beforeOnError(errorRef::set)
+                    .beforeOnError((err) -> {
+                        // Ignore racy cancellation, it's ordered safely.
+                        if (!(err instanceof IllegalStateException)) {
+                            errorRef.compareAndSet(null, err);
+                        }
+                    })
                     .afterCancel(cancelledLatch::countDown);
 
             toSource(respSingle).subscribe(new SingleSource.Subscriber<StreamingHttpResponse>() {