package it.unimi.dsi.mg4j.search;
import it.unimi.dsi.mg4j.document.StringArrayDocumentCollection;
import it.unimi.dsi.mg4j.index.BitStreamIndex;
import it.unimi.dsi.mg4j.index.DiskBasedIndex;
import it.unimi.dsi.mg4j.tool.IndexBuilder;
import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.net.URISyntaxException;
import junit.framework.TestCase;
import org.apache.commons.configuration.ConfigurationException;
import org.apache.commons.io.filefilter.PrefixFileFilter;
public class TrueFalseDocumentIteratorTest extends TestCase {
private BitStreamIndex index;
private String basename;
@Override
public void setUp() throws ConfigurationException, SecurityException, IOException, URISyntaxException, ClassNotFoundException, InstantiationException, IllegalAccessException, InvocationTargetException, NoSuchMethodException {
basename = File.createTempFile( getClass().getSimpleName(), "test" ).getCanonicalPath();
new IndexBuilder( basename, new StringArrayDocumentCollection( "a", "b", "c" ) ).run();
index = DiskBasedIndex.getInstance( basename + "-text", true, true );
}
@Override
public void tearDown() {
for( File f: new File( basename ).getParentFile().listFiles( (FileFilter)new PrefixFileFilter( new File( basename ).getName() ) ) ) f.delete();
}
@SuppressWarnings("deprecation")
public void testTrue() {
TrueDocumentIterator trueDocumentIterator = TrueDocumentIterator.getInstance( index );
assertEquals( 0, trueDocumentIterator.nextDocument() );
assertEquals( IntervalIterators.TRUE, trueDocumentIterator.intervalIterator() );
assertEquals( 1, trueDocumentIterator.nextDocument() );
assertEquals( IntervalIterators.TRUE, trueDocumentIterator.intervalIterator() );
assertEquals( 2, trueDocumentIterator.nextDocument() );
assertEquals( IntervalIterators.TRUE, trueDocumentIterator.intervalIterator() );
assertEquals( -1, trueDocumentIterator.nextDocument() );
trueDocumentIterator = TrueDocumentIterator.getInstance( index );
assertTrue( trueDocumentIterator.hasNext() );
assertEquals( 0, trueDocumentIterator.nextInt() );
assertEquals( IntervalIterators.TRUE, trueDocumentIterator.intervalIterator() );
assertTrue( trueDocumentIterator.hasNext() );
assertEquals( 1, trueDocumentIterator.nextInt() );
assertEquals( IntervalIterators.TRUE, trueDocumentIterator.intervalIterator() );
assertTrue( trueDocumentIterator.hasNext() );
assertEquals( 2, trueDocumentIterator.nextInt() );
assertEquals( IntervalIterators.TRUE, trueDocumentIterator.intervalIterator() );
assertFalse( trueDocumentIterator.hasNext() );
assertEquals( -1, trueDocumentIterator.nextDocument() );
}
public void testFalse() {
FalseDocumentIterator trueDocumentIterator = FalseDocumentIterator.getInstance( index );
assertEquals( -1, trueDocumentIterator.nextDocument() );
}
}