Test: NetworkPartitionTests should make sure MockTransportService is …

…used.
elastic · Jun 15, 2015 · 2fb8df0 · 2fb8df0
1 parent e559eaf
commit 2fb8df0
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/core/src/test/java/org/elasticsearch/test/disruption/NetworkPartitionTests.java b/core/src/test/java/org/elasticsearch/test/disruption/NetworkPartitionTests.java
@@ -22,14 +22,24 @@
 
 
 import org.apache.lucene.util.LuceneTestCase;
+import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.test.ElasticsearchIntegrationTest;
 import org.elasticsearch.test.InternalTestCluster;
+import org.elasticsearch.test.transport.MockTransportService;
+import org.elasticsearch.transport.TransportModule;
 import org.junit.Test;
 
 import java.io.IOException;
 
 @LuceneTestCase.Slow
-public class NetworkPartitionTests extends ElasticsearchIntegrationTest{
+public class NetworkPartitionTests extends ElasticsearchIntegrationTest {
+
+    @Override
+    protected Settings nodeSettings(int nodeOrdinal) {
+        return Settings.builder()
+                .put(TransportModule.TRANSPORT_SERVICE_TYPE_KEY, MockTransportService.class.getName())
+                .build();
+    }
 
     @Test
     public void testNetworkPartitionWithNodeShutdown() throws IOException {