Ruby中用undef删除类方法后无法重新定义的原因？

场景背景 #

升级Rails项目时遇到Mocha和新版Rspec的any_instance方法冲突：早期项目用Mocha给所有类添加了any_instance，新版Rspec也会给Class打猴子补丁加同名方法。为了不修改大量测试，想保留Mocha的版本，于是先通过Rspec的disable_monkey_patching!移除它的猴子补丁，再用config.mock_with :mocha让Mocha重新定义any_instance，但遇到了方法无法重新生效的问题，以下是最小复现示例。

复现代码 #

1. 初始定义与include

# 定义测试类
class A; end

# 定义包含类方法的模块
module B
  module ClassMethods
    def a; end
  end
end

# 给所有类添加方法a（通过给Class include模块实现）
Class.send :include, B::ClassMethods

A.a # 正常调用，无报错

2. 使用undef移除方法

# 用undef删除Class的a方法，这也是disable_monkey_patching!的实现方式
Class.class_exec { undef a }

A.a # 抛出NoMethodError，符合预期

3. 重新include新模块尝试恢复方法

module C
  module ClassMethods
    def a; end
  end
end

Class.send :include, C::ClassMethods

# 意外结果：仍然抛出NoMethodError
A.a # <- undefined method `a' for A:Class (NoMethodError)

原因分析 #

这是Ruby方法查找机制和undef关键字的特性导致的：

• undef的作用是在接收者的方法表中标记该方法名被禁用，无论这个方法原本是来自接收者自身还是包含的模块
• 当你给Class执行undef a后，Ruby会记住Class这个类不能响应a方法
• 后续即使给Classinclude包含a方法的模块C，在查找A.a（本质是查找Class的实例方法a）时，Ruby会先检查Class本身的方法表，发现a被标记为禁用，就直接抛出错误，不会继续查找祖先链中的模块C

解决办法 #

用remove_method替代undef来移除方法：

• remove_method只会删除接收者自身定义的方法，如果方法来自包含的模块，它会把该方法从当前类的方法表中移除，但不会禁用方法名，后续查找会回到祖先链
• 而undef是彻底禁用该方法名，阻断后续所有查找

修改后的代码：

# 替换undef为remove_method
Class.class_exec { remove_method :a }

# 重新include模块C
Class.send :include, C::ClassMethods

A.a # 正常调用，无报错

对应到你的Rspec场景，需要调整disable_monkey_patching!的逻辑，或者手动用remove_method替代Rspec使用的undef来移除any_instance方法，之后再让Mocha重新定义即可。

内容的提问来源于stack exchange，提问作者Hut8