svn commit: r1854855 - /jackrabbit/oak/trunk/oak-lucene/src/test/java/org/apache/jackrabbit/oak/plugins/index/lucene/directory/CopyOnWriteDirectoryTest.java

[thomasm]

Author: thomasm
Date: Tue Mar  5 13:55:55 2019
New Revision: 1854855

URL: http://svn.apache.org/viewvc?rev=1854855&view=rev
Log:
OAK-8097 Load Lucene index files before writing to the index (test)

Modified:
    
jackrabbit/oak/trunk/oak-lucene/src/test/java/org/apache/jackrabbit/oak/plugins/index/lucene/directory/CopyOnWriteDirectoryTest.java

Modified: 
jackrabbit/oak/trunk/oak-lucene/src/test/java/org/apache/jackrabbit/oak/plugins/index/lucene/directory/CopyOnWriteDirectoryTest.java
URL: 
http://svn.apache.org/viewvc/jackrabbit/oak/trunk/oak-lucene/src/test/java/org/apache/jackrabbit/oak/plugins/index/lucene/directory/CopyOnWriteDirectoryTest.java?rev=1854855&r1=1854854&r2=1854855&view=diff
==============================================================================
--- 
jackrabbit/oak/trunk/oak-lucene/src/test/java/org/apache/jackrabbit/oak/plugins/index/lucene/directory/CopyOnWriteDirectoryTest.java
 (original)
+++ 
jackrabbit/oak/trunk/oak-lucene/src/test/java/org/apache/jackrabbit/oak/plugins/index/lucene/directory/CopyOnWriteDirectoryTest.java
 Tue Mar  5 13:55:55 2019
@@ -28,10 +28,14 @@ import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicInteger;
 
 import com.google.common.io.Closer;
+
+import org.apache.commons.io.FileUtils;
 import org.apache.jackrabbit.oak.plugins.document.DocumentMKBuilderProvider;
 import org.apache.jackrabbit.oak.plugins.document.DocumentNodeStore;
 import org.apache.jackrabbit.oak.plugins.index.lucene.IndexCopier;
 import org.apache.jackrabbit.oak.plugins.index.lucene.LuceneIndexDefinition;
+import org.apache.jackrabbit.oak.spi.blob.GarbageCollectableBlobStore;
+import org.apache.jackrabbit.oak.spi.blob.MemoryBlobStore;
 import org.apache.jackrabbit.oak.spi.commit.CommitInfo;
 import org.apache.jackrabbit.oak.spi.commit.EmptyHook;
 import org.apache.jackrabbit.oak.spi.state.NodeBuilder;
@@ -48,6 +52,7 @@ import static org.apache.jackrabbit.oak.
 import static 
org.apache.jackrabbit.oak.plugins.index.search.FulltextIndexConstants.INDEX_DATA_CHILD_NAME;
 import static 
org.apache.jackrabbit.oak.plugins.index.search.IndexDefinition.PROP_UID;
 import static 
org.apache.jackrabbit.oak.plugins.index.search.IndexDefinition.STATUS_NODE;
+import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
@@ -80,7 +85,8 @@ public class CopyOnWriteDirectoryTest {
                 return t;
             }
         });
-        copier = new IndexCopier(executor, tempFolder.newFolder());
+        // copyToLocalBeforeWrite requires that prefetch is enabled (as normal)
+        copier = new IndexCopier(executor, tempFolder.newFolder(), true);
         ns = builderProvider.newBuilder().getNodeStore();
     }
 
@@ -139,6 +145,41 @@ public class CopyOnWriteDirectoryTest {
         }
     }
 
+    // OAK-8097
+    @Test
+    public void copyToLocalBeforeWrite() throws Exception {
+        // storage backend (in-memory)
+        GarbageCollectableBlobStore blobStore = new MemoryBlobStore();
+        LuceneIndexDefinition def = new LuceneIndexDefinition(ns.getRoot(), 
ns.getRoot(), "/foo");
+        NodeBuilder builder = ns.getRoot().builder();
+        Directory dir = new DefaultDirectoryFactory(copier, 
blobStore).newInstance(def, builder.child("foo"), INDEX_DATA_CHILD_NAME, false);
+        // add some files
+        addFiles(dir);
+        // close
+        dir.close();
+        // get the directory size
+        File dirName = copier.getIndexDir(def, def.getIndexPath(), 
INDEX_DATA_CHILD_NAME);
+        long oldSize = FileUtils.sizeOfDirectory(dirName.getParentFile());
+
+        // delete all files from the local directory
+        FileUtils.deleteQuietly(dirName);
+        // check if empty
+        assertEquals(0, FileUtils.sizeOfDirectory(dirName.getParentFile()));
+
+        // open the directory again -
+        // this is to download the files from the blob store,
+        // and store them back to the local directory
+        dir = new DefaultDirectoryFactory(copier, blobStore).newInstance(def, 
builder.child("foo"), INDEX_DATA_CHILD_NAME, false);
+
+        // check if the directory size matches,
+        // if yes then all files are restored
+        long newSize = FileUtils.sizeOfDirectory(dirName.getParentFile());
+        assertEquals(oldSize, newSize);
+
+        // done
+        dir.close();
+    }
+
     private void writeTree(NodeBuilder builder) {
         NodeBuilder test = builder.child("test");
         for (int i = 0; i < 100; i++) {