Wrap reloadable constant calls in to_prepare

rspec · Oct 15, 2023 · ecfe44d · ecfe44d
1 parent b71e7d0
commit ecfe44d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/lib/rspec-rails.rb b/lib/rspec-rails.rb
@@ -7,14 +7,18 @@ module Rails
     # Railtie to hook into Rails.
     class Railtie < ::Rails::Railtie
       # As of Rails 5.1.0 you can register directories to work with `rake notes`
-      require 'rails/source_annotation_extractor'
-      ::Rails::SourceAnnotationExtractor::Annotation.register_directories("spec")
+      config.to_prepare do
+        require 'rails/source_annotation_extractor'
+        ::Rails::SourceAnnotationExtractor::Annotation.register_directories("spec")
+      end
       generators = config.app_generators
       generators.integration_tool :rspec
       generators.test_framework :rspec
 
       generators do
-        ::Rails::Generators.hidden_namespaces.reject! { |namespace| namespace.to_s.start_with?("rspec") }
+        config.to_prepare do
+          ::Rails::Generators.hidden_namespaces.reject! { |namespace| namespace.to_s.start_with?("rspec") }
+        end
       end
 
       rake_tasks do