107 lines
3.4 KiB
Groovy
107 lines
3.4 KiB
Groovy
/*
|
|
* Copyright The OpenTelemetry Authors
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import static io.opentelemetry.api.trace.Span.Kind.CLIENT
|
|
import static org.elasticsearch.cluster.ClusterName.CLUSTER_NAME_SETTING
|
|
|
|
import groovy.json.JsonSlurper
|
|
import io.opentelemetry.api.trace.attributes.SemanticAttributes
|
|
import io.opentelemetry.instrumentation.test.AgentTestRunner
|
|
import org.apache.http.HttpHost
|
|
import org.apache.http.client.config.RequestConfig
|
|
import org.apache.http.util.EntityUtils
|
|
import org.elasticsearch.client.Response
|
|
import org.elasticsearch.client.RestClient
|
|
import org.elasticsearch.client.RestClientBuilder
|
|
import org.elasticsearch.common.io.FileSystemUtils
|
|
import org.elasticsearch.common.settings.Settings
|
|
import org.elasticsearch.common.transport.TransportAddress
|
|
import org.elasticsearch.env.Environment
|
|
import org.elasticsearch.http.HttpServerTransport
|
|
import org.elasticsearch.node.Node
|
|
import org.elasticsearch.node.internal.InternalSettingsPreparer
|
|
import org.elasticsearch.transport.Netty3Plugin
|
|
import spock.lang.Shared
|
|
|
|
class Elasticsearch5RestClientTest extends AgentTestRunner {
|
|
@Shared
|
|
TransportAddress httpTransportAddress
|
|
@Shared
|
|
Node testNode
|
|
@Shared
|
|
File esWorkingDir
|
|
@Shared
|
|
String clusterName = UUID.randomUUID().toString()
|
|
|
|
@Shared
|
|
static RestClient client
|
|
|
|
def setupSpec() {
|
|
|
|
esWorkingDir = File.createTempDir("test-es-working-dir-", "")
|
|
esWorkingDir.deleteOnExit()
|
|
println "ES work dir: $esWorkingDir"
|
|
|
|
def settings = Settings.builder()
|
|
.put("path.home", esWorkingDir.path)
|
|
.put("transport.type", "netty3")
|
|
.put("http.type", "netty3")
|
|
.put(CLUSTER_NAME_SETTING.getKey(), clusterName)
|
|
.build()
|
|
testNode = new Node(new Environment(InternalSettingsPreparer.prepareSettings(settings)), [Netty3Plugin])
|
|
testNode.start()
|
|
httpTransportAddress = testNode.injector().getInstance(HttpServerTransport).boundAddress().publishAddress()
|
|
|
|
client = RestClient.builder(new HttpHost(httpTransportAddress.address, httpTransportAddress.port))
|
|
.setMaxRetryTimeoutMillis(Integer.MAX_VALUE)
|
|
.setRequestConfigCallback(new RestClientBuilder.RequestConfigCallback() {
|
|
@Override
|
|
RequestConfig.Builder customizeRequestConfig(RequestConfig.Builder builder) {
|
|
return builder.setConnectTimeout(Integer.MAX_VALUE).setSocketTimeout(Integer.MAX_VALUE)
|
|
}
|
|
})
|
|
.build()
|
|
|
|
}
|
|
|
|
def cleanupSpec() {
|
|
testNode?.close()
|
|
if (esWorkingDir != null) {
|
|
FileSystemUtils.deleteSubDirectories(esWorkingDir.toPath())
|
|
esWorkingDir.delete()
|
|
}
|
|
}
|
|
|
|
def "test elasticsearch status"() {
|
|
setup:
|
|
Response response = client.performRequest("GET", "_cluster/health")
|
|
|
|
Map result = new JsonSlurper().parseText(EntityUtils.toString(response.entity))
|
|
|
|
expect:
|
|
result.status == "green"
|
|
|
|
assertTraces(1) {
|
|
trace(0, 1) {
|
|
span(0) {
|
|
name "GET _cluster/health"
|
|
kind CLIENT
|
|
hasNoParent()
|
|
attributes {
|
|
"${SemanticAttributes.DB_SYSTEM.key()}" "elasticsearch"
|
|
"${SemanticAttributes.DB_OPERATION.key()}" "GET _cluster/health"
|
|
"${SemanticAttributes.NET_PEER_NAME.key()}" httpTransportAddress.address
|
|
"${SemanticAttributes.NET_PEER_PORT.key()}" httpTransportAddress.port
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
String expectedOperationName(String method) {
|
|
return method != null ? "HTTP $method" : "HTTP request"
|
|
}
|
|
}
|