use of com.yelp.nrtsearch.server.config.LuceneServerConfiguration in project nrtsearch by Yelp.
the class LocalStateBackendTest method getMockGlobalState.
private GlobalState getMockGlobalState() throws IOException {
GlobalState mockState = mock(GlobalState.class);
LuceneServerConfiguration serverConfiguration = getConfig();
when(mockState.getConfiguration()).thenReturn(serverConfiguration);
when(mockState.getStateDir()).thenReturn(Paths.get(serverConfiguration.getStateDir()));
return mockState;
}
use of com.yelp.nrtsearch.server.config.LuceneServerConfiguration in project nrtsearch by Yelp.
the class RemoteStateBackendTest method testArchiverRequired.
@Test
public void testArchiverRequired() throws IOException {
GlobalState mockState = mock(GlobalState.class);
LuceneServerConfiguration serverConfiguration = getConfig(false);
when(mockState.getConfiguration()).thenReturn(serverConfiguration);
when(mockState.getStateDir()).thenReturn(Paths.get(serverConfiguration.getStateDir()));
try {
new RemoteStateBackend(mockState);
fail();
} catch (IllegalArgumentException e) {
assertEquals("Archiver must be provided for remote state usage", e.getMessage());
}
}
use of com.yelp.nrtsearch.server.config.LuceneServerConfiguration in project nrtsearch by Yelp.
the class SuggestTest method setUp.
@Before
public void setUp() throws IOException {
LuceneServerConfiguration luceneServerConfiguration = LuceneServerTestConfigurationFactory.getConfig(Mode.STANDALONE, folder.getRoot());
GlobalState globalState = GlobalState.createState(luceneServerConfiguration);
grpcServer = new GrpcServer(grpcCleanup, luceneServerConfiguration, folder, false, globalState, luceneServerConfiguration.getIndexDir(), "test_index", globalState.getPort());
Path tempDir = folder.newFolder("TestSuggest").toPath();
tempFile = tempDir.resolve("suggest.in");
}
use of com.yelp.nrtsearch.server.config.LuceneServerConfiguration in project nrtsearch by Yelp.
the class RestoreStateHandlerTest method setup.
@Before
public void setup() throws IOException {
s3Directory = folder.newFolder("s3").toPath();
archiverDirectory = folder.newFolder("archiver").toPath();
api = S3Mock.create(8011, s3Directory.toAbsolutePath().toString());
api.start();
s3 = new AmazonS3Client(new AnonymousAWSCredentials());
s3.setEndpoint("http://127.0.0.1:8011");
s3.createBucket(BUCKET_NAME);
archiver = new ArchiverImpl(s3, BUCKET_NAME, archiverDirectory, new TarImpl(TarImpl.CompressionMode.LZ4));
LuceneServerConfiguration luceneServerConfiguration = LuceneServerTestConfigurationFactory.getConfig(Mode.STANDALONE, folder.getRoot());
globalState = GlobalState.createState(luceneServerConfiguration);
}
use of com.yelp.nrtsearch.server.config.LuceneServerConfiguration in project nrtsearch by Yelp.
the class ServerTestCase method setUpGrpcServer.
private GrpcServer setUpGrpcServer(CollectorRegistry collectorRegistry) throws IOException {
String testIndex = "test_index";
LuceneServerConfiguration luceneServerConfiguration = LuceneServerTestConfigurationFactory.getConfig(Mode.STANDALONE, folder.getRoot(), getExtraConfig());
globalState = GlobalState.createState(luceneServerConfiguration);
return new GrpcServer(collectorRegistry, grpcCleanup, luceneServerConfiguration, folder, false, globalState, luceneServerConfiguration.getIndexDir(), testIndex, globalState.getPort(), null, getPlugins());
}
Aggregations