time_freeze 0.1.0

data/lib/time_freeze/time_extensions.rb

@@ -0,0 +1,43 @@
+require 'date'
+
+class Time
+  
+  # stolen from ActiveSupport...
+  # but I don't want to require ActiveSupport
+  
+  # Converts a Time object to a Date, dropping hour, minute, and second precision.
+  #
+  #   my_time = Time.now  # => Mon Nov 12 22:59:51 -0500 2007
+  #   my_time.to_date     # => Mon, 12 Nov 2007
+  #
+  #   your_time = Time.parse("1/13/2009 1:13:03 P.M.")  # => Tue Jan 13 13:13:03 -0500 2009
+  #   your_time.to_date                                 # => Tue, 13 Jan 2009
+  def to_date
+    ::Date.new(year, month, day)
+  end unless method_defined?(:to_date)
+
+  # A method to keep Time, Date and DateTime instances interchangeable on conversions.
+  # In this case, it simply returns +self+.
+  def to_time
+    self
+  end unless method_defined?(:to_time)
+
+  # Converts a Time instance to a Ruby DateTime instance, preserving UTC offset.
+  #
+  #   my_time = Time.now    # => Mon Nov 12 23:04:21 -0500 2007
+  #   my_time.to_datetime   # => Mon, 12 Nov 2007 23:04:21 -0500
+  #
+  #   your_time = Time.parse("1/13/2009 1:13:03 P.M.")  # => Tue Jan 13 13:13:03 -0500 2009
+  #   your_time.to_datetime                             # => Tue, 13 Jan 2009 13:13:03 -0500
+  def to_datetime
+    ::DateTime.civil(year, month, day, hour, min, sec, Rational(utc_offset, 86400))
+  end unless method_defined?(:to_datetime)
+  
+  # Ruby 1.8-cvs and early 1.9 series define private Time#to_date
+  %w(to_time to_date to_datetime).each do |method|
+    if private_instance_methods.include?(method) || private_instance_methods.include?(method.to_sym)
+      public method
+    end
+  end
+  
+end

data/lib/time_freeze.rb

@@ -0,0 +1,62 @@
+require 'time_freeze/time_extensions'
+
+module TimeFreeze
+  
+  @@frozen_time = nil
+  
+  def self.frozen_time
+    @@frozen_time
+  end
+  
+  def self.frozen_time=(value)
+    @@frozen_time = value
+  end
+  
+  def self.freeze!(frozen_time=Time.now)
+    raise ArgumentError, "we can only freeze to a Time" unless frozen_time.is_a?(Time)
+    TimeFreeze.frozen_time, time_before = frozen_time, TimeFreeze.frozen_time
+    yield
+  ensure
+    # no matter what happens
+    # we need to reset this
+    TimeFreeze.frozen_time = time_before
+  end
+  
+  def self.unfreeze!
+    self.frozen_time=nil
+  end
+
+end
+
+class << Time
+  alias :now_unfrozen :now
+  def now
+    if TimeFreeze.frozen_time
+      TimeFreeze.frozen_time.to_time
+    else
+      self.now_unfrozen
+    end
+  end
+end
+
+class << Date
+  alias :today_unfrozen :today
+  def today
+    if TimeFreeze.frozen_time
+      TimeFreeze.frozen_time.to_date
+    else
+      self.today_unfrozen
+    end
+  end
+end
+
+class << DateTime
+  alias :now_unfrozen :now
+  def now
+    if TimeFreeze.frozen_time
+      TimeFreeze.frozen_time.to_datetime
+    else
+      self.now_unfrozen
+    end
+  end
+end

data/test/test_helper.rb

@@ -0,0 +1,5 @@
+require "rubygems"
+require 'test/unit'
+$:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
+
+require 'time_freeze'

data/test/time_freeze_test.rb

@@ -0,0 +1,95 @@
+require 'test_helper'
+require 'time'
+require 'date'
+
+class TimeFreezeTest < Test::Unit::TestCase
+  
+  def setup
+    @time_now     = Time.now
+    @datetime_now = DateTime.now
+    @date_today    = Date.today
+    
+    TimeFreeze.unfreeze!
+  end
+  
+  def test_normal_function
+     assert_unfrozen_time
+     
+     sleep 10
+     
+     assert_unfrozen_time(10)
+   end
+  
+  def test_freeze!
+    time1 = Time.mktime(2001,12,30,13,45,59,999999)
+    
+    assert_unfrozen_time
+    
+    TimeFreeze.freeze!(time1) do
+      assert_frozen_time(time1)
+    end
+    
+    assert_unfrozen_time
+  end
+  
+  class ExampleRaisedException < Exception; end
+  
+  def test_freeze_unfreezes_after_exception
+   time1 = Time.mktime(2001,12,30,13,45,59,999999)
+    
+    assert_unfrozen_time
+    
+    begin
+      TimeFreeze.freeze!(time1) do
+        
+        assert_frozen_time(time1)
+
+        raise ExampleRaisedException
+
+        flunk "we seem to have resumed after the raise"
+      end
+
+    rescue ExampleRaisedException
+
+      assert_unfrozen_time
+      
+    end
+
+    assert_unfrozen_time
+  end
+  
+  def test_nested_freeze
+    time1 = Time.mktime(2001,12,30,13,45,59,999999)
+    time2 = Time.mktime(2015,03,16,23,22,13,764282)
+    
+    assert_unfrozen_time
+    
+    TimeFreeze.freeze!(time1) do
+      assert_frozen_time(time1)
+      
+      TimeFreeze.freeze!(time2) do
+        assert_frozen_time(time2)
+      end
+      
+      assert_frozen_time(time1)
+    end
+    
+    assert_unfrozen_time
+  end
+  
+  def assert_frozen_time(expected_time)
+    assert_equal TimeFreeze.frozen_time,    expected_time
+    assert_equal expected_time,             Time.now
+    assert_equal expected_time.to_datetime, DateTime.now
+    assert_equal expected_time.to_date,     Date.today
+  end
+  
+  def assert_unfrozen_time(offset=0)
+    assert_in_delta offset, Time.now - @time_now, 0.5
+    assert_in_delta offset*DateTime::SECONDS_IN_DAY, DateTime.now - @datetime_now, 0.5*DateTime::SECONDS_IN_DAY
+    
+    assert_equal @date_today, Date.today
+  end
+  
+end
+      

metadata

@@ -0,0 +1,70 @@
+--- !ruby/object:Gem::Specification 
+name: time_freeze
+version: !ruby/object:Gem::Version 
+  hash: 27
+  prerelease: false
+  segments: 
+  - 0
+  - 1
+  - 0
+  version: 0.1.0
+platform: ruby
+authors: 
+- Matthew Rudy Jacobs
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2010-12-02 00:00:00 +08:00
+default_executable: 
+dependencies: []
+
+description: 
+email: MatthewRudyJacobs@gmail.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: []
+
+files: 
+- test/test_helper.rb
+- test/time_freeze_test.rb
+- lib/time_freeze/time_extensions.rb
+- lib/time_freeze.rb
+has_rdoc: true
+homepage: http://yoursite.example.com
+licenses: []
+
+post_install_message: 
+rdoc_options: []
+
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  none: false
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      hash: 3
+      segments: 
+      - 0
+      version: "0"
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.7
+signing_key: 
+specification_version: 3
+summary: The simplest possible way to freeze time
+test_files: []
+