Skip to content

Commit 7698f71

Browse files
authored
Set query timeouts for camel cassandara test (#19026)
1 parent 6fd466c commit 7698f71

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

  • instrumentation/camel-2.20/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/camel/v2_20/decorators

instrumentation/camel-2.20/javaagent/src/test/java/io/opentelemetry/javaagent/instrumentation/camel/v2_20/decorators/CassandraTest.java

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,11 +27,14 @@
2727
import static io.opentelemetry.semconv.incubating.DbIncubatingAttributes.DbSystemNameIncubatingValues.CASSANDRA;
2828

2929
import com.datastax.oss.driver.api.core.CqlSession;
30+
import com.datastax.oss.driver.api.core.config.DefaultDriverOption;
31+
import com.datastax.oss.driver.api.core.config.DriverConfigLoader;
3032
import com.google.common.collect.ImmutableMap;
3133
import io.opentelemetry.api.trace.SpanKind;
3234
import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
3335
import io.opentelemetry.instrumentation.testing.junit.http.AbstractHttpServerUsingTest;
3436
import io.opentelemetry.instrumentation.testing.junit.http.HttpServerInstrumentationExtension;
37+
import java.time.Duration;
3538
import org.apache.camel.CamelContext;
3639
import org.apache.camel.ProducerTemplate;
3740
import org.junit.jupiter.api.AfterAll;
@@ -97,10 +100,17 @@ protected void cleanUp() {
97100
}
98101

99102
static void cassandraSetup() {
103+
DriverConfigLoader configLoader =
104+
DriverConfigLoader.programmaticBuilder()
105+
.withDuration(DefaultDriverOption.REQUEST_TIMEOUT, Duration.ofSeconds(0))
106+
.withDuration(DefaultDriverOption.CONNECTION_INIT_QUERY_TIMEOUT, Duration.ofSeconds(10))
107+
.build();
108+
100109
try (CqlSession cqlSession =
101110
CqlSession.builder()
102111
.addContactPoint(cassandra.getContactPoint())
103112
.withLocalDatacenter(cassandra.getLocalDatacenter())
113+
.withConfigLoader(configLoader)
104114
.build()) {
105115
cqlSession.execute(
106116
"CREATE KEYSPACE IF NOT EXISTS test WITH replication = {'class': 'SimpleStrategy', 'replication_factor': 1};");

0 commit comments

Comments
 (0)