class Sidekiq::Generators::JobGenerator

def self.default_generator_root
  File.dirname(__FILE__)
end

def create_job_file
  template "job.rb.erb", File.join("app/sidekiq", class_path, "#{file_name}_job.rb")
end

def create_job_spec
  template_file = File.join(
    "spec/sidekiq",
    class_path,
    "#{file_name}_job_spec.rb"
  )
  template "job_spec.rb.erb", template_file
end

def create_job_test
  template_file = File.join(
    "test/sidekiq",
    class_path,
    "#{file_name}_job_test.rb"
  )
  template "job_test.rb.erb", template_file
end

def create_test_file
  return unless test_framework
  if test_framework == :rspec
    create_job_spec
  else
    create_job_test
  end
end

def file_name
  @_file_name ||= super.sub(/_?job\z/i, "")
end

def test_framework
  ::Rails.application.config.generators.options[:rails][:test_framework]
end