polyrun 2.2.9 → 3.0.0

data/lib/polyrun/quick/reporter.rb

@@ -1,6 +1,9 @@
 module Polyrun
   module Quick
     class Reporter
+      SLOW_THRESHOLD_SECONDS = 0.05
+      SLOWEST_LIMIT = 5
+
       def initialize(out, err, verbose)
         @out = out
         @err = err
@@ -8,24 +11,28 @@ module Polyrun
         @passed = 0
         @failed = 0
         @errors = 0
+        @timings = []
       end
 
-      def pass(group, description)
+      def pass(group, description, duration_seconds: nil)
         @passed += 1
+        record_timing(group, description, duration_seconds)
         return unless @verbose
 
-        @out.puts "  ok  #{group} #{description}"
+        @out.puts "  ok  #{group} #{description}#{duration_suffix(duration_seconds)}"
       end
 
-      def fail(group, description, exc)
+      def fail(group, description, exc, duration_seconds: nil)
         @failed += 1
-        @err.puts "  FAIL  #{group} #{description}"
+        record_timing(group, description, duration_seconds)
+        @err.puts "  FAIL  #{group} #{description}#{duration_suffix(duration_seconds)}"
         @err.puts "         #{exc.message}"
       end
 
-      def error(group, description, exc)
+      def error(group, description, exc, duration_seconds: nil)
         @errors += 1
-        @err.puts "  ERROR #{group} #{description}"
+        record_timing(group, description, duration_seconds)
+        @err.puts "  ERROR #{group} #{description}#{duration_suffix(duration_seconds)}"
         @err.puts "         #{exc.class}: #{exc.message}"
         loc = exc.backtrace&.first
         @err.puts "         #{loc}" if loc
@@ -35,8 +42,35 @@ module Polyrun
         total = @passed + @failed + @errors
         @out.puts
         @out.puts "Polyrun::Quick: #{@passed} passed, #{@failed} failed, #{@errors} errors (#{total} examples)"
+        print_slowest!
         (@failed + @errors).positive? ? 1 : 0
       end
+
+      private
+
+      def record_timing(group, description, duration_seconds)
+        return if duration_seconds.nil?
+
+        @timings << ["#{group} #{description}".strip, duration_seconds.to_f]
+      end
+
+      def duration_suffix(duration_seconds)
+        return "" if duration_seconds.nil?
+
+        format(" (%.3fs)", duration_seconds)
+      end
+
+      def print_slowest!
+        slow = @timings.select { |(_, sec)| sec >= SLOW_THRESHOLD_SECONDS }
+          .sort_by { |(_, sec)| -sec }
+          .first(SLOWEST_LIMIT)
+        return if slow.empty?
+
+        @out.puts "Slowest:"
+        slow.each do |label, sec|
+          @out.puts format("  %.3fs  %s", sec, label)
+        end
+      end
     end
   end
 end

data/lib/polyrun/quick/runner.rb

@@ -1,20 +1,22 @@
-# rubocop:disable Polyrun/FileLength -- quick runner + spec quality wiring
+# rubocop:disable Polyrun/FileLength -- quick runner + suite config + coverage wiring
 require "pathname"
 
 require_relative "assertions"
+require_relative "config"
 require_relative "errors"
 require_relative "example_group"
 require_relative "example_runner"
 require_relative "reporter"
 
 module Polyrun
-  # Micro test runner: nested +describe+, +it+ / +test+, +before+ / +after+, +let+ / +let!+,
-  # +expect().to+ matchers, optional +Polyrun::Quick.capybara!+ when the +capybara+ gem is loaded.
+  # Micro test runner: nested +describe+ / +with+, +it+ / +test+, +before+ / +after+ / +around+,
+  # +let+ / +let!+ / +subject+, +expect().to+ matchers, optional +Polyrun::Quick.capybara!+.
   #
   # Run: +polyrun quick+ or +polyrun quick spec/foo.rb+
+  # Optional suite file: +config/polyrun_quick.rb+ (+before_suite+, +after_suite+, +add_load_path+).
   #
-  # Coverage: when +POLYRUN_COVERAGE=1+ or (+config/polyrun_coverage.yml+ and +POLYRUN_QUICK_COVERAGE=1+), starts
-  # {Polyrun::Coverage::Rails} before loading quick files so stdlib +Coverage+ records them.
+  # Coverage: when +POLYRUN_COVERAGE=1+, Covered-style +COVERAGE+, or (+config/polyrun_coverage.yml+ and
+  # +POLYRUN_QUICK_COVERAGE=1+), starts {Polyrun::Coverage::Rails} before loading quick files.
   module Quick
     module DSL
       def describe(name, &block)
@@ -103,6 +105,10 @@ module Polyrun
 
       def run(paths)
         Quick.reset_capybara_flag!
+        config = Config.load!(root: Dir.pwd)
+        config.apply_load_paths!(root: Dir.pwd)
+        status = 2
+        suite_hooks = false
 
         files = expand_paths(paths)
         if files.empty?
@@ -110,18 +116,27 @@ module Polyrun
           return 2
         end
 
-        quick_start_coverage_if_configured!
-        quick_start_spec_quality_if_configured!
+        begin
+          config.run_before_suite!
+          suite_hooks = true
+          quick_start_coverage_if_configured!
+          quick_start_spec_quality_if_configured!
 
-        collector = load_quick_files!(files)
-        return 1 unless collector
+          collector = load_quick_files!(files)
+          unless collector
+            status = 1
+            return 1
+          end
 
-        reporter = Reporter.new(@out, @err, @verbose)
-        run_examples!(collector, reporter)
-        reporter.summary
-      ensure
-        Quick.collector = nil
-        Quick.reset_capybara_flag!
+          reporter = Reporter.new(@out, @err, @verbose)
+          run_examples!(collector, reporter)
+          status = reporter.summary
+          status
+        ensure
+          config.run_after_suite!(status) if suite_hooks
+          Quick.collector = nil
+          Quick.reset_capybara_flag!
+        end
       end
 
       def load_quick_files!(files)
@@ -200,7 +215,6 @@ module Polyrun
         Runner.parallel_default_paths(Dir.pwd)
       end
 
-      # Omit RSpec/Minitest files and common helpers so +polyrun quick+ with no args does not load normal suites.
       def default_quick_exclude?(path, base)
         Runner.quick_path_excluded?(path, base)
       end

data/lib/polyrun/reporting/rspec_failure_fragment_formatter.rb

@@ -4,7 +4,7 @@ require "fileutils"
 module Polyrun
   module Reporting
     # RSpec formatter: appends one JSON object per failed example to the shard fragment file.
-    # Enable via +Polyrun::RSpec.install_failure_fragments!+ and +POLYRUN_FAILURE_FRAGMENTS=1+ (set by run-shards --merge-failures).
+    # Enable via +Polyrun::RSpecSupport.install_failure_fragments!+ and +POLYRUN_FAILURE_FRAGMENTS=1+ (set by run-shards --merge-failures).
     #
     # Output: +tmp/polyrun_failures/polyrun-failure-fragment-<workerN|shardM-workerN>.jsonl+
     # (same basename rules as {Coverage::CollectorFragmentMeta}.)

data/lib/polyrun/reporting/rspec_junit.rb

@@ -21,8 +21,8 @@ module Polyrun
         require "rspec/core"
         require "rspec/core/formatters/json_formatter"
 
-        RSpec.configure do |config|
-          config.add_formatter RSpec::Core::Formatters::JsonFormatter, json_abs
+        ::RSpec.configure do |config|
+          config.add_formatter ::RSpec::Core::Formatters::JsonFormatter, json_abs
         end
 
         at_exit do

data/lib/polyrun/{rspec → rspec_support}/active_record_timeout_recovery.rb

@@ -1,5 +1,5 @@
 module Polyrun
-  module RSpec
+  module RSpecSupport
     # Drops pooled connections after per-example Timeout interrupts database I/O.
     # Uses only ActiveRecord connection-pool APIs (PostgreSQL, MySQL, SQLite, SQL Server, etc.).
     module ActiveRecordTimeoutRecovery

data/lib/polyrun/{rspec → rspec_support}/example_debug.rb

@@ -3,7 +3,7 @@ require "timeout"
 require_relative "example_debug_instrumentation"
 
 module Polyrun
-  module RSpec
+  module RSpecSupport
     # Per-example RSpec debugging for local investigation (SQL, TracePoint, timeouts).
     #
     # +POLYRUN_EXAMPLE_DEBUG=1+ enables print helpers, Rails logging, and disables per-example timeouts.

data/lib/polyrun/{rspec → rspec_support}/example_debug_instrumentation.rb

@@ -1,5 +1,5 @@
 module Polyrun
-  module RSpec
+  module RSpecSupport
     module ExampleDebug
       SKIPPED_SQL_PREFIX = /\A(?:SELECT|SET|SHOW|BEGIN|COMMIT|ROLLBACK|RELEASE|SAVEPOINT)/
 

data/lib/polyrun/{rspec → rspec_support}/sharded_formatter_compat.rb

@@ -1,5 +1,5 @@
 module Polyrun
-  module RSpec
+  module RSpecSupport
     # Formatter tweaks for progress formatters under +POLYRUN_SHARD_*+ workers.
     module ShardedFormatterCompat
       TextFormatterSilencer = Module.new do

data/lib/polyrun/{rspec.rb → rspec_support.rb}

@@ -1,8 +1,8 @@
 require_relative "../polyrun"
 
 module Polyrun
-  # Optional RSpec wiring (require +polyrun/rspec+ explicitly).
-  module RSpec
+  # Optional RSpec wiring (require +polyrun/rspec_support+ explicitly).
+  module RSpecSupport
     module_function
 
     # Registers +before(:suite)+ to run {Data::ParallelProvisioning.run_suite_hooks!}.
@@ -74,31 +74,31 @@ module Polyrun
     end
 
     def install_example_debug!(rspec_config: nil)
-      require_relative "rspec/example_debug"
-      Polyrun::RSpec::ExampleDebug.install!(rspec_config: rspec_config || fetch_rspec_configuration!)
+      require_relative "rspec_support/example_debug"
+      Polyrun::RSpecSupport::ExampleDebug.install!(rspec_config: rspec_config || fetch_rspec_configuration!)
     end
 
     def install_example_timeout!(rspec_config: nil, seconds: ENV.fetch("RSPEC_EXAMPLE_TIMEOUT_SEC", "30").to_f)
-      require_relative "rspec/example_debug"
-      Polyrun::RSpec::ExampleDebug.install_example_timeout!(
+      require_relative "rspec_support/example_debug"
+      Polyrun::RSpecSupport::ExampleDebug.install_example_timeout!(
         rspec_config || fetch_rspec_configuration!,
         seconds: seconds
       )
     end
 
     def install_example_rails_logging!(rspec_config: nil)
-      require_relative "rspec/example_debug"
-      Polyrun::RSpec::ExampleDebug.install_rails_logging!(rspec_config: rspec_config || fetch_rspec_configuration!)
+      require_relative "rspec_support/example_debug"
+      Polyrun::RSpecSupport::ExampleDebug.install_rails_logging!(rspec_config: rspec_config || fetch_rspec_configuration!)
     end
 
     def install_example_prosopite!(rspec_config: nil)
-      require_relative "rspec/example_debug"
-      Polyrun::RSpec::ExampleDebug.install_prosopite!(rspec_config: rspec_config || fetch_rspec_configuration!)
+      require_relative "rspec_support/example_debug"
+      Polyrun::RSpecSupport::ExampleDebug.install_prosopite!(rspec_config: rspec_config || fetch_rspec_configuration!)
     end
 
     def install_sharded_formatter_compat!(rspec_config: nil)
-      require_relative "rspec/sharded_formatter_compat"
-      Polyrun::RSpec::ShardedFormatterCompat.install!(rspec_config: rspec_config || fetch_rspec_configuration!)
+      require_relative "rspec_support/sharded_formatter_compat"
+      Polyrun::RSpecSupport::ShardedFormatterCompat.install!(rspec_config: rspec_config || fetch_rspec_configuration!)
     end
 
     def fetch_rspec_configuration!

data/lib/polyrun/sus_support.rb

@@ -0,0 +1,55 @@
+require_relative "../polyrun"
+
+module Polyrun
+  # Optional Sus-oriented wiring (require +polyrun/sus_support+ explicitly).
+  #
+  # Does not load the +sus+ gem. Call installers from +config/sus.rb+ after Sus is available
+  # when a hook needs +Sus::Config+.
+  module SusSupport
+    module WorkerPingConfigHook
+      def before_tests(assertions, output: self.output)
+        Polyrun::WorkerPing.ping!(location: "sus:before_tests")
+        super
+      end
+
+      def after_tests(assertions, output: self.output)
+        super
+        Polyrun::WorkerPing.ping!(location: "sus:after_tests")
+      end
+    end
+
+    module_function
+
+    # Runs {Data::ParallelProvisioning.run_suite_hooks!} (serial vs shard worker hooks).
+    def install_parallel_provisioning!
+      Polyrun::Data::ParallelProvisioning.run_suite_hooks!
+    end
+
+    # Start {Coverage::Collector} when Polyrun or Covered coverage env is on.
+    # Prefer this for shard fragment merge. If Covered::Sus is already included on +Sus::Config+,
+    # skip starting the collector so +Coverage.start+ is not contested in one process.
+    def install_coverage!(root: Dir.pwd, **kwargs)
+      require_relative "coverage/collector"
+      return unless Polyrun::Coverage::Collector.coverage_requested_for_quick?(root)
+
+      if defined?(::Covered::Sus) && defined?(::Sus::Config) && ::Sus::Config.ancestors.include?(::Covered::Sus)
+        Polyrun::Log.warn "polyrun sus_support: install_coverage! skipped (Covered::Sus already on Sus::Config; use one coverage collector per process)"
+        return
+      end
+
+      Polyrun::Coverage::Collector.start!(root: root, **kwargs)
+    end
+
+    # Ping at Sus +before_tests+ / +after_tests+ when +Sus::Config+ is defined.
+    def install_worker_ping!
+      require_relative "worker_ping"
+      unless defined?(::Sus::Config)
+        Polyrun::Log.warn "polyrun sus_support: install_worker_ping! skipped (load Sus before calling)"
+        return
+      end
+
+      ::Sus::Config.prepend(WorkerPingConfigHook)
+      Polyrun::WorkerPing.ensure_interval_ping_thread!
+    end
+  end
+end

data/lib/polyrun/templates/POLYRUN.md

@@ -42,7 +42,7 @@ Do not combine Model A and Model B in one workflow without a documented reason (
 
 ## Spec quality (optional)
 
-- `POLYRUN_SPEC_QUALITY=1` and `Polyrun::RSpec.install_spec_quality!` in `spec_helper` (requires coverage enabled for line deltas).
+- `POLYRUN_SPEC_QUALITY=1` and `Polyrun::RSpecSupport.install_spec_quality!` in `spec_helper` (requires coverage enabled for line deltas).
 - `config/polyrun_spec_quality.yml` — thresholds and CI gates (`polyrun init --profile spec-quality`).
 - Parallel: `run-shards --merge-spec-quality` → `coverage/polyrun-spec-quality.json` → `report-spec-quality`.
 

data/lib/polyrun/timing/rspec_example_formatter.rb

@@ -10,7 +10,7 @@ module Polyrun
     # Use after RSpec is loaded:
     #   require "polyrun/timing/rspec_example_formatter"
     #   RSpec.configure { |c| c.add_formatter Polyrun::Timing::RSpecExampleFormatter }
-    # Or {Polyrun::RSpec.install_example_timing!} (+output_path:+ avoids touching +ENV+).
+    # Or {Polyrun::RSpecSupport.install_example_timing!} (+output_path:+ avoids touching +ENV+).
     #
     # Default output path: +ENV["POLYRUN_EXAMPLE_TIMING_OUT"]+ if set, else +polyrun_timing_examples.json+.
     class RSpecExampleFormatter
@@ -44,7 +44,7 @@ module Polyrun
         File.write(timing_output_path, JSON.pretty_generate(@times))
       end
 
-      # Override in a subclass from {Polyrun::RSpec.install_example_timing!(output_path: ...)}.
+      # Override in a subclass from {Polyrun::RSpecSupport.install_example_timing!(output_path: ...)}.
       def timing_output_path
         ENV["POLYRUN_EXAMPLE_TIMING_OUT"] || "polyrun_timing_examples.json"
       end

data/lib/polyrun/version.rb

@@ -1,3 +1,3 @@
 module Polyrun
-  VERSION = "2.2.9"
+  VERSION = "3.0.0"
 end

data/lib/polyrun/worker_ping.rb

@@ -7,11 +7,11 @@ module Polyrun
   #
   # Prefer framework installs (call from helpers *after* loading the runner):
   #
-  #   require "polyrun/rspec"
-  #   Polyrun::RSpec.install_worker_ping!
+  #   require "polyrun/rspec_support"
+  #   Polyrun::RSpecSupport.install_worker_ping!
   #
-  #   require "polyrun/minitest"
-  #   Polyrun::Minitest.install_worker_ping!
+  #   require "polyrun/minitest_support"
+  #   Polyrun::MinitestSupport.install_worker_ping!
   #
   # Polyrun Quick runs +ping!+ automatically when requiring the Quick stack.
   #
@@ -19,6 +19,10 @@ module Polyrun
   # {#ensure_interval_ping_thread!} once at worker startup if you rely on periodic pings without per-example {#ping!};
   # installers call this so the env toggle works out of the box.
   module WorkerPing
+    # Load-time constant: `||=` can create two mutexes.
+    INTERVAL_PING_MX = Mutex.new
+    private_constant :INTERVAL_PING_MX
+
     class << self
       def ping!(location: nil)
         path = ping_file_path
@@ -46,8 +50,7 @@ module Polyrun
         path = ping_file_path
         return if path.empty?
 
-        @interval_ping_mx ||= Mutex.new
-        @interval_ping_mx.synchronize do
+        INTERVAL_PING_MX.synchronize do
           return if @interval_ping_started
 
           raw = ENV["POLYRUN_WORKER_PING_INTERVAL_SEC"].to_s.strip

data/sig/polyrun/{minitest.rbs → minitest_support.rbs}

@@ -1,5 +1,5 @@
 module Polyrun
-  module Minitest
+  module MinitestSupport
     def self.install_parallel_provisioning!: () -> void
 
     def self.install_worker_ping!: () -> void

data/sig/polyrun/{rspec.rbs → rspec_support.rbs}

@@ -1,5 +1,5 @@
 module Polyrun
-  module RSpec
+  module RSpecSupport
     def self.install_parallel_provisioning!: (untyped rspec_config) -> void
 
     def self.install_example_timing!: (?output_path: String? ) -> void

data/sig/polyrun/sus_support.rbs

@@ -0,0 +1,7 @@
+module Polyrun
+  module SusSupport
+    def self.install_parallel_provisioning!: () -> void
+    def self.install_coverage!: (?root: String, **untyped) -> void
+    def self.install_worker_ping!: () -> void
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: polyrun
 version: !ruby/object:Gem::Version
-  version: 2.2.9
+  version: 3.0.0
 platform: ruby
 authors:
 - Andrei Makarov
@@ -261,6 +261,7 @@ files:
 - lib/polyrun/config/suite.rb
 - lib/polyrun/coverage/cobertura_zero_lines.rb
 - lib/polyrun/coverage/collector.rb
+- lib/polyrun/coverage/collector_blob.rb
 - lib/polyrun/coverage/collector_finish.rb
 - lib/polyrun/coverage/collector_fragment_meta.rb
 - lib/polyrun/coverage/config_resolve.rb
@@ -311,7 +312,7 @@ files:
 - lib/polyrun/hooks/worker_runner.rb
 - lib/polyrun/hooks/worker_shell.rb
 - lib/polyrun/log.rb
-- lib/polyrun/minitest.rb
+- lib/polyrun/minitest_support.rb
 - lib/polyrun/partition/constraints.rb
 - lib/polyrun/partition/hrw.rb
 - lib/polyrun/partition/min_heap.rb
@@ -335,10 +336,12 @@ files:
 - lib/polyrun/queue/worker_loop.rb
 - lib/polyrun/quick.rb
 - lib/polyrun/quick/assertions.rb
+- lib/polyrun/quick/config.rb
 - lib/polyrun/quick/errors.rb
 - lib/polyrun/quick/example_group.rb
 - lib/polyrun/quick/example_runner.rb
 - lib/polyrun/quick/matchers.rb
+- lib/polyrun/quick/matchers_predicates.rb
 - lib/polyrun/quick/reporter.rb
 - lib/polyrun/quick/runner.rb
 - lib/polyrun/railtie.rb
@@ -348,11 +351,11 @@ files:
 - lib/polyrun/reporting/junit_report.rb
 - lib/polyrun/reporting/rspec_failure_fragment_formatter.rb
 - lib/polyrun/reporting/rspec_junit.rb
-- lib/polyrun/rspec.rb
-- lib/polyrun/rspec/active_record_timeout_recovery.rb
-- lib/polyrun/rspec/example_debug.rb
-- lib/polyrun/rspec/example_debug_instrumentation.rb
-- lib/polyrun/rspec/sharded_formatter_compat.rb
+- lib/polyrun/rspec_support.rb
+- lib/polyrun/rspec_support/active_record_timeout_recovery.rb
+- lib/polyrun/rspec_support/example_debug.rb
+- lib/polyrun/rspec_support/example_debug_instrumentation.rb
+- lib/polyrun/rspec_support/sharded_formatter_compat.rb
 - lib/polyrun/spec_quality.rb
 - lib/polyrun/spec_quality/config.rb
 - lib/polyrun/spec_quality/fragment.rb
@@ -363,6 +366,7 @@ files:
 - lib/polyrun/spec_quality/report.rb
 - lib/polyrun/spec_quality/rspec_hook.rb
 - lib/polyrun/spec_quality/sql_counter.rb
+- lib/polyrun/sus_support.rb
 - lib/polyrun/templates/POLYRUN.md
 - lib/polyrun/templates/ci_matrix.polyrun.yml
 - lib/polyrun/templates/minimal_gem.polyrun.yml
@@ -385,9 +389,10 @@ files:
 - sig/polyrun/config.rbs
 - sig/polyrun/debug.rbs
 - sig/polyrun/log.rbs
-- sig/polyrun/minitest.rbs
+- sig/polyrun/minitest_support.rbs
 - sig/polyrun/quick.rbs
-- sig/polyrun/rspec.rbs
+- sig/polyrun/rspec_support.rbs
+- sig/polyrun/sus_support.rbs
 - sig/polyrun/worker_output.rbs
 - sig/polyrun/worker_ping.rbs
 homepage: https://github.com/amkisko/polyrun.rb