Class: Minitest::Mock

Relationships & Source Files
Inherits:	Object
Defined in:	lib/minitest/mock.rb

Overview

A simple and clean mock object framework.

All mock objects are an instance of Mock

Instance Method Summary

#__respond_to?
#expect(name, retval, args = [], &blk)

Expect that method name is called, optionally with args or a blk, and returns retval.
#verify

Verify that all methods were called as expected.

Instance Method Details

#__respond_to?

[ GitHub ]

# File 'lib/minitest/mock.rb', line 11


alias :__respond_to? :respond_to?

#expect(name, retval, args = [], &blk)

Expect that method name is called, optionally with args or a blk, and returns retval.

@mock.expect(:meaning_of_life, 42)
@mock.meaning_of_life # => 42

@mock.expect(:do_something_with, true, [some_obj, true])
@mock.do_something_with(some_obj, true) # => true

@mock.expect(:do_something_else, true) do |a1, a2|
  a1 == "buggs" && a2 == :bunny
end

args is compared to the expected args using case equality (ie, the '===' operator), allowing for less specific expectations.

@mock.expect(:uses_any_string, true, [String])
@mock.uses_any_string("foo") # => true
@mock.verify  # => true

@mock.expect(:uses_one_string, true, ["foo"])
@mock.uses_one_string("bar") # => true
@mock.verify  # => raises MockExpectationError

[ GitHub ]

# File 'lib/minitest/mock.rb', line 67


def expect(name, retval, args=[], &blk)
  name = name.to_sym

  if block_given?
    raise ArgumentError, "args ignored when block given" unless args.empty?
    @expected_calls[name] << { :retval => retval, :block => blk }
  else
    raise ArgumentError, "args must be an array" unless Array === args
    @expected_calls[name] << { :retval => retval, :args => args }
  end
  self
end

#verify

Verify that all methods were called as expected. Raises ::MockExpectationError if the mock object was not called as expected.

[ GitHub ]

# File 'lib/minitest/mock.rb', line 94


def verify
  @expected_calls.each do |name, calls|
    calls.each do |expected|
      msg1 = "expected #{__call name, expected}"
      msg2 = "#{msg1}, got [#{__call name, @actual_calls[name]}]"

      raise MockExpectationError, msg2 if
        @actual_calls.has_key?(name) and
        not @actual_calls[name].include?(expected)

      raise MockExpectationError, msg1 unless
        @actual_calls.has_key?(name) and
        @actual_calls[name].include?(expected)
    end
  end
  true
end