6e8615737f
Remove references to http://localhost when closing the ES indexes during tests. Change-Id: I7c11cb8078bfcf5a8b7a337d764dcb90f7efbf42
86 lines
2.8 KiB
Java
86 lines
2.8 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.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 ElasticV7QueryChangesTest extends AbstractQueryChangesTest {
|
|
@ConfigSuite.Default
|
|
public static Config defaultConfig() {
|
|
return IndexConfig.createForElasticsearch();
|
|
}
|
|
|
|
private static ElasticContainer container;
|
|
private static CloseableHttpAsyncClient client;
|
|
|
|
@BeforeClass
|
|
public static void startIndexService() {
|
|
if (container == null) {
|
|
// Only start Elasticsearch once
|
|
container = ElasticContainer.createAndStart(ElasticVersion.V7_6);
|
|
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://%s:%d/%s*/_close",
|
|
container.getHttpHost().getHostName(),
|
|
container.getHttpHost().getPort(),
|
|
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, container, indicesPrefix);
|
|
return Guice.createInjector(new InMemoryModule(elasticsearchConfig, notesMigration));
|
|
}
|
|
}
|