92 lines
2.5 KiB
Groovy
92 lines
2.5 KiB
Groovy
/*
|
|
* Copyright 2020, OpenTelemetry Authors
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
import io.opentelemetry.auto.test.base.HttpClientTest
|
|
import org.apache.http.HttpResponse
|
|
import org.apache.http.client.config.RequestConfig
|
|
import org.apache.http.concurrent.FutureCallback
|
|
import org.apache.http.impl.nio.client.HttpAsyncClients
|
|
import org.apache.http.message.BasicHeader
|
|
import spock.lang.AutoCleanup
|
|
import spock.lang.Shared
|
|
import spock.lang.Timeout
|
|
|
|
import java.util.concurrent.CountDownLatch
|
|
|
|
@Timeout(5)
|
|
class ApacheHttpAsyncClientTest extends HttpClientTest {
|
|
|
|
@Shared
|
|
RequestConfig requestConfig = RequestConfig.custom()
|
|
.setConnectTimeout(CONNECT_TIMEOUT_MS)
|
|
.setSocketTimeout(READ_TIMEOUT_MS)
|
|
.build()
|
|
|
|
@AutoCleanup
|
|
@Shared
|
|
def client = HttpAsyncClients.custom().setDefaultRequestConfig(requestConfig).build()
|
|
|
|
def setupSpec() {
|
|
client.start()
|
|
}
|
|
|
|
@Override
|
|
int doRequest(String method, URI uri, Map<String, String> headers, Closure callback) {
|
|
def request = new HttpUriRequest(method, uri)
|
|
headers.entrySet().each {
|
|
request.addHeader(new BasicHeader(it.key, it.value))
|
|
}
|
|
|
|
def latch = callback == null ? null : new CountDownLatch(1)
|
|
def handler = callback == null ? null : new FutureCallback<HttpResponse>() {
|
|
|
|
@Override
|
|
void completed(HttpResponse result) {
|
|
callback()
|
|
latch.countDown()
|
|
}
|
|
|
|
@Override
|
|
void failed(Exception ex) {
|
|
latch.countDown()
|
|
}
|
|
|
|
@Override
|
|
void cancelled() {
|
|
latch.countDown()
|
|
}
|
|
}
|
|
|
|
def future = client.execute(request, handler)
|
|
def response = future.get()
|
|
response.entity?.content?.close() // Make sure the connection is closed.
|
|
if (callback != null) {
|
|
// need to wait for callback to complete in case test is expecting span from it
|
|
latch.await()
|
|
}
|
|
response.statusLine.statusCode
|
|
}
|
|
|
|
@Override
|
|
Integer statusOnRedirectError() {
|
|
return 302
|
|
}
|
|
|
|
@Override
|
|
boolean testRemoteConnection() {
|
|
false // otherwise SocketTimeoutException for https requests
|
|
}
|
|
}
|