ensure coverage is loaded before extending

danmayer · Jun 24, 2020 · 42f6beb · 42f6beb
1 parent 53e2abd
commit 42f6beb
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/coverband.rb b/lib/coverband.rb
@@ -97,11 +97,6 @@ def self.runtime_coverage!
   end
   unless ENV["COVERBAND_DISABLE_AUTO_START"]
     begin
-      if RUBY_PLATFORM == "java"
-        unless Coverage.respond_to?(:line_stub)
-          require "coverband/utils/jruby_ext"
-        end
-      end
       # Coverband should be setup as early as possible
       # to capture usage of things loaded by initializers or other Rails engines
       configure

diff --git a/lib/coverband/collectors/coverage.rb b/lib/coverband/collectors/coverage.rb
@@ -72,6 +72,11 @@ def initialize
         raise NotImplementedError, "Coverage needs Ruby > 2.3.0" if Gem::Version.new(RUBY_VERSION) < Gem::Version.new("2.3.0")
 
         require "coverage"
+        if RUBY_PLATFORM == "java"
+          unless ::Coverage.respond_to?(:line_stub)
+            require "coverband/utils/jruby_ext"
+          end
+        end
         if defined?(SimpleCov) && defined?(Rails) && defined?(Rails.env) && Rails.env.test?
           puts "Coverband: detected SimpleCov in test Env, allowing it to start Coverage"
           puts "Coverband: to ensure no error logs or missing Coverage call `SimpleCov.start` prior to requiring Coverband"