diff --git a/server/src/test/java/org/elasticsearch/transport/RemoteClusterServiceTests.java b/server/src/test/java/org/elasticsearch/transport/RemoteClusterServiceTests.java index afc915edab0eb..766e0d780531b 100644 --- a/server/src/test/java/org/elasticsearch/transport/RemoteClusterServiceTests.java +++ b/server/src/test/java/org/elasticsearch/transport/RemoteClusterServiceTests.java @@ -741,7 +741,7 @@ public void testCollectSearchShards() throws Exception { AtomicReference failure = new AtomicReference<>(); remoteClusterService.collectSearchShards(IndicesOptions.lenientExpandOpen(), null, null, remoteIndicesByCluster, new LatchedActionListener<>(ActionListener.wrap(response::set, failure::set), latch)); - assertTrue(latch.await(1, TimeUnit.SECONDS)); + assertTrue(latch.await(5, TimeUnit.SECONDS)); assertNull(failure.get()); assertNotNull(response.get()); Map map = response.get(); @@ -760,7 +760,7 @@ public void testCollectSearchShards() throws Exception { remoteClusterService.collectSearchShards(IndicesOptions.lenientExpandOpen(), "index_not_found", null, remoteIndicesByCluster, new LatchedActionListener<>(ActionListener.wrap(response::set, failure::set), latch)); - assertTrue(latch.await(2, TimeUnit.SECONDS)); + assertTrue(latch.await(5, TimeUnit.SECONDS)); assertNull(response.get()); assertNotNull(failure.get()); assertThat(failure.get(), instanceOf(RemoteTransportException.class)); @@ -799,7 +799,7 @@ public void onNodeDisconnected(DiscoveryNode node) { AtomicReference failure = new AtomicReference<>(); remoteClusterService.collectSearchShards(IndicesOptions.lenientExpandOpen(), null, null, remoteIndicesByCluster, new LatchedActionListener<>(ActionListener.wrap(response::set, failure::set), latch)); - assertTrue(latch.await(1, TimeUnit.SECONDS)); + assertTrue(latch.await(5, TimeUnit.SECONDS)); assertNull(response.get()); assertNotNull(failure.get()); assertThat(failure.get(), instanceOf(RemoteTransportException.class)); @@ -817,7 +817,7 @@ public void onNodeDisconnected(DiscoveryNode node) { AtomicReference failure = new AtomicReference<>(); remoteClusterService.collectSearchShards(IndicesOptions.lenientExpandOpen(), null, null, remoteIndicesByCluster, new LatchedActionListener<>(ActionListener.wrap(response::set, failure::set), latch)); - assertTrue(latch.await(1, TimeUnit.SECONDS)); + assertTrue(latch.await(5, TimeUnit.SECONDS)); assertNull(failure.get()); assertNotNull(response.get()); Map map = response.get(); @@ -836,7 +836,7 @@ public void onNodeDisconnected(DiscoveryNode node) { //give transport service enough time to realize that the node is down, and to notify the connection listeners //so that RemoteClusterConnection is left with no connected nodes, hence it will retry connecting next - assertTrue(disconnectedLatch.await(1, TimeUnit.SECONDS)); + assertTrue(disconnectedLatch.await(5, TimeUnit.SECONDS)); service.clearAllRules(); if (randomBoolean()) { @@ -853,7 +853,7 @@ public void onNodeDisconnected(DiscoveryNode node) { AtomicReference failure = new AtomicReference<>(); remoteClusterService.collectSearchShards(IndicesOptions.lenientExpandOpen(), null, null, remoteIndicesByCluster, new LatchedActionListener<>(ActionListener.wrap(response::set, failure::set), latch)); - assertTrue(latch.await(1, TimeUnit.SECONDS)); + assertTrue(latch.await(5, TimeUnit.SECONDS)); assertNull(failure.get()); assertNotNull(response.get()); Map map = response.get(); @@ -862,7 +862,7 @@ public void onNodeDisconnected(DiscoveryNode node) { String clusterAlias = "remote" + i; assertTrue(map.containsKey(clusterAlias)); ClusterSearchShardsResponse shardsResponse = map.get(clusterAlias); - assertTrue(shardsResponse != ClusterSearchShardsResponse.EMPTY); + assertNotSame(ClusterSearchShardsResponse.EMPTY, shardsResponse); } } assertEquals(0, service.getConnectionManager().size());