Module: RSpecCandy::Helpers::ShouldReceiveAndExecute

Defined in:
lib/rspec_candy/helpers/should_receive_and_execute.rb

Instance Method Summary collapse

Instance Method Details

#should_receive_and_execute(method) ⇒ 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
 
# File 'lib/rspec_candy/helpers/should_receive_and_execute.rb', line 5

def should_receive_and_execute(method)

  method = method.to_s
  method_base = method.gsub(/([\?\!\=\[\]]+)$/, '')
  method_suffix = $1

  method_called = "_#{method_base}_called#{method_suffix}"
  method_with_spy = "#{method_base}_with_spy#{method_suffix}"
  method_without_spy = "#{method_base}_without_spy#{method_suffix}"

  prototype = respond_to?(:singleton_class) ? singleton_class : metaclass
  prototype.class_eval do

    method_defined_directly = method_defined?(method) || private_method_defined?(method) # check that a method is not "defined" by responding to method_missing

    unless method_defined?(method_called)
      define_method method_called do |*args|
      end
    end

    if method_defined_directly
      unless method_defined?(method_with_spy)
        define_method method_with_spy do |*args, &block|
          send(method_called, *args)
          send(method_without_spy, *args, &block)
        end
        alias_method_chain method, :spy
      end
    else
      define_method method do |*args, &block|
        send(method_called, *args)
        super(*args, &block)
      end
    end

  end

  should_receive(method_called)
end