path: root/railties/test/generators/task_generator_test.rb

require 'generators/generators_test_helper'
require 'rails/generators/rails/task/task_generator'

class TaskGeneratorTest < Rails::Generators::TestCase
  include GeneratorsTestHelper
  arguments %w(feeds foo bar)

  def test_task_is_created
    run_generator
    assert_file "lib/tasks/feeds.rake" do |content|
      assert_match(/namespace :feeds/, content)
      assert_match(/task foo:/, content)
      assert_match(/task bar:/, content)
    end
  end

  def test_task_on_revoke
    task_path = 'lib/tasks/feeds.rake'
    run_generator
    assert_file task_path
    run_generator ['feeds'], behavior: :revoke
    assert_no_file task_path
  end
end