Class: MiniTest::Mock

Inherits:
Object show all
Defined in:
lib/minitest/mock.rb

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Mock

Returns a new instance of Mock.



5
6
7
8
 
# File 'lib/minitest/mock.rb', line 5

def initialize
  @expected_calls = {}
  @actual_calls = Hash.new {|h,k| h[k] = [] }
end

Instance Method Details

#expect(name, retval, args = []) ⇒ Object 



10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/minitest/mock.rb', line 10

def expect(name, retval, args=[])
  n, r, a = name, retval, args # for the closure below
  @expected_calls[name] = { :retval => retval, :args => args }
  self.class.__send__ :remove_method, name if respond_to? name
  self.class.__send__(:define_method, name) { |*x|
    raise ArgumentError unless @expected_calls[n][:args].size == x.size
    @actual_calls[n] << { :retval => r, :args => x }
    retval
  }
  self
end

#verify ⇒ Object 



22
23
24
25
26
27
28
29
30
 
# File 'lib/minitest/mock.rb', line 22

def verify
  @expected_calls.each_key do |name|
    expected = @expected_calls[name]
    msg = "expected #{name}, #{expected.inspect}"
    raise MockExpectationError, msg unless
      @actual_calls.has_key? name and @actual_calls[name].include?(expected)
  end
  true
end