f34e825217
* stable-2.16: Elasticsearch: Unharcode localhost for container's Change-Id: Id798846df3bc48b4266cec545b2ad607a1a4a06b
91 lines
3.0 KiB
Java
91 lines
3.0 KiB
Java
// Copyright (C) 2018 The Android Open Source Project
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package com.google.gerrit.elasticsearch;
|
|
|
|
import com.google.gerrit.elasticsearch.ElasticTestUtils.ElasticNodeInfo;
|
|
import com.google.gerrit.server.query.change.AbstractQueryChangesTest;
|
|
import com.google.gerrit.testing.ConfigSuite;
|
|
import com.google.gerrit.testing.InMemoryModule;
|
|
import com.google.gerrit.testing.IndexConfig;
|
|
import com.google.inject.Guice;
|
|
import com.google.inject.Injector;
|
|
import org.apache.http.client.methods.HttpPost;
|
|
import org.apache.http.client.protocol.HttpClientContext;
|
|
import org.apache.http.impl.nio.client.CloseableHttpAsyncClient;
|
|
import org.apache.http.impl.nio.client.HttpAsyncClients;
|
|
import org.eclipse.jgit.lib.Config;
|
|
import org.junit.After;
|
|
import org.junit.AfterClass;
|
|
import org.junit.BeforeClass;
|
|
|
|
public class ElasticV6QueryChangesTest extends AbstractQueryChangesTest {
|
|
@ConfigSuite.Default
|
|
public static Config defaultConfig() {
|
|
return IndexConfig.createForElasticsearch();
|
|
}
|
|
|
|
private static ElasticNodeInfo nodeInfo;
|
|
private static ElasticContainer container;
|
|
private static CloseableHttpAsyncClient client;
|
|
|
|
@BeforeClass
|
|
public static void startIndexService() {
|
|
if (nodeInfo != null) {
|
|
// do not start Elasticsearch twice
|
|
return;
|
|
}
|
|
|
|
container = ElasticContainer.createAndStart(ElasticVersion.V6_8);
|
|
nodeInfo =
|
|
new ElasticNodeInfo(
|
|
container.getHttpHost().getHostName(), container.getHttpHost().getPort());
|
|
client = HttpAsyncClients.createDefault();
|
|
client.start();
|
|
}
|
|
|
|
@AfterClass
|
|
public static void stopElasticsearchServer() {
|
|
if (container != null) {
|
|
container.stop();
|
|
}
|
|
}
|
|
|
|
@After
|
|
public void closeIndex() {
|
|
client.execute(
|
|
new HttpPost(
|
|
String.format(
|
|
"http://localhost:%d/%s*/_close", nodeInfo.port, getSanitizedMethodName())),
|
|
HttpClientContext.create(),
|
|
null);
|
|
}
|
|
|
|
@Override
|
|
protected void initAfterLifecycleStart() throws Exception {
|
|
super.initAfterLifecycleStart();
|
|
ElasticTestUtils.createAllIndexes(injector);
|
|
}
|
|
|
|
@Override
|
|
protected Injector createInjector() {
|
|
Config elasticsearchConfig = new Config(config);
|
|
InMemoryModule.setDefaults(elasticsearchConfig);
|
|
String indicesPrefix = getSanitizedMethodName();
|
|
ElasticTestUtils.configure(
|
|
elasticsearchConfig, nodeInfo.hostname, nodeInfo.port, indicesPrefix);
|
|
return Guice.createInjector(new InMemoryModule(elasticsearchConfig));
|
|
}
|
|
}
|