Fix flaky CassandraTest by adding warm-up call to initialize cassandra session

Author: trask

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

@@ -115,6 +115,12 @@ void testCassandra() {
     CamelContext camelContext = appContext.getBean(CamelContext.class);
     ProducerTemplate template = camelContext.createProducerTemplate();
 
+    // warm up: force the cassandra session initialization (including "USE keyspace")
+    // so that the keyspace-switching span doesn't interfere with the test assertion
+    template.requestBody("direct:input", (Object) null);
+    testing.waitForTraces(1);
+    testing.clearData();
+
     template.requestBody("direct:input", (Object) null);
 
     if (emitStableDatabaseSemconv()) {
@@ -142,22 +148,7 @@ void testCassandra() {
                               equalTo(
                                   DB_QUERY_TEXT,
                                   "select * from test.users where id=1 ALLOW FILTERING"),
-                              equalTo(DB_QUERY_SUMMARY, "SELECT test.users"))),
-          trace ->
-              trace.hasSpansSatisfyingExactly(
-                  span ->
-                      span.hasName("USE test")
-                          .hasKind(SpanKind.CLIENT)
-                          .hasAttributesSatisfyingExactly(
-                              satisfies(NETWORK_TYPE, val -> val.isInstanceOf(String.class)),
-                              equalTo(SERVER_ADDRESS, host),
-                              equalTo(SERVER_PORT, cassandraPort),
-                              satisfies(
-                                  NETWORK_PEER_ADDRESS, val -> val.isInstanceOf(String.class)),
-                              equalTo(NETWORK_PEER_PORT, cassandraPort),
-                              equalTo(DB_SYSTEM_NAME, "cassandra"),
-                              equalTo(DB_QUERY_TEXT, "USE test"),
-                              equalTo(DB_QUERY_SUMMARY, "USE test"))));
+                              equalTo(DB_QUERY_SUMMARY, "SELECT test.users"))));
     } else {
       testing.waitAndAssertTraces(
           trace ->
@@ -185,10 +176,7 @@ void testCassandra() {
                   span ->
                       span.hasName("SELECT test.users")
                           .hasKind(SpanKind.CLIENT)
-                          .hasParent(trace.getSpan(1))),
-          trace ->
-              trace.hasSpansSatisfyingExactly(
-                  span -> span.hasName("DB Query").hasKind(SpanKind.CLIENT)));
+                          .hasParent(trace.getSpan(1))));
     }
   }
 }