class Minitest::Mock

def method_missing(sym, *args, &block) # :nodoc:
  unless @expected_calls.key?(sym) then
    raise NoMethodError, "unmocked method %p, expected one of %p" %
      [sym, @expected_calls.keys.sort_by(&:to_s)]
  end
  index = @actual_calls[sym].length
  expected_call = @expected_calls[sym][index]
  unless expected_call then
    raise MockExpectationError, "No more expects available for %p: %p" %
      [sym, args]
  end
  expected_args, retval, val_block =
    expected_call.values_at(:args, :retval, :block)
  if val_block then
    # keep "verify" happy
    @actual_calls[sym] << expected_call
    raise MockExpectationError, "mocked method %p failed block w/ %p" %
      [sym, args] unless val_block.call(*args, &block)
    return retval
  end
  if expected_args.size != args.size then
    raise ArgumentError, "mocked method %p expects %d arguments, got %d" %
      [sym, expected_args.size, args.size]
  end
  zipped_args = expected_args.zip(args)
  fully_matched = zipped_args.all? { |mod, a|
    mod === a or mod == a
  }
  unless fully_matched then
    raise MockExpectationError, "mocked method %p called with unexpected arguments %p" %
      [sym, args]
  end
  @actual_calls[sym] << {
    :retval => retval,
    :args => zipped_args.map! { |mod, a| mod === a ? mod : a },
  }
  retval
end