From c0397aa67c40dc8a7facca3f60ed2de769354a26 Mon Sep 17 00:00:00 2001
From: xvrl <xavier@metamarkets.com>
Date: Thu, 21 Mar 2013 10:36:14 -0700
Subject: [PATCH] add test case for granularities not aligned with segment
 boundaries

---
 .../timeseries/TimeseriesQueryRunnerTest.java | 47 +++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerTest.java b/server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerTest.java
index 3be91c3686f..a28a93c62cd 100644
--- a/server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerTest.java
+++ b/server/src/test/java/com/metamx/druid/query/timeseries/TimeseriesQueryRunnerTest.java
@@ -375,6 +375,53 @@ public class TimeseriesQueryRunnerTest
     TestHelper.assertExpectedResults(expectedResults2, results2);
   }
 
+  @Test
+  public void testTimeseriesGranularityNotAlignedOnSegmentBoundariesWithFilter()
+    {
+      TimeseriesQuery query1 = Druids.newTimeseriesQueryBuilder()
+                                     .dataSource(QueryRunnerTestHelper.dataSource)
+                                     .filters(QueryRunnerTestHelper.providerDimension, "spot", "upfront", "total_market")
+                                     .granularity(new PeriodGranularity(new Period("P7D"), null, DateTimeZone.forID("America/Los_Angeles")))
+                                     .intervals(
+                                         Arrays.asList(
+                                             new Interval(
+                                                 "2011-01-12T00:00:00.000-08:00/2011-01-20T00:00:00.000-08:00"
+                                             )
+                                         )
+                                     )
+                                     .aggregators(
+                                         Arrays.<AggregatorFactory>asList(
+                                             QueryRunnerTestHelper.rowsCount,
+                                             new LongSumAggregatorFactory(
+                                                 "idx",
+                                                 "index"
+                                             )
+                                         )
+                                     )
+                                     .build();
+
+      List<Result<TimeseriesResultValue>> expectedResults1 = Arrays.asList(
+          new Result<TimeseriesResultValue>(
+              new DateTime("2011-01-06T00:00:00.000-08:00", DateTimeZone.forID("America/Los_Angeles")),
+              new TimeseriesResultValue(
+                  ImmutableMap.<String, Object>of("rows", 13L, "idx", 6071L)
+              )
+          ),
+          new Result<TimeseriesResultValue>(
+              new DateTime("2011-01-13T00:00:00.000-08:00", DateTimeZone.forID("America/Los_Angeles")),
+              new TimeseriesResultValue(
+                  ImmutableMap.<String, Object>of("rows", 91L, "idx", 33382L)
+              )
+          )
+      );
+
+      Iterable<Result<TimeseriesResultValue>> results1 = Sequences.toList(
+          runner.run(query1),
+          Lists.<Result<TimeseriesResultValue>>newArrayList()
+      );
+      TestHelper.assertExpectedResults(expectedResults1, results1);
+    }
+
   @Test
   public void testTimeseriesWithVaryingGranWithFilter()
   {