Set to eager_loading when rails eager_load! called

danmayer · Apr 27, 2019 · 1181516 · 1181516
1 parent 64e038f
commit 1181516
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/lib/coverband/utils/railtie.rb b/lib/coverband/utils/railtie.rb
@@ -2,6 +2,18 @@
 
 Coverband.eager_loading_coverage!
 module Coverband
+  module RailsEagerLoad
+    def eager_load!
+      Coverband.configuration.logger&.debug('Coverband: set to eager_loading')
+      Coverband.eager_loading_coverage!
+      super
+    ensure
+      Coverband.report_coverage(true)
+      Coverband.runtime_coverage!
+    end
+  end
+  Rails::Engine.prepend(RailsEagerLoad)
+
   class Railtie < Rails::Railtie
     initializer 'coverband.configure' do |app|
       app.middleware.use Coverband::Middleware