time-warp 1.0.7 → 1.0.8

data/lib/time_warp.rb

@@ -7,18 +7,27 @@ module Test # :nodoc:
       ##
       # Time warp to the specified time for the duration of the passed block.
       def pretend_now_is(*args)
-        begin
-          Time.testing_offset = Time.now - time_from(*args)
-          yield
-        ensure
-          Time.testing_offset = 0
+        Time.testing_offset = Time.now - time_from(*args)
+        if block_given?
+          begin
+            yield
+          ensure
+            reset_to_real_time
+          end
         end
       end
+      
+      ##
+      # Reset to real time.
+      def reset_to_real_time
+        Time.testing_offset = 0
+      end
     
     private
     
       def time_from(*args)
         return args[0] if 1 == args.size && args[0].is_a?(Time)
+        return args[0].to_time if 1 == args.size && args[0].respond_to?(:to_time)  # For example, if it's a Date.
         Time.utc(*args)
       end
 

data/test/time_warp_test.rb

@@ -55,4 +55,42 @@ class TimeWarpTest < Test::Unit::TestCase
     end
     assert_equal now_with_time_argument.to_s, now_with_time_utc_arguments.to_s
   end
+  
+  def test_pretend_now_without_a_block
+    now        = Time.now
+    now_year   = now.year
+    now_month  = now.month
+    now_day    = now.day
+    now_hour   = now.hour
+    now_minute = now.min
+    
+    pretend_now_is(Time.utc(2008,"jul",25,6,15))
+    assert_equal 2008, Time.now.utc.year
+    assert_equal    7, Time.now.utc.month
+    assert_equal   25, Time.now.utc.day
+    assert_equal    6, Time.now.utc.hour
+    assert_equal   15, Time.now.utc.min
+    reset_to_real_time
+    
+    assert_equal   now_year, Time.now.year
+    assert_equal  now_month, Time.now.month
+    assert_equal    now_day, Time.now.day
+    assert_equal   now_hour, Time.now.hour
+    assert_equal now_minute, Time.now.min
+  end
+  
+  def test_pretend_now_with_an_object_that_responds_to_to_time
+    # Date objects in rails have to_time methods, but without Rails, they don't so we fake it
+    date = Object.new
+    def date.to_time
+      Time.utc(2008,"jul",25,0,0)
+    end
+    pretend_now_is(date) do #=> Fri Jul 25 00:00:00 UTC 2008
+      assert_equal 2008, Time.now.utc.year
+      assert_equal    7, Time.now.utc.month
+      assert_equal   25, Time.now.utc.day
+      assert_equal    0, Time.now.utc.hour
+      assert_equal    0, Time.now.utc.min
+    end
+  end
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: time-warp
 version: !ruby/object:Gem::Version 
-  hash: 25
-  prerelease: false
+  hash: 7
+  prerelease: 
   segments: 
   - 1
   - 0
-  - 7
-  version: 1.0.7
+  - 8
+  version: 1.0.8
 platform: ruby
 authors: 
 - Barry Hess
@@ -15,7 +15,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-08-24 00:00:00 -05:00
+date: 2011-04-15 00:00:00 -05:00
 default_executable: 
 dependencies: []
 
@@ -67,7 +67,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
 requirements: []
 
 rubyforge_project: 
-rubygems_version: 1.3.7
+rubygems_version: 1.5.0
 signing_key: 
 specification_version: 3
 summary: Warp time in your tests