diff --git a/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2NodeClientTest.groovy b/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2NodeClientTest.groovy index a56e5f0454..20d2883c8a 100644 --- a/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2NodeClientTest.groovy +++ b/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2NodeClientTest.groovy @@ -40,6 +40,7 @@ class Elasticsearch2NodeClientTest extends AgentTestRunner { testNode = NodeBuilder.newInstance().clusterName("test-cluster").settings(settings).build() testNode.start() testNode.client().admin().cluster().prepareHealth().setWaitForYellowStatus().execute().actionGet(5000) + TEST_WRITER.waitForTraces(1) } def cleanupSpec() { diff --git a/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2TransportClientTest.groovy b/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2TransportClientTest.groovy index f6be522c8e..a284f44921 100644 --- a/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2TransportClientTest.groovy +++ b/dd-java-agent/instrumentation/elasticsearch-transport-2/src/test/groovy/Elasticsearch2TransportClientTest.groovy @@ -51,6 +51,7 @@ class Elasticsearch2TransportClientTest extends AgentTestRunner { ).build() client.addTransportAddress(new InetSocketTransportAddress(InetAddress.getByName("localhost"), TCP_PORT)) client.admin().cluster().prepareHealth().setWaitForYellowStatus().execute().actionGet(5000) + TEST_WRITER.waitForTraces(1) } def cleanupSpec() { diff --git a/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5NodeClientTest.groovy b/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5NodeClientTest.groovy index 9c60abd8ed..4da1acfbbb 100644 --- a/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5NodeClientTest.groovy +++ b/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5NodeClientTest.groovy @@ -46,6 +46,7 @@ class Elasticsearch5NodeClientTest extends AgentTestRunner { testNode = new Node(new Environment(InternalSettingsPreparer.prepareSettings(settings)), [Netty3Plugin]) testNode.start() testNode.client().admin().cluster().prepareHealth().setWaitForYellowStatus().execute().actionGet(5000) + TEST_WRITER.waitForTraces(1) } def cleanupSpec() { diff --git a/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5TransportClientTest.groovy b/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5TransportClientTest.groovy index 6e758ef24d..3bc49699c9 100644 --- a/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5TransportClientTest.groovy +++ b/dd-java-agent/instrumentation/elasticsearch-transport-5/src/test/groovy/Elasticsearch5TransportClientTest.groovy @@ -58,6 +58,7 @@ class Elasticsearch5TransportClientTest extends AgentTestRunner { ) client.addTransportAddress(new InetSocketTransportAddress(InetAddress.getByName("localhost"), TCP_PORT)) client.admin().cluster().prepareHealth().setWaitForYellowStatus().execute().actionGet(5000) + TEST_WRITER.waitForTraces(1) } def cleanupSpec() { diff --git a/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6NodeClientTest.groovy b/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6NodeClientTest.groovy index aa45d596b7..2b69ab27db 100644 --- a/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6NodeClientTest.groovy +++ b/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6NodeClientTest.groovy @@ -43,6 +43,7 @@ class Elasticsearch6NodeClientTest extends AgentTestRunner { testNode = new Node(InternalSettingsPreparer.prepareEnvironment(settings, null), [Netty4Plugin]) testNode.start() testNode.client().admin().cluster().prepareHealth().setWaitForYellowStatus().execute().actionGet(5000) + TEST_WRITER.waitForTraces(1) } def cleanupSpec() { diff --git a/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6TransportClientTest.groovy b/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6TransportClientTest.groovy index dab5afcfe5..0889db29a3 100644 --- a/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6TransportClientTest.groovy +++ b/dd-java-agent/instrumentation/elasticsearch-transport-6/src/test/groovy/Elasticsearch6TransportClientTest.groovy @@ -55,6 +55,7 @@ class Elasticsearch6TransportClientTest extends AgentTestRunner { ) client.addTransportAddress(new TransportAddress(InetAddress.getByName("localhost"), TCP_PORT)) client.admin().cluster().prepareHealth().setWaitForYellowStatus().execute().actionGet(5000) + TEST_WRITER.waitForTraces(1) } def cleanupSpec() {