package org.infinispan.lucene.impl;
import org.apache.lucene.store.Directory;
import org.apache.lucene.store.IndexOutput;
import org.apache.lucene.store.Lock;
import org.apache.lucene.store.LockFactory;
import org.infinispan.Cache;
import org.infinispan.configuration.cache.ConfigurationBuilder;
import org.infinispan.lucene.CacheTestSupport;
import org.infinispan.lucene.directory.DirectoryBuilder;
import org.infinispan.manager.EmbeddedCacheManager;
import org.infinispan.test.SingleCacheManagerTest;
import org.infinispan.test.fwk.TestCacheManagerFactory;
import org.testng.AssertJUnit;
import org.testng.annotations.Test;
import java.io.IOException;
/**
* Tests covering DirecotoryImplementor class.
*
* @author Anna Manukyan
*/
@Test(groups = "functional", testName = "lucene.V3DirectoryImplementerTests")
public class V3DirectoryImplementerTests extends SingleCacheManagerTest {
private static final String INDEX_NAME = "index-A";
private static final int BUFFER_SIZE = 1024;
@Override
protected EmbeddedCacheManager createCacheManager() throws Exception {
ConfigurationBuilder configuration = CacheTestSupport.createLocalCacheConfiguration();
return TestCacheManagerFactory.createCacheManager(configuration);
}
@Test(expectedExceptions = IllegalArgumentException.class, expectedExceptionsMessageRegExp = "chunkSize must be a positive integer")
public void testInitWithInvalidChunkSize() throws IOException {
Directory dir = null;
try {
dir = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, INDEX_NAME).chunkSize(0).create();
} finally {
if (dir != null) dir.close();
}
}
@Test(expectedExceptions = IllegalArgumentException.class)
public void testFailureOfOverrideWriteLocker() throws IOException {
Directory dir = null;
try {
dir = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, INDEX_NAME).chunkSize(BUFFER_SIZE)
.overrideWriteLocker(null)
.create();
} finally {
if (dir != null) dir.close();
}
}
public void testOverrideWriteLocker() throws IOException {
Directory dir = null;
try {
dir = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, INDEX_NAME).chunkSize(BUFFER_SIZE)
.overrideWriteLocker(new LockFactory() {
@Override
public Lock makeLock(String lockName) {
return null;
}
@Override
public void clearLock(String lockName) throws IOException {
}
})
.create();
AssertJUnit.assertEquals(0, dir.listAll().length);
} finally {
if (dir != null) dir.close();
}
}
@Test
public void testFileModified() throws Exception {
Cache cache = cacheManager.getCache();
String fileName = "dummyFileName";
Directory dir = null;
try {
dir = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, INDEX_NAME).chunkSize(BUFFER_SIZE).create();
createFile(fileName, dir);
assert dir.fileExists(fileName);
assert dir.fileModified(fileName) != 0;
AssertJUnit.assertEquals(0, dir.fileModified("nonExistentFileName.txt"));
} finally {
if (dir != null) dir.close();
}
}
@Test
public void testTouchFile() throws Exception {
Cache cache = cacheManager.getCache();
String fileName = "testfile.txt";
Directory dir = null;
try {
dir = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, INDEX_NAME).chunkSize(BUFFER_SIZE).create();
createFile(fileName, dir);
long lastModifiedDate = dir.fileModified(fileName);
Thread.sleep(1);
dir.touchFile(fileName);
assert lastModifiedDate != dir.fileModified(fileName);
} finally {
if (dir != null) dir.close();
}
}
@Test
public void testTouchNonExistentFile() throws Exception {
Cache cache = cacheManager.getCache();
String fileName = "nonExistent.txt";
Directory dir = null;
try {
dir = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, INDEX_NAME).chunkSize(BUFFER_SIZE).create();
long lastModifiedDate = dir.fileModified(fileName);
AssertJUnit.assertEquals(0L, lastModifiedDate);
Thread.sleep(1);
dir.touchFile(fileName);
AssertJUnit.assertEquals(lastModifiedDate, dir.fileModified(fileName));
} finally {
if (dir != null) dir.close();
}
}
@Test
public void testGetIndexNameAndToString() throws IOException {
Cache cache = cacheManager.getCache();
Directory dir = null;
try {
dir = DirectoryBuilder.newDirectoryInstance(cache, cache, cache, INDEX_NAME).chunkSize(BUFFER_SIZE).create();
AssertJUnit.assertEquals(INDEX_NAME, ((DirectoryLuceneV3) dir).getIndexName());
AssertJUnit.assertEquals("InfinispanDirectory{indexName=\'" + INDEX_NAME + "\'}", dir.toString());
} finally {
if (dir != null) dir.close();
}
}
private void createFile(final String fileName, final Directory dir) throws IOException {
IndexOutput io = dir.createOutput(fileName);
io.writeByte((byte) 66);
io.writeByte((byte) 69);
io.flush();
io.close();
}
}