Ruby CLI工具Minitest测试：参数传递与输出验证问题

嘿，刚好我之前做过类似的Ruby CLI测试，这两个问题Minitest都有很直接的解决办法，我给你一步步说清楚！

Ruby CLI程序的命令行参数都存在全局数组ARGV里，所以在测试中直接修改这个数组的值，就能模拟用户传入不同参数的场景。不过有个小技巧：别让测试直接触发脚本的入口逻辑，最好把主程序的核心逻辑封装成一个类方法（比如MyCLI.run），这样测试时可以直接调用这个方法，而不会因为脚本执行exit导致测试中断。

另外，为了避免不同测试用例之间的参数互相干扰，记得在测试的setup和teardown方法里重置ARGV：

def setup
  # 保存原始ARGV
  @original_argv = ARGV.dup
end

def teardown
  # 测试结束后恢复ARGV
  ARGV.replace(@original_argv)
end

比如要测试传入两个正确参数的场景，就这么写：

test "valid arguments trigger function execution" do
  ARGV = ["valid_arg1", "valid_arg2"]
  # 这里调用你的主程序方法
  MyCLI.run
end

Minitest自带capture_io方法，它能捕获代码运行时的标准输出（stdout）和标准错误（stderr），返回一个包含这两个内容的数组。你可以用它来断言puts输出的内容是否符合预期。

结合退出状态的测试，完整的示例大概是这样：

test "shows help and exits with status 0 when wrong number of args" do
  ARGV = [] # 模拟不传参的场景
  # 捕获输出同时检查退出状态
  output = capture_io do
    assert_raises(SystemExit) { MyCLI.run } do |exit_error|
      assert_equal 0, exit_error.status
    end
  end
  # 断言输出包含帮助信息
  assert_match /Usage: my_cli.rb <arg1> <arg2>/, output.first
end

给你一个可直接复用的CLI结构和测试文件：

主程序（my_cli.rb） #

class MyCLI
  def self.run
    args = ARGV
    # 处理参数数量不对的情况
    if args.size != 2
      puts "Usage: my_cli.rb <valid_first_arg> <valid_second_arg>"
      exit 0
    end

    arg1, arg2 = args
    valid_arg1s = ["foo", "bar"]
    valid_arg2s = ["baz", "qux"]
    errors = []

    # 验证参数合法性
    errors << "Invalid first argument" unless valid_arg1s.include?(arg1)
    errors << "Invalid second argument" unless valid_arg2s.include?(arg2)

    # 参数错误时退出
    unless errors.empty?
      puts errors.join("\n")
      exit 1
    end

    # 参数正确时执行功能
    puts "Executing core function with #{arg1} and #{arg2}"
    exit 0
  end
end

# 脚本入口，只有直接运行时才执行
MyCLI.run if $0 == __FILE__

测试文件（test_my_cli.rb） #

require "minitest/autorun"
require_relative "my_cli"

class TestMyCLI < Minitest::Test
  def setup
    @original_argv = ARGV.dup
  end

  def teardown
    ARGV.replace(@original_argv)
  end

  test "exits with status 0 and shows help when arg count is not 2" do
    ARGV = ["only_one_arg"]
    output = capture_io do
      assert_raises(SystemExit) { MyCLI.run } do |e|
        assert_equal 0, e.status
      end
    end
    assert_match /Usage:/, output.first
  end

  test "exits with status 1 when first arg is invalid" do
    ARGV = ["invalid_arg", "baz"]
    output = capture_io do
      assert_raises(SystemExit) { MyCLI.run } do |e|
        assert_equal 1, e.status
      end
    end
    assert_includes output.first, "Invalid first argument"
  end

  test "exits with status 1 when second arg is invalid" do
    ARGV = ["foo", "invalid_arg"]
    output = capture_io do
      assert_raises(SystemExit) { MyCLI.run } do |e|
        assert_equal 1, e.status
      end
    end
    assert_includes output.first, "Invalid second argument"
  end

  test "exits with status 1 when both args are invalid" do
    ARGV = ["invalid1", "invalid2"]
    output = capture_io do
      assert_raises(SystemExit) { MyCLI.run } do |e|
        assert_equal 1, e.status
      end
    end
    assert_includes output.first, "Invalid first argument"
    assert_includes output.first, "Invalid second argument"
  end

  test "executes function and exits with status 0 when args are valid" do
    ARGV = ["foo", "baz"]
    output = capture_io do
      assert_raises(SystemExit) { MyCLI.run } do |e|
        assert_equal 0, e.status
      end
    end
    assert_includes output.first, "Executing core function with foo and baz"
  end
end

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