Use optimized loading of configuration

Author: chaadow

lib/devise/jwt/railtie.rb

@@ -9,9 +9,7 @@ class Railtie < Rails::Railtie
       initializer 'devise-jwt-middleware' do |app|
         app.middleware.use Warden::JWTAuth::Middleware
 
-        config.after_initialize do
-          Rails.application.reload_routes!
-
+        config.after_routes_loaded do
           Warden::JWTAuth.configure do |config|
             defaults = DefaultsGenerator.call
 
@@ -21,15 +19,6 @@ class Railtie < Rails::Railtie
             config.revocation_strategies = defaults[:revocation_strategies]
           end
         end
-
-        ActiveSupport::Reloader.to_prepare do
-          Warden::JWTAuth.configure do |config|
-            defaults = DefaultsGenerator.call
-
-            config.mappings = defaults[:mappings]
-            config.revocation_strategies = defaults[:revocation_strategies]
-          end
-        end
       end
     end
   end

spec/spec_helper.rb

@@ -30,4 +30,10 @@
 
 RSpec.configure do |config|
   config.use_transactional_fixtures = true
+
+  # TODO: Remove once Devise publishes a new version
+  # https://github.com/heartcombo/devise/issues/5705
+  config.before do
+    Rails.application.try(:reload_routes_unless_loaded)
+  end
 end