Apache
/
druid
mirror of https://github.com/apache/druid.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add unit test to show breaking Dimension Extraction Filter

This commit is contained in:
Charles Allen 2015-05-22 15:02:11 -07:00
parent cd6df1feba
commit e1399b7ce4
1 changed files with 50 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

55
processing/src/test/java/io/druid/segment/filter/ExtractionDimFilterTest.java
View File

@ -26,6 +26,8 @@ import com.metamx.collections.spatial.ImmutableRTree;
import io.druid.query.extraction.DimExtractionFn; import io.druid.query.extraction.DimExtractionFn;
import io.druid.query.extraction.ExtractionFn; import io.druid.query.extraction.ExtractionFn;
import io.druid.query.filter.BitmapIndexSelector; import io.druid.query.filter.BitmapIndexSelector;
import io.druid.query.filter.DimFilters;
import io.druid.query.filter.ExtractionDimFilter;
import io.druid.segment.data.ArrayIndexed; import io.druid.segment.data.ArrayIndexed;
import io.druid.segment.data.Indexed; import io.druid.segment.data.Indexed;
import it.uniroma3.mat.extendedset.intset.ConciseSet; import it.uniroma3.mat.extendedset.intset.ConciseSet;
@ -47,16 +49,19 @@ public class ExtractionDimFilterTest
); );
private static final Map<String, String> EXTRACTION_VALUES = ImmutableMap.of( private static final Map<String, String> EXTRACTION_VALUES = ImmutableMap.of(
"foo1","extractDimVal" "foo1", "extractDimVal"
); );
private static ImmutableBitmap foo1BitMap; private static ImmutableBitmap foo1BitMap;
@BeforeClass @BeforeClass
public static void setupStatic(){ public static void setupStatic()
{
final ConciseSet conciseSet = new ConciseSet(); final ConciseSet conciseSet = new ConciseSet();
conciseSet.add(1); conciseSet.add(1);
foo1BitMap = new WrappedConciseBitmap(conciseSet); foo1BitMap = new WrappedConciseBitmap(conciseSet);
} }
private static final BitmapIndexSelector BITMAP_INDEX_SELECTOR = new BitmapIndexSelector() private static final BitmapIndexSelector BITMAP_INDEX_SELECTOR = new BitmapIndexSelector()
{ {
@Override @Override
@ -113,7 +118,8 @@ public class ExtractionDimFilterTest
}; };
@Test @Test
public void testEmpty(){ public void testEmpty()
{
ExtractionFilter extractionFilter = new ExtractionFilter( ExtractionFilter extractionFilter = new ExtractionFilter(
"foo", "NFDJUKFNDSJFNS", DIM_EXTRACTION_FN "foo", "NFDJUKFNDSJFNS", DIM_EXTRACTION_FN
); );
@ -122,7 +128,8 @@ public class ExtractionDimFilterTest
} }
@Test @Test
public void testNull(){ public void testNull()
{
ExtractionFilter extractionFilter = new ExtractionFilter( ExtractionFilter extractionFilter = new ExtractionFilter(
"FDHJSFFHDS", "extractDimVal", DIM_EXTRACTION_FN "FDHJSFFHDS", "extractDimVal", DIM_EXTRACTION_FN
); );
@ -131,11 +138,49 @@ public class ExtractionDimFilterTest
} }
@Test @Test
public void testNormal(){ public void testNormal()
{
ExtractionFilter extractionFilter = new ExtractionFilter( ExtractionFilter extractionFilter = new ExtractionFilter(
"foo", "extractDimVal", DIM_EXTRACTION_FN "foo", "extractDimVal", DIM_EXTRACTION_FN
); );
ImmutableBitmap immutableBitmap = extractionFilter.getBitmapIndex(BITMAP_INDEX_SELECTOR); ImmutableBitmap immutableBitmap = extractionFilter.getBitmapIndex(BITMAP_INDEX_SELECTOR);
Assert.assertEquals(1, immutableBitmap.size()); Assert.assertEquals(1, immutableBitmap.size());
} }
@Test
public void testOr()
{
Assert.assertEquals(
1, Filters.convertDimensionFilters(
DimFilters.or(
new ExtractionDimFilter(
"foo",
"extractDimVal",
DIM_EXTRACTION_FN,
null
)
)
).getBitmapIndex(BITMAP_INDEX_SELECTOR).size()
);
Assert.assertEquals(
1,
Filters.convertDimensionFilters(
DimFilters.or(
new ExtractionDimFilter(
"foo",
"extractDimVal",
DIM_EXTRACTION_FN,
null
),
new ExtractionDimFilter(
"foo",
"DOES NOT EXIST",
DIM_EXTRACTION_FN,
null
)
)
).getBitmapIndex(BITMAP_INDEX_SELECTOR).size()
);
}
} }