lib/mini/mock.rb

   1 class MockExpectationError < StandardError; end
   2
   3 require 'mini/test'
   4
   5 class Mini::Mock
   6   def initialize
   7     @expected_calls = {}
   8     @actual_calls = Hash.new {|h,k| h[k] = [] }
   9   end
  10
  11   def expect(name, retval, args=[])
  12     n, r, a = name, retval, args # for the closure below
  13     @expected_calls[name] = { :retval => retval, :args => args }
  14     self.class.__send__(:define_method, name) { |*a|
  15       raise ArgumentError unless @expected_calls[n][:args].size == a.size
  16       @actual_calls[n] << { :retval => r, :args => a }
  17       retval
  18     }
  19     self
  20   end
  21
  22   def verify
  23     @expected_calls.each_key do |name|
  24       expected = @expected_calls[name]
  25       msg = "expected #{name}, #{expected.inspect}"
  26       raise MockExpectationError, msg unless
  27         @actual_calls.has_key? name and @actual_calls[name].include?(expected)
  28     end
  29     true
  30   end
  31 end