package org.infinispan.lucene.cachestore;
import java.io.File;
import java.io.IOException;
import org.apache.lucene.store.Directory;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.lucene.cachestore.configuration.LuceneStoreConfigurationBuilder;
import org.infinispan.lucene.directory.DirectoryBuilder;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.test.CacheManagerCallable;
import org.infinispan.test.TestingUtil;
import org.infinispan.test.fwk.TestCacheManagerFactory;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;
/**
* Verify we can write to a FSDirectory, and when using it via the {@link LuceneCacheLoader}
* we can find the same contents as by searching it directly.
* This implicitly verifies configuration settings passed it, as it won't work if the CacheLoader
* is unable to find the specific index path.
*
* @author Sanne Grinovero
* @since 5.2
*/
@Test(groups = "functional", testName = "lucene.cachestore.IndexCacheLoaderTest")
public class IndexCacheLoaderTest {
private static final int SCALE = 600;
protected final String parentDir = ".";
protected File rootDir = null;
@BeforeMethod
public void setUp() {
rootDir = TestHelper.createRootDir(parentDir, getIndexPathName());
}
/**
* @return a unique name for this test, where we store indexes
*/
protected String getIndexPathName() {
return this.getClass().getSimpleName();
}
@AfterMethod
public void tearDown() {
if(rootDir != null) {
TestingUtil.recursiveFileRemove(rootDir);
}
}
@Test
public void testReadExistingIndex() throws IOException {
TestHelper.createIndex(rootDir, "index-A", 10 * SCALE, true);
TestHelper.createIndex(rootDir, "index-B", 20 * SCALE, false);
TestHelper.verifyIndex(rootDir, "index-A", 10 * SCALE, true);
verifyDirectory(rootDir, "index-A", 10 * SCALE, true);
TestHelper.verifyIndex(rootDir, "index-B", 20 * SCALE, false);
verifyDirectory(rootDir, "index-B", 20 * SCALE, false);
}
private void verifyDirectory (final File rootDir, final String indexName, final int termsAdded, final boolean inverted) {
final EmbeddedCacheManager cacheManager = initializeInfinispan(rootDir);
TestingUtil.withCacheManager(new CacheManagerCallable(cacheManager) {
public void call() {
Cache<Object, Object> cache = cacheManager.getCache();
Directory directory = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, indexName).create();
try {
TestHelper.verifyOnDirectory(directory, termsAdded, inverted);
} catch(Exception ex) {
throw new RuntimeException(ex);
}
}
});
}
protected EmbeddedCacheManager initializeInfinispan(File rootDir) {
ConfigurationBuilder builder = new ConfigurationBuilder();
builder
.persistence()
.addStore(LuceneStoreConfigurationBuilder.class)
.autoChunkSize(1024)
.location(rootDir.getAbsolutePath());
return TestCacheManagerFactory.createCacheManager(builder);
}
}