FSDirectoryWrapperTest.java

/*
 * Carrot2 project.
 *
 * Copyright (C) 2002-2016, Dawid Weiss, Stanisław Osiński.
 * All rights reserved.
 *
 * Refer to the full license file "carrot2.LICENSE"
 * in the root folder of the repository checkout or at:
 * http://www.carrot2.org/carrot2.LICENSE
 */

package org.carrot2.source.lucene;

import java.io.StringReader;
import java.io.StringWriter;
import java.nio.file.Files;
import java.nio.file.Path;

import org.apache.lucene.analysis.core.SimpleAnalyzer;
import org.apache.lucene.store.FSDirectory;
import org.carrot2.util.ReflectionUtils;
import org.carrot2.util.simplexml.SimpleXmlWrapperValue;
import org.carrot2.util.simplexml.SimpleXmlWrappers;
import org.carrot2.util.tests.CarrotTestCase;
import org.junit.BeforeClass;
import org.junit.Test;
import org.simpleframework.xml.core.Persister;

import com.carrotsearch.randomizedtesting.LifecycleScope;
import com.carrotsearch.randomizedtesting.RandomizedTest;

/**
 * Test cases for {@link FSDirectoryWrapper}.
 */
public class FSDirectoryWrapperTest extends CarrotTestCase
{
    private static Path indexDir;
    private static FSDirectory directory;

    @BeforeClass
    public static void installFSDirectoryWrapper() throws ClassNotFoundException
    {
        // Make sure LuceneDocumentSource is loaded (this registers the wrapper).
        ReflectionUtils.classForName(LuceneDocumentSource.class.getName());
    }

    @BeforeClass
    public static void prepareIndex() throws Exception
    {
        indexDir = RandomizedTest.newTempDir(LifecycleScope.SUITE);
        directory = FSDirectory.open(indexDir);
        closeAfterSuite(directory);
        LuceneIndexUtils.createAndPopulateIndex(directory, new SimpleAnalyzer());
    }

    @Test
    public void testFSDirectorySerialization() throws Exception
    {
        FSDirectory unserializedDir = null;
        try
        {
            final Persister persister = new Persister();

            final StringWriter writer = new StringWriter();
            persister.write(SimpleXmlWrappers.wrap(directory), writer);

            final SimpleXmlWrapperValue wrapper = persister.read(
                SimpleXmlWrapperValue.class, new StringReader(writer.toString()));
            assertThat(wrapper).describedAs("Wrapper for: " + writer.toString()).isNotNull();
            unserializedDir = SimpleXmlWrappers.unwrap(wrapper);

            assertThat(unserializedDir).isNotNull();
            assertThat(Files.isSameFile(unserializedDir.getDirectory(), indexDir)).isTrue();
        }
        finally
        {
            if (unserializedDir != null)
            {
                unserializedDir.close();
            }
        }
    }
}