Enable read timeout test for HttpURLConnection (open-telemetry#4699)

RashmiRam · May 23, 2022 · 97f4fd1 · 97f4fd1
1 parent 193b109
commit 97f4fd1
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/instrumentation/http-url-connection/javaagent/src/test/groovy/HttpUrlConnectionTest.groovy b/instrumentation/http-url-connection/javaagent/src/test/groovy/HttpUrlConnectionTest.groovy
@@ -27,6 +27,9 @@ class HttpUrlConnectionTest extends HttpClientTest<HttpURLConnection> implements
 
   @Override
   int sendRequest(HttpURLConnection connection, String method, URI uri, Map<String, String> headers) {
+    if (uri.toString().contains("/read-timeout")) {
+      connection.readTimeout = READ_TIMEOUT_MS
+    }
     try {
       connection.setRequestMethod(method)
       headers.each { connection.setRequestProperty(it.key, it.value) }
@@ -65,6 +68,11 @@ class HttpUrlConnectionTest extends HttpClientTest<HttpURLConnection> implements
     return false
   }
 
+  @Override
+  boolean testReadTimeout() {
+    true
+  }
+
   @Unroll
   def "trace request (useCaches: #useCaches)"() {
     setup: