如何使用RSpec测试Ruby脚本？带参数校验脚本测试方案

你的测试未生效主要有3个核心原因：

• 脚本内部已经通过rescue Errors => e捕获了Errors::NoArgumentError异常，还调用了abort直接退出进程，异常根本不会向外泄漏到RSpec的断言上下文，自然raise_error匹配不到。
• 直接调用load运行脚本时，当前Ruby进程的ARGV存储的是RSpec运行时的参数（比如测试文件路径、格式化参数等），并非空值，所以ARGV.count.zero?的分支根本没有触发。
• 如果你用system调用脚本，相当于启动独立子进程运行，子进程的异常、退出状态不会传递到父进程的RSpec上下文，更不可能捕获到异常。

把业务逻辑和入口执行逻辑拆分，避免所有逻辑都写在脚本顶层，是最稳定的测试方案：

改造后的脚本代码 #

# frozen_string_literal: true

# 假设你已经预先定义了Errors命名空间下的异常类
module Errors
  class NoArgumentError < StandardError; end
end

class FileProcessor
  def initialize(args)
    @args = args
  end

  def run
    raise Errors::NoArgumentError if @args.count.zero?
    file = File.open(@args[0], 'r')
    # 原有文件处理逻辑
  end
end

# 只有直接执行脚本时才运行入口逻辑，被require/load时不会执行
if __FILE__ == $0
  begin
    FileProcessor.new(ARGV).run
  rescue Errors => e
    puts e.message
    abort
  end
end

对应RSpec测试用例 #

直接针对封装的类做单元测试，不需要触碰全局变量，也不需要加载整个脚本入口：

it '无参数时抛出NoArgumentError' do
  expect { FileProcessor.new([]).run }.to raise_error(Errors::NoArgumentError)
end

如果不方便调整原有脚本代码，可以用以下两种方案测试：

方案1：验证进程输出和退出状态（推荐黑盒测试用） #

直接运行脚本子进程，验证退出状态和输出内容是否符合预期：

it '无参数运行时输出错误并异常退出' do
  # 运行脚本，合并标准错误和标准输出
  output = `ruby bin/script.rb 2>&1`
  # 验证进程异常退出
  expect($?.exitstatus).not_to eq(0)
  # 验证输出内容匹配异常提示，替换为你实际定义的错误信息即可
  expect(output).to include('no argument provided')
end

方案2：临时修改全局状态捕获异常 #

通过临时修改全局ARGV、重写abort方法实现异常捕获，注意测试结束后必须还原环境避免影响其他用例：

it '无参数时抛出NoArgumentError' do
  # 备份原有环境
  original_argv = ARGV.dup
  original_abort = method(:abort)
  # 清空参数、重写abort方法让它抛出异常而不是退出进程
  ARGV.clear
  define_method(:abort) { |_| raise Errors::NoArgumentError }

  expect { load 'bin/script.rb' }.to raise_error(Errors::NoArgumentError)
ensure
  # 还原原有环境
  ARGV.replace(original_argv)
  define_method(:abort, original_abort)
end

内容的提问来源于stack exchange，提问作者Mr.Code