File: worker_generator.rb

package info (click to toggle)
ruby-sidekiq 5.2.3%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 828 kB
  • sloc: ruby: 4,065; makefile: 24; sh: 6
file content (49 lines) | stat: -rw-r--r-- 1,156 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
require 'rails/generators/named_base'

module Sidekiq
  module Generators # :nodoc:
    class WorkerGenerator < ::Rails::Generators::NamedBase # :nodoc:
      desc 'This generator creates a Sidekiq Worker in app/workers and a corresponding test'

      check_class_collision suffix: 'Worker'

      def self.default_generator_root
        File.dirname(__FILE__)
      end

      def create_worker_file
        template 'worker.rb.erb', File.join('app/workers', class_path, "#{file_name}_worker.rb")
      end

      def create_test_file
        if defined?(RSpec)
          create_worker_spec
        else
          create_worker_test
        end
      end

      private

      def create_worker_spec
        template_file = File.join(
            'spec/workers',
            class_path,
            "#{file_name}_worker_spec.rb"
        )
        template 'worker_spec.rb.erb', template_file
      end

      def create_worker_test
        template_file = File.join(
            'test/workers',
            class_path,
            "#{file_name}_worker_test.rb"
        )
        template 'worker_test.rb.erb', template_file
      end


    end
  end
end