Skip to content

Commit 043f4f9

Browse files
committed
Simplify test
1 parent c08e617 commit 043f4f9

File tree

1 file changed

+7
-30
lines changed

1 file changed

+7
-30
lines changed

instrumentation/cassandra/cassandra-3.0/javaagent/src/test/java/CassandraClientTest.java

Lines changed: 7 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
import com.datastax.driver.core.ResultSetFuture;
1111
import com.datastax.driver.core.Session;
1212
import io.opentelemetry.api.trace.SpanKind;
13+
import io.opentelemetry.instrumentation.api.internal.SemconvStability;
1314
import io.opentelemetry.instrumentation.testing.junit.AgentInstrumentationExtension;
1415
import io.opentelemetry.instrumentation.testing.junit.InstrumentationExtension;
1516
import io.opentelemetry.instrumentation.testing.junit.db.SemconvStabilityUtil;
@@ -96,11 +97,7 @@ void syncTest(Parameter parameter) {
9697
.hasAttributesSatisfyingExactly(
9798
equalTo(
9899
NetworkAttributes.NETWORK_TYPE,
99-
SemconvStabilityUtil.getAttributeKey(
100-
NetworkAttributes.NETWORK_TYPE)
101-
== null
102-
? null
103-
: "ipv4"),
100+
SemconvStability.emitOldDatabaseSemconv() ? "ipv4" : null),
104101
equalTo(ServerAttributes.SERVER_ADDRESS, cassandraHost),
105102
equalTo(ServerAttributes.SERVER_PORT, cassandraPort),
106103
equalTo(NetworkAttributes.NETWORK_PEER_ADDRESS, cassandraIp),
@@ -119,11 +116,7 @@ void syncTest(Parameter parameter) {
119116
.hasAttributesSatisfyingExactly(
120117
equalTo(
121118
NetworkAttributes.NETWORK_TYPE,
122-
SemconvStabilityUtil.getAttributeKey(
123-
NetworkAttributes.NETWORK_TYPE)
124-
== null
125-
? null
126-
: "ipv4"),
119+
SemconvStability.emitOldDatabaseSemconv() ? "ipv4" : null),
127120
equalTo(ServerAttributes.SERVER_ADDRESS, cassandraHost),
128121
equalTo(ServerAttributes.SERVER_PORT, cassandraPort),
129122
equalTo(NetworkAttributes.NETWORK_PEER_ADDRESS, cassandraIp),
@@ -156,11 +149,7 @@ void syncTest(Parameter parameter) {
156149
.hasAttributesSatisfyingExactly(
157150
equalTo(
158151
NetworkAttributes.NETWORK_TYPE,
159-
SemconvStabilityUtil.getAttributeKey(
160-
NetworkAttributes.NETWORK_TYPE)
161-
== null
162-
? null
163-
: "ipv4"),
152+
SemconvStability.emitOldDatabaseSemconv() ? "ipv4" : null),
164153
equalTo(ServerAttributes.SERVER_ADDRESS, cassandraHost),
165154
equalTo(ServerAttributes.SERVER_PORT, cassandraPort),
166155
equalTo(NetworkAttributes.NETWORK_PEER_ADDRESS, cassandraIp),
@@ -210,11 +199,7 @@ void asyncTest(Parameter parameter) {
210199
.hasAttributesSatisfyingExactly(
211200
equalTo(
212201
NetworkAttributes.NETWORK_TYPE,
213-
SemconvStabilityUtil.getAttributeKey(
214-
NetworkAttributes.NETWORK_TYPE)
215-
== null
216-
? null
217-
: "ipv4"),
202+
SemconvStability.emitOldDatabaseSemconv() ? "ipv4" : null),
218203
equalTo(ServerAttributes.SERVER_ADDRESS, cassandraHost),
219204
equalTo(ServerAttributes.SERVER_PORT, cassandraPort),
220205
equalTo(NetworkAttributes.NETWORK_PEER_ADDRESS, cassandraIp),
@@ -234,11 +219,7 @@ void asyncTest(Parameter parameter) {
234219
.hasAttributesSatisfyingExactly(
235220
equalTo(
236221
NetworkAttributes.NETWORK_TYPE,
237-
SemconvStabilityUtil.getAttributeKey(
238-
NetworkAttributes.NETWORK_TYPE)
239-
== null
240-
? null
241-
: "ipv4"),
222+
SemconvStability.emitOldDatabaseSemconv() ? "ipv4" : null),
242223
equalTo(ServerAttributes.SERVER_ADDRESS, cassandraHost),
243224
equalTo(ServerAttributes.SERVER_PORT, cassandraPort),
244225
equalTo(NetworkAttributes.NETWORK_PEER_ADDRESS, cassandraIp),
@@ -276,11 +257,7 @@ void asyncTest(Parameter parameter) {
276257
.hasAttributesSatisfyingExactly(
277258
equalTo(
278259
NetworkAttributes.NETWORK_TYPE,
279-
SemconvStabilityUtil.getAttributeKey(
280-
NetworkAttributes.NETWORK_TYPE)
281-
== null
282-
? null
283-
: "ipv4"),
260+
SemconvStability.emitOldDatabaseSemconv() ? "ipv4" : null),
284261
equalTo(ServerAttributes.SERVER_ADDRESS, cassandraHost),
285262
equalTo(ServerAttributes.SERVER_PORT, cassandraPort),
286263
equalTo(NetworkAttributes.NETWORK_PEER_ADDRESS, cassandraIp),

0 commit comments

Comments
 (0)