coverband 4.2.0 → 4.2.1.rc1

data/lib/coverband/adapters/base.rb

@@ -53,11 +53,6 @@ module Coverband
         coverage.keys || []
       end
 
-      # TODO: deprecate / remove?
-      def covered_lines_for_file(file)
-        Array(coverage.dig(file, 'data'))
-      end
-
       protected
 
       def split_coverage(types)
@@ -76,7 +71,7 @@ module Coverband
         raise 'abstract'
       end
 
-      def get_report
+      def get_report(_type = nil)
         raise 'abstract'
       end
 
@@ -87,7 +82,7 @@ module Coverband
       def expand_report(report)
         expanded = {}
         report_time = Time.now.to_i
-        updated_time = self.type == Coverband::EAGER_TYPE ? nil : report_time
+        updated_time = type == Coverband::EAGER_TYPE ? nil : report_time
         report.each_pair do |key, line_data|
           extended_data = {
             'first_updated_at' => report_time,

data/lib/coverband/adapters/file_store.rb

@@ -36,7 +36,7 @@ module Coverband
         File.open(path, 'w') { |f| f.write(report.to_json) }
       end
 
-      def get_report
+      def get_report(_local_type = nil)
         if File.exist?(path)
           JSON.parse(File.read(path))
         else

data/lib/coverband/at_exit.rb

@@ -16,19 +16,10 @@ module Coverband
         at_exit do
           ::Coverband::Background.stop
 
-          #####
-          # TODO: This is is brittle and not a great solution to avoid deploy time
-          # actions polluting the 'runtime' metrics
-          #
-          # * should we skip /bin/rails webpacker:compile ?
-          # * Perhaps detect heroku deployment ENV var opposed to tasks?
-          #####
-          default_heroku_tasks = ['assets:clean', 'assets:precompile']
-          if defined?(Rake) && Rake.respond_to?(:application) && (Rake&.application&.top_level_tasks || []).any? { |task| default_heroku_tasks.include?(task) }
+          if !Coverband.configuration.report_on_exit
             # skip reporting
           else
-            Coverband.report_coverage(true)
-            #Coverband.configuration.logger&.debug('Coverband: Reported coverage before exit')
+            Coverband.report_coverage
           end
         end
       end

data/lib/coverband/collectors/coverage.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+
 require 'singleton'
 require_relative 'delta'
 
@@ -13,10 +14,13 @@ module Coverband
     class Coverage
       include Singleton
 
+      def self.ruby_version_greater_than_or_equal_to?(version)
+        Gem::Version.new(RUBY_VERSION) >= Gem::Version.new(version)
+      end
+
       def reset_instance
         @project_directory = File.expand_path(Coverband.configuration.root)
-        @ignore_patterns = Coverband.configuration.ignore + ['internal:prelude', 'schema.rb']
-        @reporting_frequency = Coverband.configuration.reporting_frequency
+        @ignore_patterns = Coverband.configuration.ignore
         @store = Coverband.configuration.store
         @verbose  = Coverband.configuration.verbose
         @logger   = Coverband.configuration.logger
@@ -34,29 +38,29 @@ module Coverband
         @store.type = Coverband::EAGER_TYPE
       end
 
-      def report_coverage(force_report = false)
-        return if !ready_to_report? && !force_report
-        raise 'no Coverband store set' unless @store
+      def eager_loading
+        old_coverage_type = @store.type
+        eager_loading!
+        yield
+      ensure
+        report_coverage
+        @store.type = old_coverage_type
+      end
 
-        original_previous_set = Delta.previous_results
-        files_with_line_usage = filtered_files(Delta.results)
+      def report_coverage
+        @semaphore.synchronize do
+          raise 'no Coverband store set' unless @store
 
-        ###
-        # Hack to prevent processes and threads from reporting first Coverage hit
-        # when we are in runtime collection mode, which do not have a cache of previous
-        # coverage to remove the initial stdlib Coverage loading data
-        ###
-        if ((original_previous_set.nil? && @store.type == Coverband::EAGER_TYPE) ||
-           (original_previous_set && @store.type != Coverband::EAGER_TYPE))
+          files_with_line_usage = filtered_files(Delta.results)
           @store.save_report(files_with_line_usage)
         end
-      rescue StandardError => err
+      rescue StandardError => e
         if @verbose
-          @logger&.error 'coverage failed to store'
-          @logger&.error "error: #{err.inspect} #{err.message}"
-          @logger&.error err.backtrace
+          @logger.error 'coverage failed to store'
+          @logger.error "error: #{e.inspect} #{e.message}"
+          @logger.error e.backtrace
         end
-        raise err if @test_env
+        raise e if @test_env
       end
 
       protected
@@ -83,25 +87,18 @@ module Coverband
         end.select { |_file_name, coverage| coverage.any? { |value| value&.nonzero? } }
       end
 
-      def ready_to_report?
-        (rand * 100.0) >= (100.0 - @reporting_frequency)
-      end
-
       def initialize
-        if Gem::Version.new(RUBY_VERSION) < Gem::Version.new('2.3.0')
-          raise NotImplementedError, 'not supported until Ruby 2.3.0 and later'
-        end
+        @semaphore = Mutex.new
+        raise NotImplementedError, 'Coverage needs Ruby > 2.3.0' if Gem::Version.new(RUBY_VERSION) < Gem::Version.new('2.3.0')
+
         require 'coverage'
-        if Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('2.5.0')
+        if Coverage.ruby_version_greater_than_or_equal_to?('2.6.0')
+          ::Coverage.start(oneshot_lines: Coverband.configuration.use_oneshot_lines_coverage) unless ::Coverage.running?
+        elsif Coverage.ruby_version_greater_than_or_equal_to?('2.5.0')
           ::Coverage.start unless ::Coverage.running?
         else
           ::Coverage.start
         end
-        if Coverband.configuration.safe_reload_files
-          Coverband.configuration.safe_reload_files.each do |safe_file|
-            load safe_file
-          end
-        end
         reset_instance
       end
     end

data/lib/coverband/collectors/delta.rb

@@ -1,9 +1,9 @@
 # frozen_string_literal: true
+
 module Coverband
   module Collectors
     class Delta
-      @semaphore = Mutex.new
-      @@previous_coverage = nil
+      @@previous_coverage = {}
       attr_reader :current_coverage
 
       def initialize(current_coverage)
@@ -17,18 +17,9 @@ module Coverband
       end
 
       def self.results(process_coverage = RubyCoverage)
-        @semaphore.synchronize do
-          set_default_results
-          new(process_coverage.results).results
-        end
-      end
-
-      def self.previous_results
-        @@previous_coverage
-      end
-
-      def self.set_default_results
-        @@previous_coverage ||= {}
+        coverage_results = process_coverage.results
+        coverage_results = transform_oneshot_lines_results(coverage_results) if Coverband.configuration.use_oneshot_lines_coverage
+        new(coverage_results).results
       end
 
       def results
@@ -38,28 +29,33 @@ module Coverband
       end
 
       def self.reset
-        @@previous_coverage = nil
+        @@previous_coverage = {}
       end
 
       private
 
       def generate
         current_coverage.each_with_object({}) do |(file, line_counts), new_results|
-          if @@previous_coverage[file]
-            new_results[file] = array_diff(line_counts, @@previous_coverage[file])
-          else
-            new_results[file] = line_counts
-          end
+          new_results[file] = if @@previous_coverage[file]
+                                array_diff(line_counts, @@previous_coverage[file])
+                              else
+                                line_counts
+                              end
         end
       end
 
       def array_diff(latest, original)
         latest.map.with_index do |v, i|
-          if (v && original[i])
-            [0, v - original[i]].max
-          else
-            nil
+          [0, v - original[i]].max if v && original[i]
+        end
+      end
+
+      private_class_method def self.transform_oneshot_lines_results(results)
+        results.each_with_object({}) do |(file, coverage), new_results|
+          transformed_line_counts = coverage[:oneshot_lines].each_with_object(::Coverage.line_stub(file)) do |line_number, line_counts|
+            line_counts[line_number - 1] = 1
           end
+          new_results[file] = transformed_line_counts
         end
       end
     end

data/lib/coverband/configuration.rb

@@ -3,15 +3,31 @@
 module Coverband
   class Configuration
     attr_accessor :root_paths, :root,
-                  :ignore, :additional_files, :verbose,
-                  :reporter, :reporting_frequency,
-                  :redis_namespace, :redis_ttl,
-                  :safe_reload_files, :background_reporting_enabled,
+                  :additional_files, :verbose,
+                  :reporter, :redis_namespace, :redis_ttl,
+                  :background_reporting_enabled,
                   :background_reporting_sleep_seconds, :test_env,
-                  :web_enable_clear, :gem_details, :web_debug
+                  :web_enable_clear, :gem_details, :web_debug, :report_on_exit
 
     attr_writer :logger, :s3_region, :s3_bucket, :s3_access_key_id, :s3_secret_access_key
-    attr_reader :track_gems
+    attr_reader :track_gems, :ignore, :use_oneshot_lines_coverage
+
+    #####
+    # TODO: This is is brittle and not a great solution to avoid deploy time
+    # actions polluting the 'runtime' metrics
+    #
+    # * should we skip /bin/rails webpacker:compile ?
+    # * Perhaps detect heroku deployment ENV var opposed to tasks?
+    #####
+    IGNORE_TASKS = ['coverband:clear',
+                    'coverband:coverage',
+                    'coverband:coverage_server',
+                    'coverband:migrate']
+
+    # Heroku when building assets runs code from a dynamic directory
+    # /tmp was added to avoid coverage from /tmp/build directories during
+    # heroku asset compilation
+    IGNORE_DEFAULTS = %w[vendor .erb$ .slim$ /tmp internal:prelude schema.rb]
 
     def initialize
       reset
@@ -20,12 +36,8 @@ module Coverband
     def reset
       @root = Dir.pwd
       @root_paths = []
-      # Heroku when building assets runs code from a dynamic directory
-      # /tmp was added to avoid coverage from /tmp/build directories during
-      # heroku asset compilation
-      @ignore = %w[vendor .erb$ .slim$ /tmp]
+      @ignore = IGNORE_DEFAULTS.dup
       @additional_files = []
-      @reporting_frequency = 0.0
       @verbose = false
       @reporter = 'scov'
       @logger = nil
@@ -38,6 +50,8 @@ module Coverband
       @gem_details = false
       @groups = {}
       @web_debug = false
+      @report_on_exit = true
+      @use_oneshot_lines_coverage = false
 
       # TODO: should we push these to adapter configs
       @s3_region = nil
@@ -49,7 +63,7 @@ module Coverband
     end
 
     def logger
-      @logger ||= if defined?(Rails.logger)
+      @logger ||= if defined?(Rails.logger) && Rails.logger
                     Rails.logger
                   else
                     Logger.new(STDOUT)
@@ -77,21 +91,27 @@ module Coverband
     end
 
     def store=(store)
-      if store.is_a?(Coverband::Adapters::Base)
-        @store = store
-      else
-        raise 'please pass in an subclass of Coverband::Adapters for supported stores'
-      end
+      raise 'Pass in an instance of Coverband::Adapters' unless store.is_a?(Coverband::Adapters::Base)
+
+      @store = store
+    end
+
+    ###
+    # Don't allow the ignore to override things like gem tracking
+    ###
+    def ignore=(ignored_array)
+      @ignore = (@ignore + ignored_array).uniq
     end
 
     def track_gems=(value)
       @track_gems = value
       return unless @track_gems
+
       # by default we ignore vendor where many deployments put gems
       # we will remove this default if track_gems is set
       @ignore.delete('vendor')
       # while we want to allow vendored gems we don't want to track vendored ruby STDLIB
-      @ignore << 'vendor/ruby-*'
+      @ignore << 'vendor/ruby-*' unless @ignore.include?('vendor/ruby-*')
       add_group('App', root)
       # TODO: rework support for multiple gem paths
       # currently this supports GEM_HOME (which should be first path)
@@ -135,7 +155,7 @@ module Coverband
       roots
     end
 
-    SKIPPED_SETTINGS = %w(@s3_secret_access_key @store)
+    SKIPPED_SETTINGS = %w[@s3_secret_access_key @store]
     def to_h
       instance_variables
         .each_with_object('gem_paths': gem_paths) do |var, hash|
@@ -143,6 +163,16 @@ module Coverband
         end
     end
 
+    def use_oneshot_lines_coverage=(value)
+      raise(Exception, 'One shot line coverage is only available in ruby >= 2.6') unless one_shot_coverage_implemented_in_ruby_version? || !value
+
+      @use_oneshot_lines_coverage = value
+    end
+
+    def one_shot_coverage_implemented_in_ruby_version?
+      Gem::Version.new(RUBY_VERSION) >= Gem::Version.new('2.6.0')
+    end
+
     private
 
     def redis_url

data/lib/coverband/integrations/background.rb

@@ -7,6 +7,7 @@ module Coverband
 
     def self.stop
       return unless @thread
+
       @semaphore.synchronize do
         if @thread
           @thread.exit
@@ -16,7 +17,7 @@ module Coverband
     end
 
     def self.running?
-      @thread && @thread.alive?
+      @thread&.alive?
     end
 
     def self.start
@@ -25,12 +26,13 @@ module Coverband
       logger = Coverband.configuration.logger
       @semaphore.synchronize do
         return if running?
-        logger&.debug('Coverband: Starting background reporting')
+
+        logger.debug('Coverband: Starting background reporting')
         sleep_seconds = Coverband.configuration.background_reporting_sleep_seconds
         @thread = Thread.new do
           loop do
-            Coverband.report_coverage(true)
-            logger&.debug("Coverband: Reported coverage via thread. Sleeping #{sleep_seconds}s") if Coverband.configuration.verbose
+            Coverband.report_coverage
+            logger.debug("Coverband: Reported coverage via thread. Sleeping #{sleep_seconds}s") if Coverband.configuration.verbose
             sleep(sleep_seconds)
           end
         end

data/lib/coverband/integrations/{middleware.rb → background_middleware.rb}

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Coverband
-  class Middleware
+  class BackgroundMiddleware
     def initialize(app)
       @app = app
     end
@@ -10,11 +10,7 @@ module Coverband
       @app.call(env)
     ensure
       AtExit.register
-      if Coverband.configuration.background_reporting_enabled
-        Background.start
-      else
-        Collectors::Coverage.instance.report_coverage
-      end
+      Background.start if Coverband.configuration.background_reporting_enabled
     end
   end
 end

data/lib/coverband/integrations/bundler.rb

@@ -0,0 +1,8 @@
+# frozen_string_literal: true
+
+module BundlerEagerLoad
+  def require(*groups)
+    Coverband.eager_loading_coverage { super }
+  end
+end
+Bundler.singleton_class.prepend BundlerEagerLoad

data/lib/coverband/integrations/report_middleware.rb

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+module Coverband
+  class ReportMiddleware
+    def initialize(app)
+      @app = app
+    end
+
+    def call(env)
+      @app.call(env)
+    ensure
+      Collectors::Coverage.instance.report_coverage
+    end
+  end
+end