fix tests

ikhoon · Feb 20, 2025 · 07eb668 · 07eb668
1 parent 9c357a6
commit 07eb668
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/core/src/test/java/com/linecorp/armeria/common/loadbalancer/RampingUpLoadBalancerTest.java b/core/src/test/java/com/linecorp/armeria/common/loadbalancer/RampingUpLoadBalancerTest.java
@@ -500,7 +500,13 @@ private static RampingUpLoadBalancer<Endpoint> setInitialEndpoints(int numberOfS
                             .rampingUpTaskWindow(Duration.ofNanos(rampingUpTaskWindowNanos))
                             .ticker(ticker::get)
                             .totalSteps(numberOfSteps)
-                            .timestampFunction(EndpointAttributeKeys::createdAtNanos)
+                            .timestampFunction(endpoint -> {
+                                if (EndpointAttributeKeys.hasCreatedAtNanos(endpoint)) {
+                                    return EndpointAttributeKeys.createdAtNanos(endpoint);
+                                } else {
+                                    return null;
+                                }
+                            })
                             .executor(new ImmediateExecutor())
                             .build();