Rakefile: Conditionally require local dependencies

lib/generators/suspenders/advisories_generator.rb

@@ -17,10 +17,15 @@ def add_bundler_audit
       end
 
       def modify_rakefile
-        insert_into_file "Rakefile", "\nrequire \"bundler/audit/task\"",
-          after: 'require_relative "config/application"'
-        insert_into_file "Rakefile", "\nBundler::Audit::Task.new",
-          after: 'require "bundler/audit/task"'
+        content = <<~RUBY
+
+          if Rails.env.local?
+            require "bundler/audit/task"
+            Bundler::Audit::Task.new
+          end
+        RUBY
+
+        insert_into_file "Rakefile", content, after: /require_relative "config\/application"\n/
       end
     end
   end

lib/generators/suspenders/rake_generator.rb

@@ -8,7 +8,12 @@ class RakeGenerator < Rails::Generators::Base
       TEXT
 
       def configure_default_rake_task
-        append_to_file "Rakefile", %(task default: "suspenders:rake")
+        append_to_file "Rakefile", <<~RUBY
+
+          if Rails.env.local?
+            task default: "suspenders:rake"
+          end
+        RUBY
       end
     end
   end

test/fixtures/files/Rakefile

@@ -1,5 +1,10 @@
+# Add your own tasks in files placed in lib/tasks ending in .rake,
+# for example lib/tasks/capistrano.rake, and they will automatically be available to Rake.
+
 require_relative "config/application"
-require "bundler/audit/task"
-Bundler::Audit::Task.new
 
 Rails.application.load_tasks
+
+if Rails.env.local?
+  task default: "suspenders:rake"
+end

test/fixtures/files/Rakefile_advisories

@@ -0,0 +1,8 @@
+require_relative "config/application"
+
+if Rails.env.local?
+  require "bundler/audit/task"
+  Bundler::Audit::Task.new
+end
+
+Rails.application.load_tasks

test/generators/suspenders/advisories_generator_test.rb

@@ -48,7 +48,7 @@ class AdvisoriesGeneratorTest < Rails::Generators::TestCase
 
           Rails.application.load_tasks
         TEXT
-        expected_rakefile = file_fixture("Rakefile").read
+        expected_rakefile = file_fixture("Rakefile_advisories").read
 
         run_generator
 

test/generators/suspenders/rake_generator_test.rb

@@ -12,10 +12,12 @@ class RakeGeneratorTest < Rails::Generators::TestCase
       teardown :restore_destination
 
       test "modifies existing Rakefile" do
+        content = file_fixture("Rakefile").read
+
         run_generator
 
         assert_file app_root("Rakefile") do |file|
-          assert_match(/task default: "suspenders:rake"/, file)
+          assert_equal content, file
         end
       end