
package org.elasticsearch.index.store;

import com.carrotsearch.randomizedtesting.LifecycleScope;
import com.carrotsearch.randomizedtesting.annotations.Repeat;
import com.carrotsearch.randomizedtesting.annotations.Seed;
import org.apache.lucene.store.*;
import org.elasticsearch.test.ElasticsearchLuceneTestCase;
import org.elasticsearch.test.ElasticsearchTestCase;
import org.junit.Test;

import java.io.File;
import java.io.IOException;
import java.util.Collections;
import java.util.Set;

import static org.hamcrest.CoreMatchers.*;

public class DirectoryUtilsTest extends ElasticsearchLuceneTestCase {

    @Test
    public void testGetLeave() throws IOException {
        File file = ElasticsearchTestCase.newTempDir(LifecycleScope.TEST);
        final int iters = scaledRandomIntBetween(10, 100);
        for (int i = 0; i < iters; i++) {
            {
                BaseDirectoryWrapper dir = newFSDirectory(file);
                FSDirectory directory = DirectoryUtils.getLeaf(new FilterDirectory(dir) {}, FSDirectory.class, null);
                assertThat(directory, notNullValue());
                assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir)));
                dir.close();
            }

            {
                BaseDirectoryWrapper dir = newFSDirectory(file);
                FSDirectory directory = DirectoryUtils.getLeaf(dir, FSDirectory.class, null);
                assertThat(directory, notNullValue());
                assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir)));
                dir.close();
            }

            {
                Set<String> stringSet = Collections.emptySet();
                BaseDirectoryWrapper dir = newFSDirectory(file);
                FSDirectory directory = DirectoryUtils.getLeaf(new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean()), FSDirectory.class, null);
                assertThat(directory, notNullValue());
                assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir)));
                dir.close();
            }

            {
                Set<String> stringSet = Collections.emptySet();
                BaseDirectoryWrapper dir = newFSDirectory(file);
                FSDirectory directory = DirectoryUtils.getLeaf(new FilterDirectory(new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean())) {}, FSDirectory.class, null);
                assertThat(directory, notNullValue());
                assertThat(directory, sameInstance(DirectoryUtils.getLeafDirectory(dir)));
                dir.close();
            }

            {
                Set<String> stringSet = Collections.emptySet();
                BaseDirectoryWrapper dir = newFSDirectory(file);
                RAMDirectory directory = DirectoryUtils.getLeaf(new FilterDirectory(new FileSwitchDirectory(stringSet, dir, dir, random().nextBoolean())) {}, RAMDirectory.class, null);
                assertThat(directory, nullValue());
                dir.close();
            }

        }
    }
}
