mirror of
https://github.com/apache/druid.git
synced 2025-03-08 10:30:38 +00:00
fix broken UT
This commit is contained in:
parent
1b99553394
commit
a71f7152a5
@ -19,12 +19,14 @@
|
||||
|
||||
package io.druid.query.timeboundary;
|
||||
|
||||
import com.google.common.collect.ImmutableMap;
|
||||
import com.google.common.collect.Lists;
|
||||
import com.metamx.common.guava.Sequences;
|
||||
import io.druid.query.Druids;
|
||||
import io.druid.query.QueryRunner;
|
||||
import io.druid.query.QueryRunnerTestHelper;
|
||||
import io.druid.query.Result;
|
||||
import io.druid.query.TableDataSource;
|
||||
import org.joda.time.DateTime;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
@ -32,7 +34,9 @@ import org.junit.runner.RunWith;
|
||||
import org.junit.runners.Parameterized;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.util.Arrays;
|
||||
import java.util.Collection;
|
||||
import java.util.List;
|
||||
|
||||
/**
|
||||
*/
|
||||
@ -117,4 +121,45 @@ public class TimeBoundaryQueryRunnerTest
|
||||
Assert.assertEquals(new DateTime("2011-01-12T00:00:00.000Z"), minTime);
|
||||
Assert.assertNull(maxTime);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testMergeResults() throws Exception
|
||||
{
|
||||
List<Result<TimeBoundaryResultValue>> results = Arrays.asList(
|
||||
new Result<>(
|
||||
new DateTime(),
|
||||
new TimeBoundaryResultValue(
|
||||
ImmutableMap.of(
|
||||
"maxTime", "2012-01-01",
|
||||
"minTime", "2011-01-01"
|
||||
)
|
||||
)
|
||||
),
|
||||
new Result<>(
|
||||
new DateTime(),
|
||||
new TimeBoundaryResultValue(
|
||||
ImmutableMap.of(
|
||||
"maxTime", "2012-02-01",
|
||||
"minTime", "2011-01-01"
|
||||
)
|
||||
)
|
||||
)
|
||||
);
|
||||
|
||||
TimeBoundaryQuery query = new TimeBoundaryQuery(new TableDataSource("test"), null, null, null);
|
||||
Iterable<Result<TimeBoundaryResultValue>> actual = query.mergeResults(results);
|
||||
|
||||
Assert.assertTrue(actual.iterator().next().getValue().getMaxTime().equals(new DateTime("2012-02-01")));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testMergeResultsEmptyResults() throws Exception
|
||||
{
|
||||
List<Result<TimeBoundaryResultValue>> results = Lists.newArrayList();
|
||||
|
||||
TimeBoundaryQuery query = new TimeBoundaryQuery(new TableDataSource("test"), null, null, null);
|
||||
Iterable<Result<TimeBoundaryResultValue>> actual = query.mergeResults(results);
|
||||
|
||||
Assert.assertFalse(actual.iterator().hasNext());
|
||||
}
|
||||
}
|
||||
|
@ -126,6 +126,7 @@ public class QueryHostFinderTest
|
||||
new TimeBoundaryQuery(
|
||||
new TableDataSource("test"),
|
||||
new MultipleIntervalSegmentSpec(Arrays.<Interval>asList(new Interval("2011-08-31/2011-09-01"))),
|
||||
null,
|
||||
null
|
||||
)
|
||||
);
|
||||
|
Loading…
x
Reference in New Issue
Block a user