测试 rake 任务中定义的方法

我想测试 rake 任务中定义的方法。

耙文件

#lib/tasks/simple_task.rake
namespace :xyz do
    task :simple_task => :environment do
        begin
            if task_needs_to_run?
                puts "Lets run this..."
                #some code which I don't wish to test
                ...
            end
        end
    end
    def task_needs_to_run?
        # code that needs testing
        return 2 > 1
    end

end

现在，我想测试一下这个方法，task_needs_to_run?在测试文件中 我该怎么做呢 ？

附加说明：理想情况下，我也想在 rake 任务中测试另一个私有方法......但我可以稍后再担心。

通常的方法是将所有实际代码移至模块中，并将任务实现保留为：

require 'that_new_module'

namespace :xyz do
  task :simple_task => :environment do
    ThatNewModule.doit!
  end
end

如果您使用环境变量或命令参数，只需将它们传入：

ThatNewModule.doit!(ENV['SOMETHING'], ARGV[1])

通过这种方式，您可以测试和重构实现，而无需触及 rake 任务。