Skip to content

Commit b95d849

Browse files
committed
address review comment
1 parent ce63587 commit b95d849

File tree

4 files changed

+34
-30
lines changed

4 files changed

+34
-30
lines changed

instrumentation/ktor/ktor-2.0/testing/src/main/kotlin/io/opentelemetry/instrumentation/ktor/v2_0/client/AbstractKtorHttpClientTest.kt

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import io.opentelemetry.instrumentation.testing.junit.http.HttpClientTestOptions
1818
import io.opentelemetry.instrumentation.testing.junit.http.HttpClientTestOptions.DEFAULT_HTTP_ATTRIBUTES
1919
import io.opentelemetry.semconv.NetworkAttributes
2020
import kotlinx.coroutines.*
21+
import org.junit.jupiter.api.AfterAll
2122
import java.net.URI
2223

2324
abstract class AbstractKtorHttpClientTest : AbstractHttpClientTest<HttpRequestBuilder>() {
@@ -27,6 +28,19 @@ abstract class AbstractKtorHttpClientTest : AbstractHttpClientTest<HttpRequestBu
2728

2829
installTracing()
2930
}
31+
private val singleConnectionClient = HttpClient(CIO) {
32+
engine {
33+
maxConnectionsCount = 1
34+
}
35+
36+
installTracing()
37+
}
38+
39+
@AfterAll
40+
fun tearDown() {
41+
client.close()
42+
singleConnectionClient.close()
43+
}
3044

3145
abstract fun HttpClientConfig<*>.installTracing()
3246

@@ -67,7 +81,7 @@ abstract class AbstractKtorHttpClientTest : AbstractHttpClientTest<HttpRequestBu
6781
setHttpAttributes { DEFAULT_HTTP_ATTRIBUTES - setOf(NetworkAttributes.NETWORK_PROTOCOL_VERSION) }
6882

6983
setSingleConnectionFactory { host, port ->
70-
KtorHttpClientSingleConnection(host, port) { installTracing() }
84+
KtorHttpClientSingleConnection(singleConnectionClient, host, port)
7185
}
7286
}
7387
}

instrumentation/ktor/ktor-2.0/testing/src/main/kotlin/io/opentelemetry/instrumentation/ktor/v2_0/client/KtorHttpClientSingleConnection.kt

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -12,23 +12,11 @@ import io.opentelemetry.instrumentation.testing.junit.http.SingleConnection
1212
import kotlinx.coroutines.runBlocking
1313

1414
class KtorHttpClientSingleConnection(
15+
private val client: HttpClient,
1516
private val host: String,
16-
private val port: Int,
17-
private val installTracing: HttpClientConfig<*>.() -> Unit,
17+
private val port: Int
1818
) : SingleConnection {
1919

20-
private val client: HttpClient
21-
22-
init {
23-
val engine = CIO.create {
24-
maxConnectionsCount = 1
25-
}
26-
27-
client = HttpClient(engine) {
28-
installTracing()
29-
}
30-
}
31-
3220
override fun doRequest(path: String, requestHeaders: MutableMap<String, String>) = runBlocking {
3321
val request = HttpRequestBuilder(
3422
scheme = "http",

instrumentation/ktor/ktor-3.0/testing/src/main/kotlin/io/opentelemetry/instrumentation/ktor/v3_0/client/AbstractKtorHttpClientTest.kt

Lines changed: 15 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ import io.opentelemetry.instrumentation.testing.junit.http.HttpClientTestOptions
1818
import io.opentelemetry.instrumentation.testing.junit.http.HttpClientTestOptions.DEFAULT_HTTP_ATTRIBUTES
1919
import io.opentelemetry.semconv.NetworkAttributes
2020
import kotlinx.coroutines.*
21+
import org.junit.jupiter.api.AfterAll
2122
import java.net.URI
2223

2324
abstract class AbstractKtorHttpClientTest : AbstractHttpClientTest<HttpRequestBuilder>() {
@@ -27,6 +28,19 @@ abstract class AbstractKtorHttpClientTest : AbstractHttpClientTest<HttpRequestBu
2728

2829
installTracing()
2930
}
31+
private val singleConnectionClient = HttpClient(CIO) {
32+
engine {
33+
maxConnectionsCount = 1
34+
}
35+
36+
installTracing()
37+
}
38+
39+
@AfterAll
40+
fun tearDown() {
41+
client.close()
42+
singleConnectionClient.close()
43+
}
3044

3145
abstract fun HttpClientConfig<*>.installTracing()
3246

@@ -67,7 +81,7 @@ abstract class AbstractKtorHttpClientTest : AbstractHttpClientTest<HttpRequestBu
6781
setHttpAttributes { DEFAULT_HTTP_ATTRIBUTES - setOf(NetworkAttributes.NETWORK_PROTOCOL_VERSION) }
6882

6983
setSingleConnectionFactory { host, port ->
70-
KtorHttpClientSingleConnection(host, port) { installTracing() }
84+
KtorHttpClientSingleConnection(singleConnectionClient, host, port)
7185
}
7286
}
7387
}

instrumentation/ktor/ktor-3.0/testing/src/main/kotlin/io/opentelemetry/instrumentation/ktor/v3_0/client/KtorHttpClientSingleConnection.kt

Lines changed: 2 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -12,23 +12,11 @@ import io.opentelemetry.instrumentation.testing.junit.http.SingleConnection
1212
import kotlinx.coroutines.runBlocking
1313

1414
class KtorHttpClientSingleConnection(
15+
private val client: HttpClient,
1516
private val host: String,
16-
private val port: Int,
17-
private val installTracing: HttpClientConfig<*>.() -> Unit,
17+
private val port: Int
1818
) : SingleConnection {
1919

20-
private val client: HttpClient
21-
22-
init {
23-
val engine = CIO.create {
24-
maxConnectionsCount = 1
25-
}
26-
27-
client = HttpClient(engine) {
28-
installTracing()
29-
}
30-
}
31-
3220
override fun doRequest(path: String, requestHeaders: MutableMap<String, String>) = runBlocking {
3321
val request = HttpRequestBuilder(
3422
scheme = "http",

0 commit comments

Comments
 (0)