Module: Synthesis::ExpectationRecorder

Defined in:
lib/synthesis/expectation_recorder.rb

Overview

Extend by the mock object framework’s construct for declaring a mock object so that Synthesis can tap into it in order to record the expectation.

Instance Method Summary collapse

Instance Method Details

#record_expectations_on(method_name) ⇒ Object

Intercept the mock object framework’s method for declaring a mock object so that Synthesis can record it.



8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/synthesis/expectation_recorder.rb', line 8

def record_expectations_on(method_name)
  @original_expects = method_name
  
  class_eval do
    alias_method "intercepted_#{method_name}", method_name

    define_method(:get_expectation_method_name) {method_name}

    def temp_expectation_record(meth, *expected_parameters, &matching_block)
      s_expectation = ExpectationRecord.add_expectation(self, meth, caller[0])
      m_expectation = send("intercepted_#{get_expectation_method_name}", meth, *expected_parameters, &matching_block)
      m_expectation.synthesis_expectation = s_expectation
      m_expectation
    end
      
    alias_method method_name, :temp_expectation_record
    undef temp_expectation_record
  end
end

#stop_recording! ⇒ Object

Restore the original methods ExpectationRecorder has rewritten and undefine their intercepted counterparts.



30
31
32
33
34
35
36
37
 
# File 'lib/synthesis/expectation_recorder.rb', line 30

def stop_recording!
  method_name = @original_expects
  class_eval do
    alias_method method_name, "intercepted_#{method_name}"
    remove_method "intercepted_#{method_name}"
    remove_method :get_expectation_method_name
  end
end