|
25 | 25 | import static io.opentelemetry.semconv.incubating.DbIncubatingAttributes.DB_STATEMENT; |
26 | 26 | import static io.opentelemetry.semconv.incubating.DbIncubatingAttributes.DB_SYSTEM; |
27 | 27 | import static io.opentelemetry.semconv.incubating.DbIncubatingAttributes.DB_USER; |
| 28 | +import static io.opentelemetry.semconv.incubating.PeerIncubatingAttributes.PEER_SERVICE; |
28 | 29 | import static java.util.Arrays.asList; |
29 | 30 | import static org.assertj.core.api.Assertions.assertThatThrownBy; |
30 | 31 |
|
@@ -90,6 +91,10 @@ protected DataSource wrap(DataSource dataSource) { |
90 | 91 | return dataSource; |
91 | 92 | } |
92 | 93 |
|
| 94 | + protected boolean hasPeerService() { |
| 95 | + return true; |
| 96 | + } |
| 97 | + |
93 | 98 | private static final String dbName = "jdbcUnitTest"; |
94 | 99 | private static final String dbNameLower = dbName.toLowerCase(Locale.ROOT); |
95 | 100 | private static final Map<String, String> jdbcUrls = |
@@ -1166,6 +1171,7 @@ void testGetClientInfoException(String query) throws SQLException { |
1166 | 1171 | equalTo( |
1167 | 1172 | DB_CONNECTION_STRING, |
1168 | 1173 | emitStableDatabaseSemconv() ? null : "testdb://localhost"), |
| 1174 | + equalTo(PEER_SERVICE, hasPeerService() ? "test-peer-service" : null), |
1169 | 1175 | equalTo(SERVER_ADDRESS, "localhost")))); |
1170 | 1176 | } |
1171 | 1177 |
|
@@ -1254,6 +1260,7 @@ void testProduceProperSpanName( |
1254 | 1260 | equalTo(maybeStable(DB_STATEMENT), sanitizedQuery), |
1255 | 1261 | equalTo(maybeStable(DB_OPERATION), operation), |
1256 | 1262 | equalTo(maybeStable(DB_SQL_TABLE), table), |
| 1263 | + equalTo(PEER_SERVICE, hasPeerService() ? "test-peer-service" : null), |
1257 | 1264 | equalTo(SERVER_ADDRESS, "localhost")))); |
1258 | 1265 | } |
1259 | 1266 |
|
@@ -1379,6 +1386,7 @@ void testHandleRecursiveStatements( |
1379 | 1386 | equalTo(maybeStable(DB_STATEMENT), "SELECT * FROM table"), |
1380 | 1387 | equalTo(maybeStable(DB_OPERATION), "SELECT"), |
1381 | 1388 | equalTo(maybeStable(DB_SQL_TABLE), "table"), |
| 1389 | + equalTo(PEER_SERVICE, hasPeerService() ? "test-peer-service" : null), |
1382 | 1390 | equalTo(SERVER_ADDRESS, "localhost")))); |
1383 | 1391 | } |
1384 | 1392 |
|
|
0 commit comments