module Mocha
  
  class ArgumentIterator
  
    def initialize(argument)
      @argument = argument
    end
  
    def each(&block)
      if @argument.is_a?(Hash) then
        @argument.each do |method_name, return_value|
          block.call(method_name, return_value)
        end
      else
        block.call(@argument)
      end
    end
  
  end
  
end
