我见过的一些项目将他们的命令行工具实现为 Command 对象(例如:Rubygems和我的 linebreak gem)。这些对象使用 ARGV 进行初始化,只需调用或执行方法即可启动整个过程。这使这些项目能够将其命令行应用程序放入虚拟环境中。例如,它们可以将输入和输出流对象保存在命令对象的实例变量中,以使应用程序独立于使用 STDOUT/STDIN。因此,可以测试命令行应用程序的输入/输出。就像我想象的那样,您可以将当前工作目录保存在一个实例变量中,以使您的命令行应用程序独立于您的实际工作目录。然后,您可以为每个测试创建一个临时目录,并将其设置为您的 Command 对象的工作目录。
现在一些代码:
require 'pathname'
class MyCommand
attr_accessor :input, :output, :error, :working_dir
def initialize(options = {})
@input = options[:input] ? options[:input] : STDIN
@output = options[:output] ? options[:output] : STDOUT
@error = options[:error] ? options[:error] : STDERR
@working_dir = options[:working_dir] ? Pathname.new(options[:working_dir]) : Pathname.pwd
end
# Override the puts method to use the specified output stream
def puts(output = nil)
@output.puts(output)
end
def execute(arguments = ARGV)
# Change to the given working directory
Dir.chdir(working_dir) do
# Analyze the arguments
if arguments[0] == '--readfile'
posts_dir = Pathname.new('posts')
my_file = posts_dir + 'myfile'
puts my_file.read
end
end
end
end
# Start the command without mockups if the ruby script is called directly
if __FILE__ == $PROGRAM_NAME
MyCommand.new.execute
end
现在在您的测试设置和拆卸方法中,您可以执行以下操作:
require 'pathname'
require 'tmpdir'
require 'stringio'
def setup
@working_dir = Pathname.new(Dir.mktmpdir('mycommand'))
@output = StringIO.new
@error = StringIO.new
@command = MyCommand.new(:working_dir => @working_dir, :output => @output, :error => @error)
end
def test_some_stuff
@command.execute(['--readfile'])
# ...
end
def teardown
@working_dir.rmtree
end
(在示例中,我使用了 Pathname,这是一个非常好的来自 Ruby 标准库和 StringIO 的面向对象的文件系统 API,它对于模拟 STDOUT 很有用,因为它是一个流入简单字符串的 IO 对象)
在实际测试中,您现在可以使用 @working_dir 变量来测试文件的存在或内容:
path = @working_dir + 'posts' + 'myfile'
path.exist?
path.file?
path.directory?
path.read == "abc\n"