class Mocha::ClassMethod

# File lib/mocha/class_method.rb, line 9
def initialize(stubbee, method)
  @stubbee, @original_method = stubbee, nil
  @method = RUBY_VERSION < '1.9' ? method.to_s : method.to_sym
end

# File lib/mocha/class_method.rb, line 55
def define_new_method
  stubbee.__metaclass__.class_eval(%{
    def #{method}(*args, &block)
      mocha.method_missing(:#{method}, *args, &block)
    end
  }, __FILE__, __LINE__)
end

# File lib/mocha/class_method.rb, line 36
def hide_original_method
  if method_exists?(method)
    begin
      @original_method = stubbee._method(method)
      if @original_method && @original_method.owner == stubbee.__metaclass__
        @original_visibility = :public
        if stubbee.__metaclass__.protected_instance_methods.include?(method)
          @original_visibility = :protected
        elsif stubbee.__metaclass__.private_instance_methods.include?(method)
          @original_visibility = :private
        end
        stubbee.__metaclass__.send(:remove_method, method)
      end
    rescue NameError
      # deal with nasties like ActiveRecord::Associations::AssociationProxy
    end
  end
end

# File lib/mocha/class_method.rb, line 81
def matches?(other)
  return false unless (other.class == self.class)
  (stubbee.object_id == other.stubbee.object_id) and (method == other.method)
end

# File lib/mocha/class_method.rb, line 92
def method_exists?(method)
  symbol = method.to_sym
  __metaclass__ = stubbee.__metaclass__
  __metaclass__.public_method_defined?(symbol) || __metaclass__.protected_method_defined?(symbol) || __metaclass__.private_method_defined?(symbol)
end

# File lib/mocha/class_method.rb, line 28
def mock
  stubbee.mocha
end

# File lib/mocha/class_method.rb, line 63
def remove_new_method
  stubbee.__metaclass__.send(:remove_method, method)
end

# File lib/mocha/class_method.rb, line 32
def reset_mocha
  stubbee.reset_mocha
end

# File lib/mocha/class_method.rb, line 67
def restore_original_method
  if @original_method && @original_method.owner == stubbee.__metaclass__
    if RUBY_VERSION < '1.9'
      original_method = @original_method
      stubbee.__metaclass__.send(:define_method, method) do |*args, &block|
        original_method.call(*args, &block)
      end
    else
      stubbee.__metaclass__.send(:define_method, method, @original_method)
    end
    stubbee.__metaclass__.send(@original_visibility, method)
  end
end

# File lib/mocha/class_method.rb, line 14
def stub
  hide_original_method
  define_new_method
end

# File lib/mocha/class_method.rb, line 88
def to_s
  "#{stubbee}.#{method}"
end

# File lib/mocha/class_method.rb, line 19
def unstub
  remove_new_method
  restore_original_method
  mock.unstub(method.to_sym)
  unless mock.any_expectations?
    reset_mocha
  end
end