fix unit test poll

open-telemetry · Oct 4, 2024 · fb1cd6e · fb1cd6e
1 parent b6c2d6e
commit fb1cd6e
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/aws-xray/src/test/java/io/opentelemetry/contrib/awsxray/AwsXrayRemoteSamplerTest.java b/aws-xray/src/test/java/io/opentelemetry/contrib/awsxray/AwsXrayRemoteSamplerTest.java
@@ -170,8 +170,15 @@ void defaultInitialSampler() {
 
   @Test
   void parentBasedXraySamplerAfterDefaultSampler() {
-    try (AwsXrayRemoteSampler sampler = AwsXrayRemoteSampler.newBuilder(Resource.empty()).build()) {
+    rulesResponse.set(RULE_RESPONSE_1);
+    try (AwsXrayRemoteSampler samplerWithLongerPollingInterval =
+        AwsXrayRemoteSampler.newBuilder(Resource.empty())
+            .setInitialSampler(Sampler.alwaysOn())
+            .setEndpoint(server.httpUri().toString())
+            .setPollingInterval(Duration.ofMillis(5))
+            .build()) {
       await()
+          .pollDelay(Duration.ofMillis(10))
           .untilAsserted(
               () -> {
                 assertThat(sampler.getDescription())