From 25f8a25c3ea107dcd0688307ac0ce19c4306f6b4 Mon Sep 17 00:00:00 2001 From: David Heinemeier Hansson Date: Mon, 21 Feb 2005 14:46:33 +0000 Subject: Added next_week and made beginning_of_week be a Monday, not a Sunday git-svn-id: http://svn-commit.rubyonrails.org/rails/trunk@733 5ecf4fe2-1ee6-0310-87b1-e25e094e27de --- .../lib/active_support/core_ext/time/calculations.rb | 12 +++++++++--- activesupport/test/core_ext/time_ext_test.rb | 8 +++++++- 2 files changed, 16 insertions(+), 4 deletions(-) diff --git a/activesupport/lib/active_support/core_ext/time/calculations.rb b/activesupport/lib/active_support/core_ext/time/calculations.rb index f31c282434..c62a2447e8 100644 --- a/activesupport/lib/active_support/core_ext/time/calculations.rb +++ b/activesupport/lib/active_support/core_ext/time/calculations.rb @@ -53,13 +53,19 @@ module ActiveSupport #:nodoc: end end - # Returns a new Time representing the "start" of this week (sunday, 0:00) + # Returns a new Time representing the "start" of this week (Monday, 0:00) def beginning_of_week - (self - self.wday.days).midnight + (self - self.wday.days).midnight + 1.day end - alias :sunday :beginning_of_week + alias :monday :beginning_of_week alias :at_beginning_of_week :beginning_of_week + # Returns a new Time representing the start of the given day in next week (default is Monday). + def next_week(day = :monday) + days_into_week = { :monday => 0, :tuesday => 1, :wednesday => 2, :thursday => 3, :friday => 4, :saturday => 5, :sunday => 6} + since(1.week).beginning_of_week.since(days_into_week[day].day).change(:hour => 0) + end + # Returns a new Time representing the start of the day (0:00) def beginning_of_day self - self.seconds_since_midnight diff --git a/activesupport/test/core_ext/time_ext_test.rb b/activesupport/test/core_ext/time_ext_test.rb index 1918b01b23..a948d21e17 100644 --- a/activesupport/test/core_ext/time_ext_test.rb +++ b/activesupport/test/core_ext/time_ext_test.rb @@ -12,7 +12,7 @@ class TimeExtCalculationsTest < Test::Unit::TestCase end def test_begining_of_week - assert_equal Time.local(2005,1,30), Time.local(2005,2,4,10,10,10).beginning_of_week + assert_equal Time.local(2005,1,31), Time.local(2005,2,4,10,10,10).beginning_of_week end def test_beginning_of_day @@ -80,4 +80,10 @@ class TimeExtCalculationsTest < Test::Unit::TestCase assert_equal Time.utc(2005,2,22,16,45), Time.utc(2005,2,22,15,15,10).change(:hour => 16, :min => 45) assert_equal Time.utc(2005,2,22,15,45), Time.utc(2005,2,22,15,15,10).change(:min => 45) end + + def test_next_week + assert_equal Time.local(2005,2,28), Time.local(2005,2,22,15,15,10).next_week + assert_equal Time.local(2005,2,29), Time.local(2005,2,22,15,15,10).next_week(:tuesday) + assert_equal Time.local(2005,3,4), Time.local(2005,2,22,15,15,10).next_week(:friday) + end end \ No newline at end of file -- cgit v1.2.3