Better date range handling for range option - #149

Andrew Kane
1 parent 9df6106d
Showing 3 changed files with 35 additions and 1 deletions Show diff stats
CHANGELOG.md
lib/groupdate/magic.rb
test/test_helper.rb
+## 3.1.0 [unreleased]
+
+- Better date range handling for `range` option
+
 ## 3.0.2
 - Fixed `group_by_period` with associations
@@ -174,7 +174,12 @@ module Groupdate
     def time_range
       @time_range ||= begin
         time_range = options[:range]
-        if !time_range && options[:last]
+        if time_range.is_a?(Range) && time_range.first.is_a?(Date)
+          # convert range of dates to range of times
+          last = time_range.last.in_time_zone(time_zone)
+          last += 1.day unless time_range.exclude_end?
+          time_range = Range.new(time_range.first.in_time_zone(time_zone), last, true)
+        elsif !time_range && options[:last]
           step = 1.send(field) if 1.respond_to?(field)
           if step
             now = Time.now
@@ -1059,6 +1059,31 @@ module TestGroupdate
     assert_equal expected, result(:day, "2013-05-03", true, dates: true)
   end
+  # date range
+
+  def test_date_range
+    ENV["TZ"] = "Europe/Oslo"
+    expected = {
+      Date.parse("2013-05-01") => 0,
+      Date.parse("2013-05-02") => 0,
+      Date.parse("2013-05-03") => 0
+    }
+    assert_equal expected, User.group_by_day(:created_at, range: Date.parse("2013-05-01")..Date.parse("2013-05-03")).count
+  ensure
+    ENV["TZ"] = "UTC"
+  end
+
+  def test_date_range_exclude_end
+    ENV["TZ"] = "Europe/Oslo"
+    expected = {
+      Date.parse("2013-05-01") => 0,
+      Date.parse("2013-05-02") => 0
+    }
+    assert_equal expected, User.group_by_day(:created_at, range: Date.parse("2013-05-01")...Date.parse("2013-05-03")).count
+  ensure
+    ENV["TZ"] = "UTC"
+  end
+
   # day start
   def test_day_start_decimal_end_of_day
		1	+## 3.1.0 [unreleased]
		2	+
		3	+- Better date range handling for `range` option
		4	+
1	## 3.0.2	5	## 3.0.2
2		6
3	- Fixed `group_by_period` with associations	7	- Fixed `group_by_period` with associations
	@@ -1059,6 +1059,31 @@ module TestGroupdate		@@ -1059,6 +1059,31 @@ module TestGroupdate
1059	assert_equal expected, result(:day, "2013-05-03", true, dates: true)	1059	assert_equal expected, result(:day, "2013-05-03", true, dates: true)
1060	end	1060	end
1061		1061
		1062	+ # date range
		1063	+
		1064	+ def test_date_range
		1065	+ ENV["TZ"] = "Europe/Oslo"
		1066	+ expected = {
		1067	+ Date.parse("2013-05-01") => 0,
		1068	+ Date.parse("2013-05-02") => 0,
		1069	+ Date.parse("2013-05-03") => 0
		1070	+ }
		1071	+ assert_equal expected, User.group_by_day(:created_at, range: Date.parse("2013-05-01")..Date.parse("2013-05-03")).count
		1072	+ ensure
		1073	+ ENV["TZ"] = "UTC"
		1074	+ end
		1075	+
		1076	+ def test_date_range_exclude_end
		1077	+ ENV["TZ"] = "Europe/Oslo"
		1078	+ expected = {
		1079	+ Date.parse("2013-05-01") => 0,
		1080	+ Date.parse("2013-05-02") => 0
		1081	+ }
		1082	+ assert_equal expected, User.group_by_day(:created_at, range: Date.parse("2013-05-01")...Date.parse("2013-05-03")).count
		1083	+ ensure
		1084	+ ENV["TZ"] = "UTC"
		1085	+ end
		1086	+
1062	# day start	1087	# day start
1063		1088
1064	def test_day_start_decimal_end_of_day	1089	def test_day_start_decimal_end_of_day