add r2dbc test

Author: zeitlinger

smoke-tests-otel-starter/spring-boot-3-reactive/src/main/java/io/opentelemetry/spring/smoketest/OtelReactiveSpringStarterSmokeTestController.java

@@ -13,9 +13,16 @@
 public class OtelReactiveSpringStarterSmokeTestController {
 
   public static final String WEBFLUX = "/webflux";
+  private final PlayerRepository playerRepository;
+
+  public OtelReactiveSpringStarterSmokeTestController(PlayerRepository playerRepository) {
+    this.playerRepository = playerRepository;
+  }
 
   @GetMapping(WEBFLUX)
   public Mono<String> getStock() {
-    return Mono.just("pong");
+    return playerRepository
+        .findById(1)
+        .map(player -> "Player: " + player.getName() + " Age: " + player.getAge());
   }
 }

smoke-tests-otel-starter/spring-boot-3-reactive/src/main/java/io/opentelemetry/spring/smoketest/Player.java

@@ -0,0 +1,34 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.spring.smoketest;
+
+import org.springframework.data.annotation.Id;
+
+public class Player {
+  @Id Integer id;
+  String name;
+  Integer age;
+
+  public Player() {}
+
+  public Player(Integer id, String name, Integer age) {
+    this.id = id;
+    this.name = name;
+    this.age = age;
+  }
+
+  public Integer getId() {
+    return id;
+  }
+
+  public String getName() {
+    return name;
+  }
+
+  public Integer getAge() {
+    return age;
+  }
+}

smoke-tests-otel-starter/spring-boot-3-reactive/src/main/java/io/opentelemetry/spring/smoketest/PlayerRepository.java

@@ -0,0 +1,10 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.spring.smoketest;
+
+import org.springframework.data.repository.reactive.ReactiveCrudRepository;
+
+public interface PlayerRepository extends ReactiveCrudRepository<Player, Integer> {}

smoke-tests-otel-starter/spring-boot-3-reactive/src/main/resources/application.yaml

@@ -1,3 +1,6 @@
 spring:
   r2dbc:
     url: r2dbc:h2:mem:///testdb
+  jpa:
+    hibernate:
+      ddl-auto: create

smoke-tests-otel-starter/spring-boot-3-reactive/src/main/resources/schema.sql

@@ -0,0 +1 @@
+CREATE TABLE IF NOT EXISTS player(id INT NOT NULL AUTO_INCREMENT, name VARCHAR(255), age INT, PRIMARY KEY (id));

smoke-tests-otel-starter/spring-boot-3-reactive/src/test/java/io/opentelemetry/smoketest/OtelReactiveSpringStarterSmokeTest.java

@@ -12,6 +12,7 @@
 import io.opentelemetry.instrumentation.testing.junit.LibraryInstrumentationExtension;
 import io.opentelemetry.semconv.HttpAttributes;
 import io.opentelemetry.semconv.UrlAttributes;
+import io.opentelemetry.semconv.incubating.DbIncubatingAttributes;
 import io.opentelemetry.spring.smoketest.OtelReactiveSpringStarterSmokeTestApplication;
 import io.opentelemetry.spring.smoketest.OtelReactiveSpringStarterSmokeTestController;
 import org.junit.jupiter.api.AfterEach;
@@ -64,7 +65,7 @@ void tearDown(CapturedOutput output) {
   }
 
   @Test
-  void webflux() {
+  void webClientAndWebFluxAndR2dbc() {
     webClient
         .get()
         .uri(OtelReactiveSpringStarterSmokeTestController.WEBFLUX)
@@ -73,19 +74,28 @@ void webflux() {
         .blockLast();
 
     testing.waitAndAssertTraces(
-            trace ->
-                trace.hasSpansSatisfyingExactly(
-                    clientSpan ->
-                        clientSpan
-                            .hasKind(SpanKind.CLIENT)
-                            .hasAttributesSatisfying(
-                                a ->
-                                    assertThat(a.get(UrlAttributes.URL_FULL)).endsWith("/webflux")),
-                    serverSpan ->
-                        serverSpan
-                            .hasKind(SpanKind.SERVER)
-                            .hasAttribute(HttpAttributes.HTTP_REQUEST_METHOD, "GET")
-                            .hasAttribute(HttpAttributes.HTTP_RESPONSE_STATUS_CODE, 200L)
-                            .hasAttribute(HttpAttributes.HTTP_ROUTE, "/webflux")));
+        trace ->
+            trace.hasSpansSatisfyingExactly(
+                span ->
+                    span.hasKind(SpanKind.CLIENT)
+                        .hasName("GET")
+                        .hasAttributesSatisfying(
+                            a -> assertThat(a.get(UrlAttributes.URL_FULL)).endsWith("/webflux")),
+                span ->
+                    span.hasKind(SpanKind.SERVER)
+                        .hasName("GET /webflux")
+                        .hasAttribute(HttpAttributes.HTTP_REQUEST_METHOD, "GET")
+                        .hasAttribute(HttpAttributes.HTTP_RESPONSE_STATUS_CODE, 200L)
+                        .hasAttribute(HttpAttributes.HTTP_ROUTE, "/webflux"),
+                span ->
+                    span.hasKind(SpanKind.CLIENT)
+                        .hasName("SELECT testdb.PLAYER")
+                        .hasAttribute(DbIncubatingAttributes.DB_NAME, "testdb")
+                        .hasAttribute(DbIncubatingAttributes.DB_SQL_TABLE, "PLAYER")
+                        .hasAttribute(DbIncubatingAttributes.DB_OPERATION, "SELECT")
+                        .hasAttribute(
+                            DbIncubatingAttributes.DB_STATEMENT,
+                            "SELECT PLAYER.* FROM PLAYER WHERE PLAYER.ID = $? LIMIT ?")
+                        .hasAttribute(DbIncubatingAttributes.DB_SYSTEM, "h2")));
   }
 }