honeymoose
/
OpenSearch
mirror of https://github.com/honeymoose/OpenSearch.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove LocalEsCluster 

We aren't using it anymore.

Original commit: elastic/x-pack-elasticsearch@67f47c4f11
This commit is contained in:
Nik Everett 2017-07-05 08:21:35 -04:00
parent 9d83eccf27
commit 18bc094dbc
1 changed files with 0 additions and 127 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

127
sql/jdbc/src/test/java/org/elasticsearch/xpack/sql/jdbc/framework/LocalEsCluster.java
View File

@ -1,127 +0,0 @@
/*
* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
* or more contributor license agreements. Licensed under the Elastic License;
* you may not use this file except in compliance with the Elastic License.
*/
package org.elasticsearch.xpack.sql.jdbc.framework;
import org.elasticsearch.client.RestClient;
import org.elasticsearch.cluster.routing.allocation.DiskThresholdSettings;
import org.elasticsearch.common.CheckedSupplier;
import org.elasticsearch.common.network.NetworkModule;
import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.painless.PainlessPlugin;
import org.elasticsearch.plugins.Plugin;
import org.elasticsearch.test.InternalTestCluster;
import org.elasticsearch.test.NodeConfigurationSource;
import org.elasticsearch.transport.Netty4Plugin;
import org.elasticsearch.xpack.XPackPlugin;
import org.elasticsearch.xpack.sql.jdbc.util.IOUtils;
import org.elasticsearch.xpack.sql.net.client.SuppressForbidden;
import org.elasticsearch.xpack.sql.util.StringUtils;
import org.junit.rules.ExternalResource;
import java.net.InetSocketAddress;
import java.nio.file.Path;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Arrays;
import java.util.Collection;
import java.util.function.Function;
import static java.util.Collections.emptySet;
import static org.apache.lucene.util.LuceneTestCase.createTempDir;
import static org.apache.lucene.util.LuceneTestCase.random;
import static org.elasticsearch.test.ESTestCase.randomBoolean;
import static org.elasticsearch.test.ESTestCase.randomLong;
/**
* Hack to run an {@link InternalTestCluster} if this is being run
* in an environment without {@code tests.rest.cluster} set for easier
* debugging. Note that this doesn't work in the security manager is
* enabled.
*/
public class LocalEsCluster extends ExternalResource implements CheckedSupplier<Connection, SQLException> {
private InternalTestCluster internalTestCluster;
private RestClient client;
private String serverAddress = StringUtils.EMPTY;
@Override
@SuppressForbidden(reason = "it is a hack anyway")
protected void before() throws Throwable {
long seed = randomLong();
String name = InternalTestCluster.clusterName("", seed);
NodeConfigurationSource config = new NodeConfigurationSource() {
@Override
public Settings nodeSettings(int nodeOrdinal) {
Settings.Builder builder = Settings.builder()
// Enable http because the tests use it
.put(NetworkModule.HTTP_ENABLED.getKey(), true)
.put(NetworkModule.HTTP_TYPE_KEY, Netty4Plugin.NETTY_HTTP_TRANSPORT_NAME)
// Default the watermarks to absurdly low to prevent the tests
// from failing on nodes without enough disk space
.put(DiskThresholdSettings.CLUSTER_ROUTING_ALLOCATION_LOW_DISK_WATERMARK_SETTING.getKey(), "1b")
.put(DiskThresholdSettings.CLUSTER_ROUTING_ALLOCATION_HIGH_DISK_WATERMARK_SETTING.getKey(), "1b")
// Mimic settings in build.gradle so we're closer to real
.put("xpack.security.enabled", false)
.put("xpack.monitoring.enabled", false)
.put("xpack.ml.enabled", false)
.put("xpack.watcher.enabled", false);
return builder.build();
}
@Override
public Path nodeConfigPath(int nodeOrdinal) {
return null;
}
@Override
public Collection<Class<? extends Plugin>> nodePlugins() {
// Use netty4 plugin to enable rest
return Arrays.asList(Netty4Plugin.class, XPackPlugin.class, PainlessPlugin.class);
}
};
internalTestCluster = new InternalTestCluster(seed, createTempDir(), false, true, 1, 1, name, config, 0, randomBoolean(), "", emptySet(), Function.identity());
internalTestCluster.beforeTest(random(), 0);
internalTestCluster.ensureAtLeastNumDataNodes(1);
InetSocketAddress httpBound = internalTestCluster.httpAddresses()[0];
String http = httpBound.getHostString() + ":" + httpBound.getPort();
try {
System.setProperty("tests.rest.cluster", http);
} catch (SecurityException e) {
throw new RuntimeException("Failed to set system property required for tests. Security manager must be disabled to use this hack.", e);
}
client = TestUtils.restClient(httpBound.getAddress());
// load data
TestUtils.loadDatasetInEs(client);
serverAddress = httpBound.getAddress().getHostAddress();
}
@Override
protected void after() {
serverAddress = StringUtils.EMPTY;
if (internalTestCluster == null) {
return;
}
IOUtils.close(client);
IOUtils.close(internalTestCluster);
}
public RestClient client() {
return client;
}
public String address() {
return serverAddress;
}
@Override
public Connection get() throws SQLException {
return DriverManager.getConnection("jdbc:es://" + serverAddress);
}
}