use ctx.eventLoop()

line · Nov 8, 2023 · d73eabc · d73eabc
1 parent 422583d
commit d73eabc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...src/test/java/com/linecorp/armeria/client/retry/RetryingClientEventLoopSchedulerTest.java b/...src/test/java/com/linecorp/armeria/client/retry/RetryingClientEventLoopSchedulerTest.java
@@ -27,6 +27,7 @@
 import org.junit.jupiter.api.extension.RegisterExtension;
 
 import com.linecorp.armeria.client.BlockingWebClient;
+import com.linecorp.armeria.client.ClientRequestContext;
 import com.linecorp.armeria.client.ClientRequestContextCaptor;
 import com.linecorp.armeria.client.Clients;
 import com.linecorp.armeria.client.Endpoint;
@@ -35,7 +36,6 @@
 import com.linecorp.armeria.common.AggregatedHttpResponse;
 import com.linecorp.armeria.common.HttpResponse;
 import com.linecorp.armeria.common.HttpStatus;
-import com.linecorp.armeria.common.RequestContext;
 import com.linecorp.armeria.common.SessionProtocol;
 import com.linecorp.armeria.common.logging.RequestLogAccess;
 import com.linecorp.armeria.internal.testing.AnticipatedException;
@@ -101,9 +101,9 @@ void shouldReturnCorrectEventLoop() {
             final List<RequestLogAccess> children = captor.get().log().children();
             assertThat(children.size()).isEqualTo(6);
             for (int i = 0; i < 6; i++) {
-                final RequestContext childCtx = children.get(i).context();
+                final ClientRequestContext childCtx = (ClientRequestContext) children.get(i).context();
                 assertThat(childCtx.eventLoop().withoutContext())
-                        .isSameAs(eventLoopMapping.get(endpoints.get(i % 3)));
+                        .isSameAs(eventLoopMapping.get(childCtx.endpoint()));
             }
         }
     }