Merge branch 'heya/fix-cassandra-target' of https://github.com/heyams…

…/opentelemetry-java-instrumentation into heya/fix-cassandra-target
open-telemetry · Feb 15, 2024 · 23df5ac · 23df5ac
2 parents a5cb42c + bce1ca4
commit 23df5ac
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/...esting/src/main/java/io/opentelemetry/testing/cassandra/v4_4/AbstractCassandra44Test.java b/...esting/src/main/java/io/opentelemetry/testing/cassandra/v4_4/AbstractCassandra44Test.java
@@ -24,9 +24,12 @@
 import static org.junit.jupiter.api.Named.named;
 
 import com.datastax.oss.driver.api.core.CqlSession;
+import com.datastax.oss.driver.api.core.CqlSessionBuilder;
+import com.datastax.oss.driver.internal.core.metadata.SniEndPoint;
 import io.opentelemetry.api.trace.SpanKind;
 import io.opentelemetry.cassandra.v4.common.AbstractCassandraTest;
 import io.opentelemetry.instrumentation.api.semconv.network.internal.NetworkAttributes;
+import java.net.InetSocketAddress;
 import java.util.stream.Stream;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
@@ -139,4 +142,11 @@ private static Stream<Arguments> provideReactiveParameters() {
                     "SELECT",
                     "users"))));
   }
+
+  @Override
+  protected CqlSessionBuilder addContactPoint(CqlSessionBuilder sessionBuilder) {
+    InetSocketAddress address = new InetSocketAddress("localhost", cassandraPort);
+    sessionBuilder.addContactEndPoint(new SniEndPoint(address, "dummy"));
+    return sessionBuilder;
+  }
 }