lib/generators/bullet/install_generator.rb in bullet-6.1.0 vs lib/generators/bullet/install_generator.rb in bullet-6.1.1

@@ -8,43 +8,41 @@
             Enable bullet in development/test for your application.
       DESC
 
       def enable_in_development
         environment(nil, env: 'development') do
-          <<-"FILE"
+          <<~FILE
+            config.after_initialize do
+              Bullet.enable        = true
+              Bullet.alert         = true
+              Bullet.bullet_logger = true
+              Bullet.console       = true
+            # Bullet.growl         = true
+              Bullet.rails_logger  = true
+              Bullet.add_footer    = true
+            end
 
-  config.after_initialize do
-    Bullet.enable        = true
-    Bullet.alert         = true
-    Bullet.bullet_logger = true
-    Bullet.console       = true
-  # Bullet.growl         = true
-    Bullet.rails_logger  = true
-    Bullet.add_footer    = true
-  end
           FILE
-            .strip
         end
 
         say 'Enabled bullet in config/environments/development.rb'
       end
 
       def enable_in_test
-        if yes?('Would you like to enable bullet in test environment? (y/n)')
-          environment(nil, env: 'test') do
-            <<-"FILE"
+        return unless yes?('Would you like to enable bullet in test environment? (y/n)')
 
-  config.after_initialize do
-    Bullet.enable        = true
-    Bullet.bullet_logger = true
-    Bullet.raise         = true # raise an error if n+1 query occurs
-  end
-            FILE
-              .strip
-          end
+        environment(nil, env: 'test') do
+          <<~FILE
+            config.after_initialize do
+              Bullet.enable        = true
+              Bullet.bullet_logger = true
+              Bullet.raise         = true # raise an error if n+1 query occurs
+            end
 
-          say 'Enabled bullet in config/environments/test.rb'
+          FILE
         end
+
+        say 'Enabled bullet in config/environments/test.rb'
       end
     end
   end
 end