test-prof 0.6.0 → 0.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 03ab3753dd924402ef2d09973165d82d17d2e72c26ad317e6513111ff1b93166
-  data.tar.gz: 93cbd1952a5054e0c2f76058e227a6e03c0bb6b050f20a74d54fd57b85401e69
+  metadata.gz: e8baffb743630e819928190645aa63d73436f2dc54fc2c56ab1ad5df8141ed3c
+  data.tar.gz: 7738b154073f150e23df40193f350e93e2fb047758023021acef237e08988da3
 SHA512:
-  metadata.gz: 6f6a2ed0ba54bf9c4e0633b842fa5c961ab832f4a45013e44ae68804c1714b6708be22505c7b6de38a6e90a2661f64c684efefbb07d1569ef22cbdf17d5cef57
-  data.tar.gz: 79fa0b40361b533cde8f26c45ed66add0a70cad73f59c5ddcabe6f2997fb944293b36a33f557e4870667a7f023e84219216ef6d4e1ee9d54b316ac038cb281e3
+  metadata.gz: f2f75e27df1795a31f3e39312bfa6268502881f1fd6cb28376603f0486048376d12d5e0266edb2eb7e76f6a54723ae03c43a6c35167242fe2c361b12535cda93
+  data.tar.gz: 56b472304e1291e9f392a2c62c36bc40fc2ad575c0b976d46a8f2fd881bfbf2db3f0948014dd29b99cf63eccded2a63b0ddd834676221f354e872622a80dc883

data/CHANGELOG.md

@@ -2,6 +2,26 @@
 
 ## master
 
+## 0.7.0 (2018-08-12)
+
+- **Ruby 2.3+ is required**. ([@palkan][])
+
+Ruby 2.2 EOL was on 2018-03-31.
+
+- Upgrade RubyProf integration to `ruby-prof >= 0.17`. ([@palkan][])
+
+Use `exclude_common_methods!` instead of the deprecated `eliminate_methods!`.
+
+Add RSpec specific exclusions.
+
+Add ability to specify custom exclusions through `config.custom_exclusions`, e.g.:
+
+```ruby
+TestProf::RubyProf.configure do |config|
+  config.custom_exclusions = { User => %i[save save!] }
+end
+```
+
 ## 0.6.0 (2018-06-29)
 
 ### Features

data/README.md

@@ -32,7 +32,7 @@ Of course, we have some [solutions](https://test-prof.evilmartians.io/#/?id=reci
 
 Supported Ruby versions:
 
-- Ruby (MRI) >= 2.2.0 (**NOTE:** the next release will require 2.3+)
+- Ruby (MRI) >= 2.3.0 (**NOTE:** for Ruby 2.2 use TestProf < 0.7.0)
 
 - JRuby >= 9.1.0.0
 

data/lib/minitest/event_prof_formatter.rb

@@ -2,14 +2,12 @@
 
 require "test_prof/ext/float_duration"
 require "test_prof/ext/string_truncate"
-require "test_prof/ext/string_strip_heredoc"
 
 module Minitest
   module TestProf
     class EventProfFormatter # :nodoc:
       using ::TestProf::FloatDuration
       using ::TestProf::StringTruncate
-      using ::TestProf::StringStripHeredoc
 
       def initialize(profilers)
         @profilers = profilers
@@ -29,7 +27,7 @@ module Minitest
 
       def total_results(profiler)
         @results <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             EventProf results for #{profiler.event}
 
             Total time: #{profiler.total_time.duration}
@@ -46,7 +44,7 @@ module Minitest
           location = group[:id][:location]
 
           @results <<
-            <<-GROUP.strip_heredoc
+            <<~GROUP
               #{description.truncate} (#{location}) – #{group[:time].duration} (#{group[:count]} / #{group[:examples]})
             GROUP
         end
@@ -61,7 +59,7 @@ module Minitest
           location = example[:id][:location]
 
           @results <<
-            <<-GROUP.strip_heredoc
+            <<~GROUP
               #{description.truncate} (#{location}) – #{example[:time].duration} (#{example[:count]})
             GROUP
         end

data/lib/test_prof/any_fixture.rb

@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 
 require "test_prof/ext/float_duration"
-require "test_prof/ext/string_strip_heredoc"
 
 module TestProf
   # Make DB fixtures from blocks.
@@ -9,7 +8,6 @@ module TestProf
     INSERT_RXP = /^INSERT INTO ([\S]+)/
 
     using FloatDuration
-    using StringStripHeredoc
 
     class Cache # :nodoc:
       attr_reader :store, :stats
@@ -85,7 +83,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             AnyFixture usage stats:
           MSG
 
@@ -119,7 +117,7 @@ module TestProf
         end
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
 
             Total time spent: #{total_spent.duration}
             Total time saved: #{total_saved.duration}

data/lib/test_prof/before_all.rb

@@ -5,7 +5,7 @@ module TestProf
   module BeforeAll
     class AdapterMissing < StandardError # :nodoc:
       MSG = "Please, provide an adapter for `before_all` " \
-            "through `TestProf::BeforeAll.adapter = MyAdapter`".freeze
+            "through `TestProf::BeforeAll.adapter = MyAdapter`"
 
       def initialize
         super(MSG)

data/lib/test_prof/cops/rspec/aggregate_failures.rb

@@ -50,7 +50,7 @@ module RuboCop
           return unless self.class.supported?
 
           method, _args, body = *node
-          return unless body && body.begin_type?
+          return unless body&.begin_type?
 
           _receiver, method_name, _object = *method
           return unless GROUP_BLOCKS.include?(method_name)
@@ -120,7 +120,7 @@ module RuboCop
         end
 
         def oneliner?(node)
-          node && node.block_type? &&
+          node&.block_type? &&
             (node.source.lines.size == 1) &&
             example_node?(node)
         end

data/lib/test_prof/event_prof/custom_events/factory_create.rb

@@ -1,10 +1,7 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
 require "test_prof/factory_bot"
 
-using TestProf::StringStripHeredoc
-
 module TestProf::EventProf::CustomEvents
   module FactoryCreate # :nodoc: all
     module RunnerPatch
@@ -49,7 +46,7 @@ TestProf::EventProf::CustomEvents.register("factory.create") do
     TestProf::EventProf::CustomEvents::FactoryCreate.setup!
   else
     TestProf.log(:error,
-                 <<-MSG.strip_heredoc
+                 <<~MSG
                    Failed to load factory_bot / factory_girl.
 
                    Make sure that any of them is in your Gemfile.

data/lib/test_prof/event_prof/custom_events/sidekiq_inline.rb

@@ -1,9 +1,5 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
-
-using TestProf::StringStripHeredoc
-
 module TestProf::EventProf::CustomEvents
   module SidekiqInline # :nodoc: all
     module ClientPatch
@@ -43,7 +39,7 @@ end
 TestProf::EventProf::CustomEvents.register("sidekiq.inline") do
   if TestProf.require(
     'sidekiq/testing',
-    <<-MSG.strip_heredoc
+    <<~MSG
       Failed to load Sidekiq.
 
       Make sure that "sidekiq" gem is in your Gemfile.

data/lib/test_prof/event_prof/custom_events/sidekiq_jobs.rb

@@ -1,9 +1,5 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
-
-using TestProf::StringStripHeredoc
-
 module TestProf::EventProf::CustomEvents
   module SidekiqJobs # :nodoc: all
     module ClientPatch
@@ -30,7 +26,7 @@ end
 TestProf::EventProf::CustomEvents.register("sidekiq.jobs") do
   if TestProf.require(
     'sidekiq/testing',
-    <<-MSG.strip_heredoc
+    <<~MSG
       Failed to load Sidekiq.
 
       Make sure that "sidekiq" gem is in your Gemfile.

data/lib/test_prof/event_prof/monitor.rb

@@ -1,18 +1,14 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
-
 module TestProf
   module EventProf
     # Wrap methods with instrumentation
     module Monitor
-      using StringStripHeredoc
-
       class << self
         def call(mod, event, *mids)
           patch = Module.new do
             mids.each do |mid|
-              module_eval <<-SRC.strip_heredoc, __FILE__, __LINE__
+              module_eval <<~SRC, __FILE__, __LINE__ + 1
                 def #{mid}(*)
                   TestProf::EventProf.instrumenter.instrument(
                     '#{event}'

data/lib/test_prof/event_prof/rspec.rb

@@ -2,7 +2,6 @@
 
 require "test_prof/ext/float_duration"
 require "test_prof/ext/string_truncate"
-require "test_prof/ext/string_strip_heredoc"
 
 module TestProf
   module EventProf
@@ -10,7 +9,6 @@ module TestProf
       include Logging
       using FloatDuration
       using StringTruncate
-      using StringStripHeredoc
 
       NOTIFICATIONS = %i[
         example_group_started
@@ -60,7 +58,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             EventProf results for #{profiler.event}
 
             Total time: #{profiler.total_time.duration}
@@ -75,7 +73,7 @@ module TestProf
           location = group[:id].metadata[:location]
 
           msgs <<
-            <<-GROUP.strip_heredoc
+            <<~GROUP
               #{description.truncate} (#{location}) – #{group[:time].duration} (#{group[:count]} / #{group[:examples]})
             GROUP
         end
@@ -87,7 +85,7 @@ module TestProf
             description = example[:id].description
             location = example[:id].metadata[:location]
             msgs <<
-              <<-GROUP.strip_heredoc
+              <<~GROUP
                 #{description.truncate} (#{location}) – #{example[:time].duration} (#{example[:count]})
               GROUP
           end
@@ -118,7 +116,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             RSpec Stamp results
 
             Total patches: #{stamper.total}
@@ -148,6 +146,6 @@ TestProf.activate('EVENT_PROF') do
       )
     end
 
-    config.after(:suite) { listener.print unless listener.nil? }
+    config.after(:suite) { listener&.print }
   end
 end

data/lib/test_prof/factory_doctor/minitest.rb

@@ -2,7 +2,6 @@
 
 require 'minitest/base_reporter'
 require 'test_prof/ext/float_duration'
-require 'test_prof/ext/string_strip_heredoc'
 
 module Minitest
   module TestProf # :nodoc:
@@ -17,9 +16,8 @@ module Minitest
 
     class FactoryDoctorReporter < BaseReporter # :nodoc:
       using ::TestProf::FloatDuration
-      using ::TestProf::StringStripHeredoc
 
-      SUCCESS_MESSAGE = 'FactoryDoctor says: "Looks good to me!"'.freeze
+      SUCCESS_MESSAGE = 'FactoryDoctor says: "Looks good to me!"'
 
       def initialize(io = $stdout, options = {})
         super
@@ -65,7 +63,7 @@ module Minitest
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             FactoryDoctor report
 
             Total (potentially) bad examples: #{@count}

data/lib/test_prof/factory_doctor/rspec.rb

@@ -1,16 +1,14 @@
 # frozen_string_literal: true
 
 require "test_prof/ext/float_duration"
-require "test_prof/ext/string_strip_heredoc"
 
 module TestProf
   module FactoryDoctor
     class RSpecListener # :nodoc:
       include Logging
       using FloatDuration
-      using StringStripHeredoc
 
-      SUCCESS_MESSAGE = 'FactoryDoctor says: "Looks good to me!"'.freeze
+      SUCCESS_MESSAGE = 'FactoryDoctor says: "Looks good to me!"'
 
       NOTIFICATIONS = %i[
         example_started
@@ -58,7 +56,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             FactoryDoctor report
 
             Total (potentially) bad examples: #{@count}
@@ -100,7 +98,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             RSpec Stamp results
 
             Total patches: #{stamper.total}
@@ -137,7 +135,7 @@ TestProf.activate('FDOC') do
       )
     end
 
-    config.after(:suite) { listener.print unless listener.nil? }
+    config.after(:suite) { listener&.print }
   end
 
   RSpec.shared_context "factory_doctor:ignore", fd_ignore: true do

data/lib/test_prof/factory_prof/printers/flamegraph.rb

@@ -5,8 +5,8 @@ require "test_prof/utils/html_builder"
 module TestProf::FactoryProf
   module Printers
     module Flamegraph # :nodoc: all
-      TEMPLATE = "flamegraph.template.html".freeze
-      OUTPUT_NAME = "factory-flame.html".freeze
+      TEMPLATE = "flamegraph.template.html"
+      OUTPUT_NAME = "factory-flame.html"
 
       class << self
         include TestProf::Logging

data/lib/test_prof/factory_prof/printers/simple.rb

@@ -1,20 +1,17 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
-
 module TestProf::FactoryProf
   module Printers
     module Simple # :nodoc: all
       class << self
         include TestProf::Logging
-        using TestProf::StringStripHeredoc
 
         def dump(result)
           return log(:info, "No factories detected") if result.raw_stats == {}
           msgs = []
 
           msgs <<
-            <<-MSG.strip_heredoc
+            <<~MSG
               Factories usage
 
                total      top-level                            name

data/lib/test_prof/recipes/active_record_one_love.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require_relative "./active_record_shared_connection"
 
 # One ❤️

data/lib/test_prof/recipes/logging.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 if defined?(RSpec)
   RSpec.shared_context "logging:verbose", log: true do
     around(:each) do |ex|
@@ -16,10 +18,13 @@ if defined?(RSpec)
 
   RSpec.shared_context "logging:active_record", log: :ar do
     around(:each) do |ex|
-      logger = ActiveRecord::Base.logger
-      ActiveRecord::Base.logger = Logger.new(STDOUT)
+      *loggers = ActiveRecord::Base.logger,
+                 ActiveSupport::LogSubscriber.logger
+      ActiveSupport::LogSubscriber.logger =
+        ActiveRecord::Base.logger = Logger.new(STDOUT)
       ex.run
-      ActiveRecord::Base.logger = logger
+      ActiveSupport::LogSubscriber.logger,
+      ActiveRecord::Base.logger = *loggers
     end
   end
 end
@@ -33,5 +38,6 @@ end
 
 TestProf.activate("LOG", "ar") do
   TestProf.log :info, "Active Record verbose logging enabled"
-  ActiveRecord::Base.logger = Logger.new(STDOUT)
+  ActiveSupport::LogSubscriber.logger =
+    ActiveRecord::Base.logger = Logger.new(STDOUT)
 end

data/lib/test_prof/recipes/rspec/let_it_be.rb

@@ -9,7 +9,7 @@ module TestProf
   module LetItBe
     class << self
       def module_for(group)
-        modules.fetch(group) do
+        modules[group] ||= begin
           Module.new.tap { |mod| group.prepend(mod) }
         end
       end
@@ -28,7 +28,7 @@ module TestProf
     # Use uniq prefix for instance variables to avoid collisions
     # We want to use the power of Ruby's unicode support)
     # And we love cats!)
-    PREFIX = RUBY_ENGINE == 'jruby' ? "@__jruby_is_not_cat_friendly__".freeze : "@😸".freeze
+    PREFIX = RUBY_ENGINE == 'jruby' ? "@__jruby_is_not_cat_friendly__" : "@😸"
 
     def let_it_be(identifier, **options, &block)
       unless LetItBe.supported?

data/lib/test_prof/rspec_dissect/collectors/base.rb

@@ -3,12 +3,10 @@
 require "test_prof/utils/sized_ordered_set"
 require "test_prof/ext/float_duration"
 require "test_prof/ext/string_truncate"
-require "test_prof/ext/string_strip_heredoc"
 
 module TestProf # :nodoc: all
   using FloatDuration
   using StringTruncate
-  using StringStripHeredoc
 
   module RSpecDissect
     module Collectors
@@ -44,7 +42,7 @@ module TestProf # :nodoc: all
         end
 
         def print_result_header
-          <<-MSG.strip_heredoc
+          <<~MSG
 
             Top #{top_count} slowest suites (by `#{print_name}` time):
 
@@ -52,7 +50,7 @@ module TestProf # :nodoc: all
         end
 
         def print_group_result(group)
-          <<-GROUP.strip_heredoc
+          <<~GROUP
             #{group[:desc].truncate} (#{group[:loc]}) – #{group[name].duration} of #{group[:total].duration} (#{group[:count]})
           GROUP
         end

data/lib/test_prof/rspec_dissect/rspec.rb

@@ -1,14 +1,12 @@
 # frozen_string_literal: true
 
 require "test_prof/ext/float_duration"
-require "test_prof/ext/string_strip_heredoc"
 
 module TestProf
   module RSpecDissect
     class Listener # :nodoc:
       include Logging
       using FloatDuration
-      using StringStripHeredoc
 
       NOTIFICATIONS = %i[
         example_group_finished
@@ -63,7 +61,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             RSpecDissect report
 
             Total time: #{@total_examples_time.duration}
@@ -104,7 +102,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             RSpec Stamp results
 
             Total patches: #{stamper.total}
@@ -141,6 +139,6 @@ TestProf.activate('RD_PROF') do
       )
     end
 
-    config.after(:suite) { listener.print unless listener.nil? }
+    config.after(:suite) { listener&.print }
   end
 end

data/lib/test_prof/rspec_stamp/parser.rb

@@ -22,8 +22,8 @@ module TestProf
         end
 
         def remove_tag(tag)
-          @tags.delete(tag) if @tags
-          @htags.delete_if { |(k, _v)| k == tag } if @htags
+          @tags&.delete(tag)
+          @htags&.delete_if { |(k, _v)| k == tag }
         end
       end
 

data/lib/test_prof/rspec_stamp/rspec.rb

@@ -1,12 +1,9 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
-
 module TestProf
   module RSpecStamp
     class RSpecListener # :nodoc:
       include Logging
-      using StringStripHeredoc
 
       NOTIFICATIONS = %i[
         example_failed
@@ -39,7 +36,7 @@ module TestProf
         msgs = []
 
         msgs <<
-          <<-MSG.strip_heredoc
+          <<~MSG
             RSpec Stamp results
 
             Total patches: #{stamper.total}
@@ -67,6 +64,6 @@ TestProf.activate('RSTAMP') do
       )
     end
 
-    config.after(:suite) { listener.stamp! unless listener.nil? }
+    config.after(:suite) { listener&.stamp! }
   end
 end

data/lib/test_prof/ruby_prof.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
-
 module TestProf
   # RubyProf wrapper.
   #
@@ -24,17 +22,6 @@ module TestProf
   module RubyProf
     # RubyProf configuration
     class Configuration
-      # Default list of methods to exclude from profile.
-      # Contains a lot of RSpec stuff.
-      ELIMINATE_METHODS = [
-        /instance_exec/,
-        /ExampleGroup>?#run/,
-        /Procsy/,
-        /AroundHook#execute_with/,
-        /HookCollections/,
-        /Array#(map|each)/
-      ].freeze
-
       PRINTERS = {
         'flat' => 'FlatPrinter',
         'flat_wln' => 'FlatPrinterWithLineNumbers',
@@ -56,10 +43,12 @@ module TestProf
         'call_stack' => 'html'
       }.freeze
 
-      LOGFILE_PREFIX = "ruby-prof-report".freeze
+      LOGFILE_PREFIX = "ruby-prof-report"
 
       attr_accessor :printer, :mode, :min_percent,
-                    :include_threads, :eliminate_methods
+                    :include_threads, :exclude_common_methods,
+                    :test_prof_exclusions_enabled,
+                    :custom_exclusions
 
       def initialize
         @printer = ENV['TEST_RUBY_PROF'].to_sym if PRINTERS.key?(ENV['TEST_RUBY_PROF'])
@@ -67,16 +56,21 @@ module TestProf
         @mode = ENV.fetch('TEST_RUBY_PROF_MODE', :wall).to_sym
         @min_percent = 1
         @include_threads = false
-        @eliminate_methods = ELIMINATE_METHODS
+        @exclude_common_methods = true
+        @test_prof_exclusions_enabled = true
+        @custom_exclusions = {}
       end
 
       def include_threads?
         include_threads == true
       end
 
-      def eliminate_methods?
-        !eliminate_methods.nil? &&
-          !eliminate_methods.empty?
+      def exclude_common_methods?
+        exclude_common_methods == true
+      end
+
+      def test_prof_exclusions_enabled?
+        @test_prof_exclusions_enabled == true
       end
 
       # Returns an array of printer type (ID) and class.
@@ -105,9 +99,6 @@ module TestProf
       def dump(name)
         result = @profiler.stop
 
-        result.eliminate_methods!(config.eliminate_methods) if
-          config.eliminate_methods?
-
         printer_type, printer_class = config.resolve_printer
 
         if %w[call_tree multi].include?(printer_type)
@@ -145,7 +136,6 @@ module TestProf
 
     class << self
       include Logging
-      using StringStripHeredoc
 
       def config
         @config ||= Configuration.new
@@ -183,6 +173,20 @@ module TestProf
           config.include_threads?
 
         profiler = ::RubyProf::Profile.new(options)
+        profiler.exclude_common_methods! if config.exclude_common_methods?
+
+        if config.test_prof_exclusions_enabled?
+          # custom test-prof exclusions
+          exclude_rspec_methods(profiler)
+
+          # custom global exclusions
+          exclude_common_methods(profiler)
+        end
+
+        config.custom_exclusions.each do |klass, mids|
+          profiler.exclude_methods! klass, *mids
+        end
+
         profiler.start
 
         Report.new(profiler)
@@ -199,7 +203,7 @@ module TestProf
         ENV["RUBY_PROF_MEASURE_MODE"] = config.mode.to_s
         @initialized = TestProf.require(
           'ruby-prof',
-          <<-MSG.strip_heredoc
+          <<~MSG
             Please, install 'ruby-prof' first:
                # Gemfile
               gem 'ruby-prof', '>= 0.16.0', require: false
@@ -208,20 +212,49 @@ module TestProf
       end
 
       def check_ruby_prof_version
-        if Utils.verify_gem_version('ruby-prof', at_least: '0.16.0')
+        if Utils.verify_gem_version('ruby-prof', at_least: '0.17.0')
           true
         else
-          log :error, <<-MGS.strip_heredoc
-            Please, upgrade 'ruby-prof' to version >= 0.16.0.
+          log :error, <<~MGS
+            Please, upgrade 'ruby-prof' to version >= 0.17.0.
           MGS
           false
         end
       end
+
+      def exclude_rspec_methods(profiler)
+        return unless defined?(RSpec::Core)
+
+        RSpecExclusions.generate.each do |klass, mids|
+          profiler.exclude_methods!(klass, *mids)
+        end
+      end
+
+      def exclude_common_methods(profiler)
+        profiler.exclude_methods!(
+          TSort,
+          :tsort_each
+        )
+
+        profiler.exclude_methods!(
+          TSort.singleton_class,
+          :tsort_each, :each_strongly_connected_component,
+          :each_strongly_connected_component_from
+        )
+
+        profiler.exclude_methods!(
+          BasicObject,
+          :instance_exec
+        )
+      end
     end
   end
 end
 
-require "test_prof/ruby_prof/rspec" if defined?(RSpec::Core)
+if defined?(RSpec::Core)
+  require "test_prof/ruby_prof/rspec"
+  require "test_prof/ruby_prof/rspec_exclusions"
+end
 
 # Hook to run RubyProf globally
 TestProf.activate('TEST_RUBY_PROF') do

data/lib/test_prof/ruby_prof/rspec_exclusions.rb

@@ -0,0 +1,93 @@
+# frozen_string_literal: true
+
+module TestProf
+  module RubyProf
+    # Generates the list of RSpec (framework internal) methods
+    # to exclude from profiling
+    module RSpecExclusions
+      module_function
+
+      def generate
+        {
+          RSpec::Core::Runner => %i[
+            run
+            run_specs
+          ],
+
+          RSpec::Core::ExampleGroup => %i[
+            run
+            run_examples
+          ],
+
+          RSpec::Core::ExampleGroup.singleton_class => %i[
+            run
+            run_examples
+          ],
+
+          RSpec::Core::Example => %i[
+            run
+            with_around_and_singleton_context_hooks
+            with_around_example_hooks
+            instance_exec
+            run_before_example
+          ],
+
+          RSpec::Core::Example.singleton_class => %i[
+            run
+            with_around_and_singleton_context_hooks
+            with_around_example_hooks
+          ],
+
+          RSpec::Core::Example::Procsy => [
+            :call
+          ],
+
+          RSpec::Core::Hooks::HookCollections => %i[
+            run
+            run_around_example_hooks_for
+            run_example_hooks_for
+            run_owned_hooks_for
+          ],
+
+          RSpec::Core::Hooks::BeforeHook => [
+            :run
+          ],
+
+          RSpec::Core::Hooks::AroundHook => [
+            :execute_with
+          ],
+
+          RSpec::Core::Configuration => [
+            :with_suite_hooks
+          ],
+
+          RSpec::Core::Reporter => [
+            :report
+          ]
+        }.tap do |data|
+          if defined?(RSpec::Support::ReentrantMutex)
+            data[RSpec::Support::ReentrantMutex] = [
+              :synchronize
+            ]
+          end
+
+          if defined?(RSpec::Core::MemoizedHelpers::ThreadsafeMemoized)
+            data.merge!(
+              RSpec::Core::MemoizedHelpers::ThreadsafeMemoized => [
+                :fetch_or_store
+              ],
+
+              RSpec::Core::MemoizedHelpers::NonThreadSafeMemoized => [
+                :fetch_or_store
+              ],
+
+              RSpec::Core::MemoizedHelpers::ContextHookMemoized => [
+                :fetch_or_store
+              ]
+            )
+          end
+        end
+      end
+    end
+  end
+end

data/lib/test_prof/stack_prof.rb

@@ -1,7 +1,5 @@
 # frozen_string_literal: true
 
-require "test_prof/ext/string_strip_heredoc"
-
 module TestProf
   # StackProf wrapper.
   #
@@ -47,7 +45,6 @@ module TestProf
 
     class << self
       include Logging
-      using StringStripHeredoc
 
       def config
         @config ||= Configuration.new
@@ -103,7 +100,7 @@ module TestProf
 
         html_path = path.gsub(/\.dump$/, '.html')
 
-        log :info, <<-MSG.strip_heredoc
+        log :info, <<~MSG
           Run the following command to generate a flame graph report:
 
           stackprof --flamegraph #{path} > #{html_path} && stackprof --flamegraph-viewer=#{html_path}
@@ -126,7 +123,7 @@ module TestProf
         return @initialized if instance_variable_defined?(:@initialized)
         @initialized = TestProf.require(
           'stackprof',
-          <<-MSG.strip_heredoc
+          <<~MSG
             Please, install 'stackprof' first:
                # Gemfile
               gem 'stackprof', '>= 0.2.9', require: false
@@ -138,7 +135,7 @@ module TestProf
         if Utils.verify_gem_version('stackprof', at_least: '0.2.9')
           true
         else
-          log :error, <<-MSG.strip_heredoc
+          log :error, <<~MSG
             Please, upgrade 'stackprof' to version >= 0.2.9.
           MSG
           false

data/lib/test_prof/tag_prof/printers/html.rb

@@ -3,8 +3,8 @@
 module TestProf::TagProf
   module Printers
     module HTML # :nodoc: all
-      TEMPLATE = "tagprof.template.html".freeze
-      OUTPUT_NAME = "tag-prof.html".freeze
+      TEMPLATE = "tagprof.template.html"
+      OUTPUT_NAME = "tag-prof.html"
 
       class << self
         include TestProf::Logging

data/lib/test_prof/tag_prof/printers/simple.rb

@@ -1,21 +1,19 @@
 # frozen_string_literal: true
 
 require "test_prof/ext/float_duration"
-require "test_prof/ext/string_strip_heredoc"
 
 module TestProf::TagProf
   module Printers
     module Simple # :nodoc: all
       class << self
         include TestProf::Logging
-        using TestProf::StringStripHeredoc
         using TestProf::FloatDuration
 
         def dump(result)
           msgs = []
 
           msgs <<
-            <<-MSG.strip_heredoc
+            <<~MSG
               TagProf report for #{result.tag}
             MSG
 

data/lib/test_prof/tag_prof/rspec.rb

@@ -33,7 +33,7 @@ module TestProf
       def example_started(_notification)
         @ts = TestProf.now
         # enable event profiling
-        @events_profiler.group_started(true) if @events_profiler
+        @events_profiler&.group_started(true)
       end
 
       def example_finished(notification)
@@ -42,7 +42,7 @@ module TestProf
         result.track(tag, time: TestProf.now - @ts, events: fetch_events_data)
 
         # reset and disable event profilers
-        @events_profiler.group_started(nil) if @events_profiler
+        @events_profiler&.group_started(nil)
       end
 
       # NOTE: RSpec < 3.4.0 doesn't have example_finished event
@@ -80,7 +80,7 @@ TestProf.activate('TAG_PROF') do
       )
     end
 
-    config.after(:suite) { listener.report unless listener.nil? }
+    config.after(:suite) { listener&.report }
   end
 end
 

data/lib/test_prof/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module TestProf
-  VERSION = "0.6.0".freeze
+  VERSION = "0.7.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: test-prof
 version: !ruby/object:Gem::Version
-  version: 0.6.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Vladimir Dementyev
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-06-29 00:00:00.000000000 Z
+date: 2018-08-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -72,14 +72,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.57.0
+        version: 0.58.0
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.57.0
+        version: 0.58.0
 - !ruby/object:Gem::Dependency
   name: rubocop-md
   requirement: !ruby/object:Gem::Requirement
@@ -142,7 +142,6 @@ files:
 - lib/test_prof/ext/active_record_refind.rb
 - lib/test_prof/ext/array_bsearch_index.rb
 - lib/test_prof/ext/float_duration.rb
-- lib/test_prof/ext/string_strip_heredoc.rb
 - lib/test_prof/ext/string_truncate.rb
 - lib/test_prof/factory_all_stub.rb
 - lib/test_prof/factory_all_stub/factory_bot_patch.rb
@@ -183,6 +182,7 @@ files:
 - lib/test_prof/rubocop.rb
 - lib/test_prof/ruby_prof.rb
 - lib/test_prof/ruby_prof/rspec.rb
+- lib/test_prof/ruby_prof/rspec_exclusions.rb
 - lib/test_prof/stack_prof.rb
 - lib/test_prof/stack_prof/rspec.rb
 - lib/test_prof/tag_prof.rb
@@ -206,7 +206,7 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.2.0
+      version: 2.3.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="

data/lib/test_prof/ext/string_strip_heredoc.rb

@@ -1,15 +0,0 @@
-# frozen_string_literal: true
-
-module TestProf
-  # Add #strip_heredoc method to use instead of
-  # squiggly docs (to support older Rubies)
-  module StringStripHeredoc
-    refine String do
-      def strip_heredoc
-        min = scan(/^[ \t]*(?=\S)/).min
-        indent = min ? min.size : 0
-        gsub(/^[ \t]{#{indent}}/, '')
-      end
-    end
-  end
-end