Skip to content

Commit 26dcbcc

Browse files
authored
Remove MockTcpTransport for ESIntegTestCase (#36089)
This commit removes the `MockTcpTransport` as a transport option for `ESIntegTestCase`. It is the first step in replacing the usages of `MockTcpTransport` with `MockNioTransport`.
1 parent da100c5 commit 26dcbcc

File tree

6 files changed

+10
-92
lines changed

6 files changed

+10
-92
lines changed

qa/smoke-test-client/src/test/java/org/elasticsearch/smoketest/ESSmokeClientTestCase.java

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,8 @@
1919

2020
package org.elasticsearch.smoketest;
2121

22-
import org.apache.logging.log4j.Logger;
2322
import org.apache.logging.log4j.LogManager;
23+
import org.apache.logging.log4j.Logger;
2424
import org.apache.lucene.util.LuceneTestCase;
2525
import org.elasticsearch.action.admin.cluster.health.ClusterHealthResponse;
2626
import org.elasticsearch.client.Client;
@@ -30,7 +30,6 @@
3030
import org.elasticsearch.common.transport.TransportAddress;
3131
import org.elasticsearch.env.Environment;
3232
import org.elasticsearch.plugins.Plugin;
33-
import org.elasticsearch.transport.MockTcpTransportPlugin;
3433
import org.elasticsearch.transport.client.PreBuiltTransportClient;
3534
import org.elasticsearch.transport.nio.MockNioTransportPlugin;
3635
import org.junit.After;
@@ -82,16 +81,8 @@ private static Client startClient(Path tempDir, TransportAddress... transportAdd
8281
.put("client.transport.ignore_cluster_name", true)
8382
.put(Environment.PATH_HOME_SETTING.getKey(), tempDir);
8483
final Collection<Class<? extends Plugin>> plugins;
85-
boolean usNio = random().nextBoolean();
86-
String transportKey;
87-
Class<? extends Plugin> transportPlugin;
88-
if (usNio) {
89-
transportKey = MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME;
90-
transportPlugin = MockNioTransportPlugin.class;
91-
} else {
92-
transportKey = MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME;
93-
transportPlugin = MockTcpTransportPlugin.class;
94-
}
84+
String transportKey = MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME;
85+
Class<? extends Plugin> transportPlugin = MockNioTransportPlugin.class;
9586
if (random().nextBoolean()) {
9687
builder.put(NetworkModule.TRANSPORT_TYPE_KEY, transportKey);
9788
plugins = Collections.singleton(transportPlugin);

qa/smoke-test-http/src/test/java/org/elasticsearch/http/HttpSmokeTestCase.java

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@
2222
import org.elasticsearch.common.settings.Settings;
2323
import org.elasticsearch.plugins.Plugin;
2424
import org.elasticsearch.test.ESIntegTestCase;
25-
import org.elasticsearch.transport.MockTcpTransportPlugin;
2625
import org.elasticsearch.transport.Netty4Plugin;
2726
import org.elasticsearch.transport.nio.MockNioTransportPlugin;
2827
import org.elasticsearch.transport.nio.NioTransportPlugin;
@@ -46,9 +45,7 @@ public static void setUpTransport() {
4645
}
4746

4847
private static String getTypeKey(Class<? extends Plugin> clazz) {
49-
if (clazz.equals(MockTcpTransportPlugin.class)) {
50-
return MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME;
51-
} else if (clazz.equals(MockNioTransportPlugin.class)) {
48+
if (clazz.equals(MockNioTransportPlugin.class)) {
5249
return MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME;
5350
} else if (clazz.equals(NioTransportPlugin.class)) {
5451
return NioTransportPlugin.NIO_TRANSPORT_NAME;

test/framework/src/main/java/org/elasticsearch/test/ESTestCase.java

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,6 @@
114114
import org.elasticsearch.test.junit.listeners.LoggingListener;
115115
import org.elasticsearch.test.junit.listeners.ReproduceInfoPrinter;
116116
import org.elasticsearch.threadpool.ThreadPool;
117-
import org.elasticsearch.transport.MockTcpTransportPlugin;
118117
import org.elasticsearch.transport.nio.MockNioTransportPlugin;
119118
import org.joda.time.DateTimeZone;
120119
import org.junit.After;
@@ -992,19 +991,12 @@ public static String randomGeohash(int minPrecision, int maxPrecision) {
992991
return geohashGenerator.ofStringLength(random(), minPrecision, maxPrecision);
993992
}
994993

995-
private static boolean useNio;
996-
997-
@BeforeClass
998-
public static void setUseNio() throws Exception {
999-
useNio = randomBoolean();
1000-
}
1001-
1002994
public static String getTestTransportType() {
1003-
return useNio ? MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME : MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME;
995+
return MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME;
1004996
}
1005997

1006998
public static Class<? extends Plugin> getTestTransportPlugin() {
1007-
return useNio ? MockNioTransportPlugin.class : MockTcpTransportPlugin.class;
999+
return MockNioTransportPlugin.class;
10081000
}
10091001

10101002
private static final GeohashGenerator geohashGenerator = new GeohashGenerator();

test/framework/src/main/java/org/elasticsearch/test/ExternalTestCluster.java

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@
3434
import org.elasticsearch.common.transport.TransportAddress;
3535
import org.elasticsearch.env.Environment;
3636
import org.elasticsearch.plugins.Plugin;
37-
import org.elasticsearch.transport.MockTcpTransportPlugin;
3837
import org.elasticsearch.transport.MockTransportClient;
3938
import org.elasticsearch.transport.nio.MockNioTransportPlugin;
4039

@@ -84,14 +83,9 @@ public ExternalTestCluster(Path tempDir, Settings additionalSettings, Collection
8483
if (addMockTcpTransport) {
8584
String transport = getTestTransportType();
8685
clientSettingsBuilder.put(NetworkModule.TRANSPORT_TYPE_KEY, transport);
87-
if (pluginClasses.contains(MockTcpTransportPlugin.class) == false &&
88-
pluginClasses.contains(MockNioTransportPlugin.class) == false) {
86+
if (pluginClasses.contains(MockNioTransportPlugin.class) == false) {
8987
pluginClasses = new ArrayList<>(pluginClasses);
90-
if (transport.equals(MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME)) {
91-
pluginClasses.add(MockNioTransportPlugin.class);
92-
} else {
93-
pluginClasses.add(MockTcpTransportPlugin.class);
94-
}
88+
pluginClasses.add(MockNioTransportPlugin.class);
9589
}
9690
}
9791
Settings clientSettings = clientSettingsBuilder.build();

test/framework/src/main/java/org/elasticsearch/transport/MockTcpTransportPlugin.java

Lines changed: 0 additions & 48 deletions
This file was deleted.

test/framework/src/main/java/org/elasticsearch/transport/MockTransportClient.java

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@
3333
@Deprecated
3434
public class MockTransportClient extends TransportClient {
3535
private static final Settings DEFAULT_SETTINGS = Settings.builder().put("transport.type.default",
36-
MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME).build();
36+
MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME).build();
3737

3838
public MockTransportClient(Settings settings, Class<? extends Plugin>... plugins) {
3939
this(settings, Arrays.asList(plugins));
@@ -51,15 +51,7 @@ private static Collection<Class<? extends Plugin>> addMockTransportIfMissing(Set
5151
Collection<Class<? extends Plugin>> plugins) {
5252
boolean settingExists = NetworkModule.TRANSPORT_TYPE_SETTING.exists(settings);
5353
String transportType = NetworkModule.TRANSPORT_TYPE_SETTING.get(settings);
54-
if (settingExists == false || MockTcpTransportPlugin.MOCK_TCP_TRANSPORT_NAME.equals(transportType)) {
55-
if (plugins.contains(MockTcpTransportPlugin.class)) {
56-
return plugins;
57-
} else {
58-
plugins = new ArrayList<>(plugins);
59-
plugins.add(MockTcpTransportPlugin.class);
60-
return plugins;
61-
}
62-
} else if (MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME.equals(transportType)) {
54+
if (settingExists == false || MockNioTransportPlugin.MOCK_NIO_TRANSPORT_NAME.equals(transportType)) {
6355
if (plugins.contains(MockNioTransportPlugin.class)) {
6456
return plugins;
6557
} else {

0 commit comments

Comments
 (0)