Added RakeTasks bootstrap stage.

lib/bootstrap/rake.rb

@@ -3,6 +3,7 @@
 require_relative 'stage/database'
 require_relative 'stage/jobs'
 require_relative 'stage/logger'
+require_relative 'stage/rake_tasks'
 
 module DocumentTransfer
   module Bootstrap
@@ -25,6 +26,8 @@ def bootstrap
         rescue Sequel::DatabaseConnectionError => e
           warn("Database connection error: #{e.message}")
         end
+
+        Stage::RakeTasks.new(@config).bootstrap
       end
     end
   end

lib/bootstrap/stage/rake_tasks.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+
+require 'delayed_job'
+
+require_relative 'base'
+require_relative '../../document_transfer'
+
+module DocumentTransfer
+  module Bootstrap
+    module Stage
+      # Bootstrap rake tasks.
+      class RakeTasks < Base
+        # Load all custom rake tasks.
+        def bootstrap
+          DocumentTransfer.load_rake_tasks
+        end
+      end
+    end
+  end
+end

spec/support/examples/bootstrap_stages.rb

@@ -11,7 +11,9 @@
     DocumentTransfer::Bootstrap.load_stages
 
     stages.to_h do |stage|
-      klass = DocumentTransfer::Bootstrap::Stage.const_get(stage.capitalize)
+      klass = DocumentTransfer::Bootstrap::Stage.const_get(
+        stage.to_s.split('_').map(&:capitalize).join
+      )
       double = instance_double(klass, bootstrap: nil)
       allow(klass).to receive(:new).and_return(double)
       [stage, double]

spec/unit/document_transfer/bootstrap/rake_spec.rb

@@ -3,5 +3,5 @@
 require_relative '../../../../lib/bootstrap/rake'
 
 RSpec.describe DocumentTransfer::Bootstrap::Rake do
-  include_examples 'bootstrap_stages', %i[logger database jobs]
+  include_examples 'bootstrap_stages', %i[logger database jobs rake_tasks]
 end