Skip to content

Commit 0218ce9

Browse files
committed
Revert "RUM-9899: fix test compilation(30 fallen): removing drop methods"
This reverts commit e837968.
1 parent 412d91c commit 0218ce9

File tree

3 files changed

+26
-1
lines changed

3 files changed

+26
-1
lines changed

integrations/dd-sdk-android-okhttp/src/main/kotlin/com/datadog/android/okhttp/trace/TracingInterceptor.kt

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -608,7 +608,15 @@ internal constructor(
608608
}
609609

610610
private fun Span.finishRumAware(isSampled: Boolean) {
611-
if (canSendSpan() && isSampled) finish()
611+
if (canSendSpan()) {
612+
if (isSampled) finish() else drop()
613+
} else {
614+
drop()
615+
}
616+
}
617+
618+
private fun Span.drop() {
619+
// TODO (this as? MutableSpan)?.drop()
612620
}
613621

614622
private fun Span.sample(request: Request): Boolean {

integrations/dd-sdk-android-okhttp/src/test/kotlin/com/datadog/android/okhttp/DatadogInterceptorWithoutTracesTest.kt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -367,6 +367,8 @@ internal class DatadogInterceptorWithoutTracesTest {
367367
val response = testedInterceptor.intercept(mockChain)
368368

369369
verify(mockSpanBuilder).withOrigin(DatadogInterceptor.ORIGIN_RUM)
370+
//// TODO - fix drop method
371+
// verify(mockSpan).drop()
370372
assertThat(response).isSameAs(fakeResponse)
371373
}
372374

@@ -382,6 +384,8 @@ internal class DatadogInterceptorWithoutTracesTest {
382384
verify(mockSpanBuilder).withOrigin(DatadogInterceptor.ORIGIN_RUM)
383385
verify(mockSpan as MutableSpan).setResourceName(fakeUrl.lowercase(Locale.US))
384386
verify(mockSpan as MutableSpan).setError(true)
387+
// TODO - fix drop method
388+
// verify(mockSpan).drop()
385389
assertThat(response).isSameAs(fakeResponse)
386390
}
387391

integrations/dd-sdk-android-okhttp/src/test/kotlin/com/datadog/android/okhttp/trace/TracingInterceptorNotSendingSpanTest.kt

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ import com.datadog.android.okhttp.utils.config.GlobalRumMonitorTestConfiguration
2020
import com.datadog.android.okhttp.utils.verifyLog
2121
import com.datadog.android.rum.RumResourceMethod
2222
import com.datadog.android.trace.TracingHeaderType
23+
import com.datadog.legacy.trace.api.interceptor.MutableSpan
2324
import com.datadog.legacy.trace.api.sampling.PrioritySampling
2425
import com.datadog.tools.unit.annotations.TestConfigurationsProvider
2526
import com.datadog.tools.unit.extensions.TestConfigurationExtension
@@ -103,6 +104,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
103104
@Mock
104105
lateinit var mockSpanContext: SpanContext
105106

107+
@Mock(extraInterfaces = [MutableSpan::class])
106108
lateinit var mockSpan: Span
107109

108110
@Mock
@@ -919,6 +921,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
919921
verify(mockSpan).setTag("http.status_code", statusCode)
920922
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
921923
verify(mockSpan, never()).finish()
924+
verify(mockSpan as MutableSpan).drop()
922925
assertThat(response).isSameAs(fakeResponse)
923926
}
924927

@@ -939,6 +942,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
939942
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
940943
verify(mockSpan).setError(true)
941944
verify(mockSpan, never()).finish()
945+
verify(mockSpan as MutableSpan).drop()
942946
assertThat(response).isSameAs(fakeResponse)
943947
}
944948

@@ -959,6 +963,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
959963
verify(mockSpan, never()).setError(true)
960964
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
961965
verify(mockSpan, never()).finish()
966+
verify(mockSpan as MutableSpan).drop()
962967
assertThat(response).isSameAs(fakeResponse)
963968
}
964969

@@ -981,6 +986,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
981986
verify(mockSpan, never()).setResourceName(TracingInterceptor.RESOURCE_NAME_404)
982987
}
983988
verify(mockSpan, never()).finish()
989+
verify(mockSpan as MutableSpan).drop()
984990
assertThat(response).isSameAs(fakeResponse)
985991
}
986992

@@ -1004,6 +1010,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
10041010
verify(mockSpan).setTag("error.stack", throwable.loggableStackTrace())
10051011
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
10061012
verify(mockSpan, never()).finish()
1013+
verify(mockSpan as MutableSpan).drop()
10071014
}
10081015

10091016
@Test
@@ -1052,6 +1059,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
10521059
verify(localSpan).setTag("http.status_code", statusCode)
10531060
verify(localSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
10541061
verify(localSpan, never()).finish()
1062+
verify(localSpan as MutableSpan).drop()
10551063
assertThat(response).isSameAs(fakeResponse)
10561064
mockInternalLogger.verifyLog(
10571065
InternalLogger.Level.WARN,
@@ -1091,13 +1099,15 @@ internal open class TracingInterceptorNotSendingSpanTest {
10911099
verify(localSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
10921100

10931101
verify(localSpan, never()).finish()
1102+
verify(localSpan as MutableSpan).drop()
10941103
verify(mockSpanBuilder).withOrigin(getExpectedOrigin())
10951104
verify(mockSpan).setTag("http.url", fakeUrl)
10961105
verify(mockSpan).setTag("http.method", fakeMethod.name)
10971106
verify(mockSpan).setTag("http.status_code", statusCode)
10981107
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
10991108

11001109
verify(mockSpan, never()).finish()
1110+
verify(mockSpan as MutableSpan).drop()
11011111
assertThat(response1).isSameAs(expectedResponse1)
11021112
assertThat(response2).isSameAs(expectedResponse2)
11031113
mockInternalLogger.verifyLog(
@@ -1133,6 +1143,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
11331143
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
11341144

11351145
verify(mockSpan, never()).finish()
1146+
verify(mockSpan as MutableSpan).drop()
11361147
assertThat(response).isSameAs(fakeResponse)
11371148
}
11381149

@@ -1162,6 +1173,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
11621173
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
11631174

11641175
verify(mockSpan, never()).finish()
1176+
verify(mockSpan as MutableSpan).drop()
11651177
assertThat(response).isSameAs(fakeResponse)
11661178
}
11671179

@@ -1213,6 +1225,7 @@ internal open class TracingInterceptorNotSendingSpanTest {
12131225
verify(mockSpan).setTag(Tags.SPAN_KIND, Tags.SPAN_KIND_CLIENT)
12141226

12151227
verify(mockSpan, never()).finish()
1228+
verify(mockSpan as MutableSpan).drop()
12161229
}
12171230

12181231
@Test

0 commit comments

Comments
 (0)