Retry validation on NodeNotConnectedException

elastic · Jun 26, 2018 · 9ea4166 · 9ea4166
1 parent d1e43c6
commit 9ea4166
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/server/src/test/java/org/elasticsearch/discovery/ClusterDisruptionIT.java b/server/src/test/java/org/elasticsearch/discovery/ClusterDisruptionIT.java
@@ -47,6 +47,7 @@
 import org.elasticsearch.test.disruption.NetworkDisruption.TwoPartitions;
 import org.elasticsearch.test.disruption.ServiceDisruptionScheme;
 import org.elasticsearch.test.junit.annotations.TestLogging;
+import org.elasticsearch.transport.NodeNotConnectedException;
 
 import java.util.ArrayList;
 import java.util.Collections;
@@ -207,7 +208,7 @@ public void testAckedIndexing() throws Exception {
                                 assertTrue("doc [" + id + "] indexed via node [" + ackedDocs.get(id) + "] not found",
                                     client(node).prepareGet("test", "type", id).setPreference("_local").get().isExists());
                             }
-                        } catch (AssertionError | NoShardAvailableActionException e) {
+                        } catch (AssertionError | NoShardAvailableActionException | NodeNotConnectedException e) {
                             throw new AssertionError(e.getMessage() + " (checked via node [" + node + "]", e);
                         }
                     }