Timecop#return will temporarily return to the actual time for the giv…

…en block (fix #45)

Add in_realtime method

Timecop#return will temporarily return to the actual time for the given block

lib/timecop/timecop.rb

@@ -89,10 +89,16 @@ def baseline=(baseline)
       instance.send(:baseline=, baseline)
     end
 
-    # Reverts back to system's Time.now, Date.today and DateTime.now (if it exists).
-    def return
-      instance.send(:unmock!)
-      nil
+    # Reverts back to system's Time.now, Date.today and DateTime.now (if it exists) permamently when
+    # no block argument is given, or temporarily reverts back to the system's time temporarily for
+    # the given block.
+    def return(&block)
+      if block_given?
+        instance.send(:return, &block)
+      else
+        instance.send(:unmock!)
+        nil
+      end
     end
 
     def return_to_baseline
@@ -130,6 +136,15 @@ def travel(mock_type, *args, &block) #:nodoc:
     end
   end
 
+  def return(&block)
+    current_stack = @_stack
+    current_baseline = @baseline
+    unmock!
+    yield
+    @_stack = current_stack
+    @baseline = current_baseline
+  end
+
   def unmock! #:nodoc:
     @baseline = nil
     @_stack = []

test/timecop_test.rb

@@ -337,6 +337,19 @@ def test_travel_time_returns_nil
     assert_nil Timecop.travel(t_future)
   end
 
+  def test_return_temporarily_returns_to_current_time_in_given_block
+    time_after_travel = Time.local(1990, 7, 16)
+    now = Time.now
+
+    Timecop.travel(time_after_travel)
+
+    assert_times_effectively_equal(time_after_travel, Time.now)
+    Timecop.return do
+      assert_times_effectively_equal(now, Time.now)
+    end
+    assert_times_effectively_equal(time_after_travel, Time.now)
+  end
+
   def test_travel_time_with_block_returns_the_value_of_the_block
     t_future = Time.local(2030, 10, 10, 10, 10, 10)
     expected = :foo