Module: Tins::TimeDummy

Defined in:
lib/tins/time_dummy.rb

Class Method Summary collapse

Class Method Details

.included(modul) ⇒ Object



5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
# File 'lib/tins/time_dummy.rb', line 5

def self.included(modul)
  class << modul
    alias really_new new
    alias really_now now

    remove_method :now rescue nil
    remove_method :new rescue nil

    attr_writer :dummy

    def dummy=(value)
      if value.respond_to?(:to_str)
        value = Time.parse(value.to_str)
      end
      @dummy = value
    end

    def dummy(value = nil)
      if value.nil?
        @dummy
      else
        begin
          old_dummy = @dummy
          self.dummy = value
          yield
        ensure
          self.dummy = old_dummy
        end
      end
    end

    def new(*a)
      if dummy
        dummy.dup
      elsif caller.first =~ /now/
        really_now
      else
        really_new(*a)
      end
    end

    def now
      new
    end
  end
  super
end