lib/gemaker/commands/customize_engine.rb in gemaker-0.6.0 vs lib/gemaker/commands/customize_engine.rb in gemaker-1.0.0

@@ -1,13 +1,40 @@
 module Gemaker
   module Cmd
     class CustomizeEngine < Gemaker::Cmd::Base
       def in_engine_context
-        copy_template("engine/engine.rb",
-          "lib/#{@config.gem_name}/engine.rb", config: @config)
-        copy_template("engine/example_class.rb",
-          "lib/#{@config.gem_name}/example_class.rb", config: @config)
+        replace_development_file
+        replace_engine_file
+        add_example_class
+
         info "Customize engine.rb"
+      end
+
+      private
+
+      def replace_development_file
+        remove_in_gem("spec/dummy/config/environments/development.rb")
+        copy_template(
+          "engine/development.rb",
+          "spec/dummy/config/environments/development.rb",
+          config: @config
+        )
+      end
+
+      def replace_engine_file
+        copy_template(
+          "engine/engine.rb",
+          "lib/#{@config.gem_name}/engine.rb",
+          config: @config
+        )
+      end
+
+      def add_example_class
+        copy_template(
+          "engine/example_class.rb",
+          "lib/#{@config.gem_name}/example_class.rb",
+          config: @config
+        )
       end
     end
   end
 end