@@ -20,6 +20,7 @@ import com.datadog.android.okhttp.utils.config.GlobalRumMonitorTestConfiguration
20
20
import com.datadog.android.okhttp.utils.verifyLog
21
21
import com.datadog.android.rum.RumResourceMethod
22
22
import com.datadog.android.trace.TracingHeaderType
23
+ import com.datadog.legacy.trace.api.interceptor.MutableSpan
23
24
import com.datadog.legacy.trace.api.sampling.PrioritySampling
24
25
import com.datadog.tools.unit.annotations.TestConfigurationsProvider
25
26
import com.datadog.tools.unit.extensions.TestConfigurationExtension
@@ -103,6 +104,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
103
104
@Mock
104
105
lateinit var mockSpanContext: SpanContext
105
106
107
+ @Mock(extraInterfaces = [MutableSpan ::class ])
106
108
lateinit var mockSpan: Span
107
109
108
110
@Mock
@@ -919,6 +921,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
919
921
verify(mockSpan).setTag(" http.status_code" , statusCode)
920
922
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
921
923
verify(mockSpan, never()).finish()
924
+ verify(mockSpan as MutableSpan ).drop()
922
925
assertThat(response).isSameAs(fakeResponse)
923
926
}
924
927
@@ -939,6 +942,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
939
942
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
940
943
verify(mockSpan).setError(true )
941
944
verify(mockSpan, never()).finish()
945
+ verify(mockSpan as MutableSpan ).drop()
942
946
assertThat(response).isSameAs(fakeResponse)
943
947
}
944
948
@@ -959,6 +963,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
959
963
verify(mockSpan, never()).setError(true )
960
964
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
961
965
verify(mockSpan, never()).finish()
966
+ verify(mockSpan as MutableSpan ).drop()
962
967
assertThat(response).isSameAs(fakeResponse)
963
968
}
964
969
@@ -981,6 +986,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
981
986
verify(mockSpan, never()).setResourceName(TracingInterceptor .RESOURCE_NAME_404 )
982
987
}
983
988
verify(mockSpan, never()).finish()
989
+ verify(mockSpan as MutableSpan ).drop()
984
990
assertThat(response).isSameAs(fakeResponse)
985
991
}
986
992
@@ -1004,6 +1010,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
1004
1010
verify(mockSpan).setTag(" error.stack" , throwable.loggableStackTrace())
1005
1011
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
1006
1012
verify(mockSpan, never()).finish()
1013
+ verify(mockSpan as MutableSpan ).drop()
1007
1014
}
1008
1015
1009
1016
@Test
@@ -1052,6 +1059,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
1052
1059
verify(localSpan).setTag(" http.status_code" , statusCode)
1053
1060
verify(localSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
1054
1061
verify(localSpan, never()).finish()
1062
+ verify(localSpan as MutableSpan ).drop()
1055
1063
assertThat(response).isSameAs(fakeResponse)
1056
1064
mockInternalLogger.verifyLog(
1057
1065
InternalLogger .Level .WARN ,
@@ -1091,13 +1099,15 @@ internal open class TracingInterceptorNotSendingSpanTest {
1091
1099
verify(localSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
1092
1100
1093
1101
verify(localSpan, never()).finish()
1102
+ verify(localSpan as MutableSpan ).drop()
1094
1103
verify(mockSpanBuilder).withOrigin(getExpectedOrigin())
1095
1104
verify(mockSpan).setTag(" http.url" , fakeUrl)
1096
1105
verify(mockSpan).setTag(" http.method" , fakeMethod.name)
1097
1106
verify(mockSpan).setTag(" http.status_code" , statusCode)
1098
1107
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
1099
1108
1100
1109
verify(mockSpan, never()).finish()
1110
+ verify(mockSpan as MutableSpan ).drop()
1101
1111
assertThat(response1).isSameAs(expectedResponse1)
1102
1112
assertThat(response2).isSameAs(expectedResponse2)
1103
1113
mockInternalLogger.verifyLog(
@@ -1133,6 +1143,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
1133
1143
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
1134
1144
1135
1145
verify(mockSpan, never()).finish()
1146
+ verify(mockSpan as MutableSpan ).drop()
1136
1147
assertThat(response).isSameAs(fakeResponse)
1137
1148
}
1138
1149
@@ -1162,6 +1173,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
1162
1173
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
1163
1174
1164
1175
verify(mockSpan, never()).finish()
1176
+ verify(mockSpan as MutableSpan ).drop()
1165
1177
assertThat(response).isSameAs(fakeResponse)
1166
1178
}
1167
1179
@@ -1213,6 +1225,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
1213
1225
verify(mockSpan).setTag(Tags .SPAN_KIND , Tags .SPAN_KIND_CLIENT )
1214
1226
1215
1227
verify(mockSpan, never()).finish()
1228
+ verify(mockSpan as MutableSpan ).drop()
1216
1229
}
1217
1230
1218
1231
@Test
0 commit comments