Question

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

rake file

#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任务中测试另一个私有方法......但我可以稍后再担心。

Answer 1

执行此操作的常用方法是将所有实际代码移动到模块中，并将任务实现保留为：

require 'that_new_module'

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

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

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

通过这种方式，您可以在不触及rake任务的情况下测试和重构实现。

Answer 2

你可以这样做：

require 'rake'
load 'simple_task.rake'
task_needs_to_run?
=> true

我自己尝试过这个...在Rake命名空间中定义一个方法与在顶层定义一个方法相同。

load Rakefile不会运行任何任务......它只是定义它们。因此，在测试脚本中load使用Rakefile没有任何害处，因此您可以测试相关的方法。

Answer 3

在已定义了rake上下文(something like this)的项目中工作时：

punc='!"#$%&\'()*+,-./:;<=>?@[\\]^_`{|}~'

def lower_clean_str(x):
    lowercased_str = x.lower()
    clean_str = lowercased_str.translate(punc) 
    return clean_str

one_RDD = l.flatMap(lambda x: lower_clean_str(x).split())
one_RDD.collect()

测试rake任务中定义的方法

3 个答案: