如何为main:Object全局mock input方法？（Inspec代码Ruby运行场景）

我们需要在不依赖InSpec wrapper的情况下，直接用Ruby运行InSpec控制代码，同时支持在不同测试场景中为input方法传入不同值，替代原有不够灵活的猴子补丁方案。

方案一：自定义输入存储类（可控性强） #

通过封装一个输入管理类，实现输入值的动态设置与重置，同时为全局Object添加input方法绑定到该类的逻辑，既保留InSpecinput的默认值特性，又能隔离测试用例的状态。

require 'minitest/autorun'

# 封装输入存储与管理逻辑
class InputStore
  def initialize
    reset!
  end

  # 设置单个输入值
  def set(key, value)
    @inputs[key] = value
  end

  # 获取输入值，支持InSpec的默认值逻辑
  def get(key, **opts)
    @inputs.fetch(key, opts[:value])
  end

  # 重置输入存储，避免用例间状态污染
  def reset!
    @inputs = {}
  end
end

# 初始化全局输入存储实例
$input_store = InputStore.new

# 为全局Object添加input方法，关联到输入存储
Object.define_method(:input) do |key, **opts|
  $input_store.get(key, **opts)
end

# 编写测试用例
describe "Big Rock Show 控制逻辑" do
  before do
    # 每个用例前重置输入
    $input_store.reset!
  end

  it "当音量为11时通过断言" do
    $input_store.set('amplifier_max_volume', 11)
    # 加载并执行控制代码
    require_relative 'rock_critic/controls.rb'
  end

  it "当音量为10时触发断言失败" do
    $input_store.set('amplifier_max_volume', 10)
    # 验证原控制逻辑的断言会失败
    assert_raises(Minitest::Assertion) do
      require_relative 'rock_critic/controls.rb'
    end
  end

  it "未设置输入时使用默认值" do
    # 不设置输入，会使用control里定义的默认值10
    assert_raises(Minitest::Assertion) do
      require_relative 'rock_critic/controls.rb'
    end
  end
end

方案优势 #

• 完整模拟InSpecinput的默认值行为
• 支持测试用例间的状态隔离，通过reset!避免污染
• 输入管理逻辑集中，便于扩展（比如从不同数据源加载输入）

方案二：使用Minitest Stub（轻量灵活） #

利用Minitest内置的stub方法，临时替换Object的input方法实现，作用域仅限定在代码块内，适合快速mock单个用例的输入值。

require 'minitest/autorun'

describe "Big Rock Show 控制逻辑" do
  it "使用自定义输入值11" do
    # 临时替换input方法，返回固定值11
    Object.stub(:input, ->(key, **opts) { 11 }) do
      require_relative 'rock_critic/controls.rb'
    end
  end

  it "使用默认输入值10" do
    # 模拟InSpec的默认值逻辑，返回opts中的value参数
    Object.stub(:input, ->(key, **opts) { opts[:value] }) do
      assert_raises(Minitest::Assertion) do
        require_relative 'rock_critic/controls.rb'
      end
    end
  end
end

方案优势 #

• 无需额外封装类，代码更简洁
• 每个用例的mock逻辑完全独立，无全局状态污染
• 快速适配不同的输入场景

原有猴子补丁方案的问题 #

• 全局固定输入哈希，无法在不同测试用例中动态切换值
• 未实现InSpecinput的默认值逻辑，与原框架行为不一致
• 直接修改Object的方法，容易引发全局命名冲突

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