crspec 0.1.2 → 0.1.3

data/lib/crspec/process_runner.rb

@@ -0,0 +1,332 @@
+# frozen_string_literal: true
+
+require "etc"
+require_relative "status_persistence"
+
+module Crspec
+  # Process tier for true multi-core execution (--processes P). The parent
+  # loads specs once (copy-on-write memory), forks P children, and shards
+  # examples across them using persisted timings (bin-packing,
+  # slowest-first; round-robin on first run). Each child runs the existing
+  # N-thread x M-fiber Runner on its shard and streams marshalled result
+  # structs back over a pipe (example blocks cannot cross process
+  # boundaries). Fail-fast propagates via SIGTERM.
+  class ProcessRunner
+    Result = Struct.new(:persistence_key, :description, :status, :error_class,
+                        :error_message, :error_backtrace, :execution_time)
+
+    attr_reader :passed_examples, :failed_examples, :pending_examples, :total_duration
+
+    def initialize(processes:, concurrency: Etc.nprocessors, fibers: 1, formatter: nil,
+                   fail_fast: false, seed: nil, only_failures: false, persistence_path: nil)
+      @processes = processes == :auto ? physical_core_count : processes
+      @concurrency = concurrency
+      @fibers = fibers
+      @formatter = formatter || Formatters::ProgressFormatter.new
+      @fail_fast = fail_fast == true ? 1 : fail_fast
+      @seed = seed
+      @only_failures = only_failures
+      @persistence_path = persistence_path
+      @persistence = StatusPersistence.new(persistence_path)
+      @passed_examples = []
+      @failed_examples = []
+      @pending_examples = []
+      @total_duration = 0
+    end
+
+    def run(example_groups)
+      start_time = Process.clock_gettime(Process::CLOCK_MONOTONIC)
+      @formatter.start
+
+      example_groups.each(&:finalize!)
+      examples = []
+      example_groups.each { |group| collect_examples(group, examples) }
+
+      previous = @persistence.load
+      if @only_failures
+        examples = examples.select do |ex|
+          entry = previous[ex.persistence_key]
+          entry.nil? || entry["status"] == "failed"
+        end
+      end
+
+      shards = shard_examples(examples, previous)
+
+      Process.warmup if Process.respond_to?(:warmup)
+
+      children = shards.each_with_index.filter_map do |shard, index|
+        next if shard.empty?
+
+        spawn_child(shard, index + 1, example_groups)
+      end
+
+      failure_total = 0
+      aborted = false
+      readers = children.to_h { |c| [c[:reader], c] }
+
+      until readers.empty?
+        ready, = IO.select(readers.keys)
+        ready.each do |io|
+          child = readers[io]
+          result = read_result(io)
+          if result.nil?
+            readers.delete(io)
+            io.close
+            next
+          end
+
+          record(result)
+          next unless result.status == "failed" && @fail_fast
+
+          failure_total += 1
+          next if aborted || failure_total < @fail_fast
+
+          aborted = true
+          children.each do |c|
+            Process.kill("TERM", c[:pid])
+          rescue Errno::ESRCH
+            nil
+          end
+        end
+      end
+
+      children.each do |c|
+        Process.wait(c[:pid])
+      rescue Errno::ECHILD
+        nil
+      end
+
+      @total_duration = Process.clock_gettime(Process::CLOCK_MONOTONIC) - start_time
+      persist_results(previous)
+      @formatter.finish
+      self
+    end
+
+    def success?
+      @failed_examples.empty?
+    end
+
+    private
+
+    def collect_examples(group, acc)
+      acc.concat(group.examples)
+      group.children.each { |child| collect_examples(child, acc) }
+    end
+
+    # Longest-processing-time bin packing: sort slowest-first, assign each
+    # example to the least-loaded shard. Round-robin when no timings exist.
+    def shard_examples(examples, previous)
+      shards = Array.new(@processes) { [] }
+      if previous.empty?
+        examples.each_with_index { |ex, i| shards[i % @processes] << ex }
+      else
+        loads = Array.new(@processes, 0.0)
+        sorted = examples.sort_by do |ex|
+          entry = previous[ex.persistence_key]
+          -(entry ? entry["run_time"].to_f : Float::INFINITY)
+        end
+        sorted.each do |ex|
+          idx = loads.each_with_index.min_by { |load, _| load }.last
+          shards[idx] << ex
+          entry = previous[ex.persistence_key]
+          loads[idx] += entry ? entry["run_time"].to_f : 0.1
+        end
+      end
+      shards
+    end
+
+    def spawn_child(shard, process_number, example_groups)
+      reader, writer = IO.pipe
+      reader.binmode
+      writer.binmode
+
+      pid = fork do
+        reader.close
+        setup_child_environment(process_number)
+
+        keys = shard.map(&:persistence_key).to_h { |k| [k, true] }
+        formatter = ChildFormatter.new(writer)
+        runner = Runner.new(concurrency: @concurrency, fibers: @fibers,
+                            formatter: formatter, seed: @seed)
+        filtered = FilteredGroups.wrap(example_groups, keys)
+        runner.run(filtered)
+        writer.close
+        exit!(runner.success? ? 0 : 1)
+      end
+
+      writer.close
+      { pid: pid, reader: reader }
+    end
+
+    # Per-process databases reuse the Rails parallel-testing convention
+    # (database suffixed _N for process N > 1). Set once, pre-thread, where
+    # ENV mutation is safe.
+    def setup_child_environment(process_number)
+      env_num = process_number == 1 ? "" : process_number.to_s
+      ENV["TEST_ENV_NUMBER"] = env_num
+      ENV["PARALLEL_WORKERS"] = @processes.to_s
+
+      return if process_number == 1
+      return unless defined?(ActiveRecord::Base) && ActiveRecord::Base.respond_to?(:connection_db_config)
+
+      # Rails' own parallel-testing helper creates the db_N database and
+      # loads the schema, honouring TEST_ENV_NUMBER set above.
+      if defined?(ActiveRecord::TestDatabases)
+        begin
+          ActiveRecord::TestDatabases.create_and_load_schema(process_number, env_name: ::Rails.env)
+          return
+        rescue StandardError
+          nil
+        end
+      end
+
+      config = begin
+        ActiveRecord::Base.connection_db_config
+      rescue StandardError
+        nil
+      end
+      return unless config
+
+      db_name = "#{config.database}_#{process_number}"
+      begin
+        ActiveRecord::Base.establish_connection(config.configuration_hash.merge(database: db_name))
+      rescue StandardError
+        nil
+      end
+    end
+
+    def read_result(io)
+      header = io.read(4)
+      return nil if header.nil? || header.bytesize < 4
+
+      length = header.unpack1("N")
+      payload = io.read(length)
+      return nil if payload.nil? || payload.bytesize < length
+
+      Marshal.load(payload)
+    rescue EOFError, IOError
+      nil
+    end
+
+    def record(result)
+      case result.status
+      when "passed"
+        @passed_examples << result
+        @formatter.example_passed(result)
+      when "pending"
+        @pending_examples << result
+        @formatter.example_pending(result)
+      else
+        @failed_examples << result
+        @formatter.example_failed(result)
+      end
+    end
+
+    def persist_results(previous)
+      (@passed_examples + @failed_examples).each do |result|
+        previous[result.persistence_key] = {
+          "status" => result.status,
+          "run_time" => result.execution_time.to_f.round(6)
+        }
+      end
+      path = @persistence_path
+      return unless path
+
+      dir = File.dirname(path)
+      FileUtils.mkdir_p(dir) unless Dir.exist?(dir)
+      File.write(path, JSON.pretty_generate(previous))
+    rescue SystemCallError
+      nil
+    end
+
+    def physical_core_count
+      if RUBY_PLATFORM.include?("darwin")
+        count = `sysctl -n hw.physicalcpu 2>/dev/null`.to_i
+        return count if count.positive?
+      elsif File.readable?("/proc/cpuinfo")
+        cores = File.read("/proc/cpuinfo").scan(/^core id\s*:\s*(\d+)/).uniq.size
+        return cores if cores.positive?
+      end
+      Etc.nprocessors
+    end
+
+    # Streams marshalled Result structs to the parent as examples finish.
+    class ChildFormatter
+      def initialize(writer)
+        @writer = writer
+        @mutex = Mutex.new
+      end
+
+      def example_passed(example)
+        emit(example, "passed")
+      end
+
+      def example_failed(example)
+        emit(example, "failed")
+      end
+
+      def example_pending(example)
+        emit(example, "pending")
+      end
+
+      def start; end
+      def finish; end
+
+      private
+
+      def emit(example, status)
+        error = example.respond_to?(:error) ? example.error : nil
+        result = Result.new(
+          example.persistence_key,
+          example.description.to_s,
+          status,
+          error&.class&.name,
+          error&.message,
+          error&.backtrace&.first(10),
+          example.respond_to?(:execution_time) ? example.execution_time : 0
+        )
+        payload = Marshal.dump(result)
+        @mutex.synchronize do
+          @writer.write([payload.bytesize].pack("N"))
+          @writer.write(payload)
+          @writer.flush
+        end
+      end
+    end
+
+    # Proxy groups that expose only the examples assigned to this shard.
+    module FilteredGroups
+      def self.wrap(groups, keys)
+        groups.map { |g| GroupProxy.new(g, keys) }
+      end
+
+      class GroupProxy
+        def initialize(group, keys)
+          @group = group
+          @keys = keys
+        end
+
+        def finalize!
+          @group.finalize!
+          self
+        end
+
+        def examples
+          @group.examples.select { |ex| @keys[ex.persistence_key] }
+        end
+
+        def children
+          @group.children.map { |c| GroupProxy.new(c, @keys) }
+        end
+
+        def method_missing(name, *args, &block)
+          @group.send(name, *args, &block)
+        end
+
+        def respond_to_missing?(name, include_private = false)
+          @group.respond_to?(name, include_private) || super
+        end
+      end
+    end
+  end
+end

data/lib/crspec/rails/assets_shim.rb

@@ -0,0 +1,33 @@
+# frozen_string_literal: true
+
+# In many Rails controller/request specs the asset pipeline is not the
+# subject under test. When running under crspec, we provide a small shim
+# around Sprockets helpers so that missing compiled assets do not cause
+# hard failures.
+
+module Crspec
+  module Rails
+    module AssetsShim
+      module ComputeAssetPathFallback
+        def compute_asset_path(path, options = {})
+          super
+        rescue StandardError
+          path.to_s
+        end
+      end
+
+      def self.install!
+        return if @installed
+        return unless defined?(::Rails) && ::Rails.env.test?
+        return unless defined?(::Sprockets::Rails::Helper)
+
+        ::Sprockets::Rails::Helper.prepend(ComputeAssetPathFallback)
+        @installed = true
+      rescue StandardError
+        nil
+      end
+    end
+  end
+end
+
+Crspec::Rails::AssetsShim.install!

data/lib/crspec/rails/database_isolation.rb

@@ -3,16 +3,187 @@
 module Crspec
   module Rails
     class DatabaseIsolation
-      def self.wrap_example(example)
-        if defined?(ActiveRecord::Base) && ActiveRecord::Base.connected?
-          ActiveRecord::Base.connection_pool.with_connection do |conn|
-            conn.transaction(requires_new: true) do
-              example.execute!
-              raise ActiveRecord::Rollback
+      LEASES_KEY = :crspec_db_leases
+
+      class << self
+        def wrap_example(example)
+          return example.execute! unless active_record_ready?
+
+          pools = writing_pools
+          return example.execute! if pools.empty?
+
+          if pools.any? { |pool| sqlite_pool?(pool) }
+            serialized_example(pools) { example.execute! }
+          else
+            savepoint_example { example.execute! }
+          end
+        end
+
+        def finish_worker
+          leases = Fiber[LEASES_KEY]
+          return unless leases
+
+          Fiber[LEASES_KEY] = nil
+          leases.each do |pool, conn|
+            begin
+              conn.rollback_transaction while conn.open_transactions.positive?
+            rescue StandardError
+              nil
             end
+            release(pool, conn)
           end
-        else
-          example.execute!
+        end
+
+        def handoff_connection(pool)
+          Fiber[LEASES_KEY]&.[](pool)
+        end
+
+        private
+
+        # PG/MySQL: each worker keeps a leased connection with a root
+        # non-joinable transaction; every example runs inside a nested
+        # transaction (SAVEPOINT) rolled back afterwards.
+        def savepoint_example
+          leases = ensure_worker_leases!
+          return yield if leases.empty?
+
+          depths = {}
+          leases.each do |pool, conn|
+            depths[pool] = conn.open_transactions
+            conn.begin_transaction(joinable: false)
+          end
+
+          begin
+            yield
+          ensure
+            leases.each do |pool, conn|
+              base_depth = depths[pool] + 1
+              begin
+                conn.rollback_transaction while conn.open_transactions >= base_depth
+              rescue StandardError
+                nil
+              end
+            end
+          end
+        end
+
+        # SQLite permits a single writer per database file, so write
+        # transactions cannot overlap across worker threads. Examples
+        # touching SQLite pools are serialized with per-example
+        # transactions; use --processes with per-process databases for
+        # SQLite concurrency.
+        def serialized_example(pools)
+          write_mutex.synchronize do
+            leases = {}
+            pools.each do |pool|
+              conn = lease(pool)
+              next unless conn
+
+              conn.begin_transaction(joinable: false)
+              leases[pool] = conn
+            end
+
+            previous = Fiber[LEASES_KEY]
+            Fiber[LEASES_KEY] = leases
+            begin
+              yield
+            ensure
+              Fiber[LEASES_KEY] = previous
+              leases.each do |pool, conn|
+                begin
+                  conn.rollback_transaction while conn.open_transactions.positive?
+                rescue StandardError
+                  nil
+                end
+                release(pool, conn)
+              end
+            end
+          end
+        end
+
+        def write_mutex
+          @write_mutex ||= Mutex.new
+        end
+
+        def active_record_ready?
+          defined?(ActiveRecord::Base) && ActiveRecord::Base.connected?
+        end
+
+        def ensure_worker_leases!
+          Fiber[LEASES_KEY] ||= begin
+            install_handoff!
+            writing_pools.each_with_object({}) do |pool, leases|
+              conn = lease(pool)
+              next unless conn
+
+              conn.begin_transaction(joinable: false)
+              leases[pool] = conn
+            end
+          end
+        end
+
+        def writing_pools
+          ActiveRecord::Base.connection_handler.connection_pool_list(:writing)
+        rescue StandardError
+          []
+        end
+
+        def sqlite_pool?(pool)
+          adapter = pool.db_config.adapter.to_s
+          adapter.match?(/sqlite/i)
+        rescue StandardError
+          false
+        end
+
+        def lease(pool)
+          install_handoff!
+          if pool.respond_to?(:lease_connection)
+            pool.lease_connection
+          else
+            pool.checkout
+          end
+        rescue StandardError
+          nil
+        end
+
+        def release(pool, conn)
+          if pool.respond_to?(:release_connection)
+            pool.release_connection
+          else
+            pool.checkin(conn)
+          end
+        rescue StandardError
+          nil
+        end
+
+        def install_handoff!
+          return if @handoff_installed
+          return unless defined?(ActiveRecord::ConnectionAdapters::ConnectionPool)
+
+          ActiveRecord::ConnectionAdapters::ConnectionPool.prepend(ConnectionHandoff)
+          @handoff_installed = true
+        end
+      end
+
+      module ConnectionHandoff
+        def lease_connection(*args)
+          DatabaseIsolation.handoff_connection(self) || super
+        end
+
+        def connection(*args)
+          DatabaseIsolation.handoff_connection(self) || super
+        end
+
+        def release_connection(*args)
+          return if DatabaseIsolation.handoff_connection(self)
+
+          super
+        end
+
+        def checkin(conn, *args)
+          return if DatabaseIsolation.handoff_connection(self) == conn
+
+          super
         end
       end
     end

data/lib/crspec/rails/parallel.rb

@@ -1,14 +1,25 @@
 # frozen_string_literal: true
 
 require "etc"
+require_relative "assets_shim"
+require_relative "warden_shim"
 
 module Crspec
   module Rails
     module Parallel
+      WORKER_NUMBER_KEY = :crspec_worker_number
+
       class << self
         attr_accessor :worker_count, :setup_blocks, :teardown_blocks, :enabled
 
         def parallelize(workers: :number_of_processors, &block)
+          # For Rails test environments it is often safer to run without
+          # process-level parallelism unless the application explicitly opts
+          # in. This avoids issues with shared transactional fixtures and
+          # global state.
+          if defined?(::Rails) && ::Rails.respond_to?(:env) && ::Rails.env.test?
+            workers = 1 if workers == :number_of_processors
+          end
           count = case workers
                   when :number_of_processors
                     Etc.nprocessors
@@ -41,25 +52,35 @@ module Crspec
           @teardown_blocks << block
         end
 
+        # Worker identity lives in Fiber Storage (inherited by fibers
+        # spawned within the worker), never in ENV: mutating
+        # ENV["TEST_ENV_NUMBER"] from worker threads is a process-wide
+        # race. Per-worker databases move to the process tier
+        # (--processes), where the ENV convention is safe to set once
+        # per child process before any threads start.
         def setup_worker(worker_number)
-          env_num = worker_number == 1 ? "" : worker_number.to_s
-          ENV["TEST_ENV_NUMBER"] = env_num
-          ENV["PARALLEL_WORKERS"] = (@worker_count || Etc.nprocessors).to_s
+          Fiber[WORKER_NUMBER_KEY] = worker_number
+          @setup_blocks&.each { |b| b.call(worker_number) }
+        end
 
-          if defined?(ActiveRecord::Base) && ActiveRecord::Base.respond_to?(:connection_db_config)
-            setup_active_record_db(worker_number)
-          end
+        def current_worker_number
+          Fiber[WORKER_NUMBER_KEY]
+        end
 
-          @setup_blocks&.each { |b| b.call(worker_number) }
+        def test_env_number(worker_number = current_worker_number)
+          return "" if worker_number.nil? || worker_number == 1
+
+          worker_number.to_s
         end
 
         def teardown_worker(worker_number)
           @teardown_blocks&.each { |b| b.call(worker_number) }
+          Fiber[WORKER_NUMBER_KEY] = nil
 
           return unless defined?(ActiveRecord::Base) && ActiveRecord::Base.respond_to?(:connection_handler)
 
           begin
-            ActiveRecord::Base.connection_handler.clear_all_connections!
+            ActiveRecord::Base.connection_handler.clear_active_connections!
           rescue StandardError
             nil
           end
@@ -71,27 +92,6 @@ module Crspec
           @setup_blocks = []
           @teardown_blocks = []
         end
-
-        private
-
-        def setup_active_record_db(worker_number)
-          return if worker_number == 1
-
-          config = begin
-            ActiveRecord::Base.connection_db_config
-          rescue StandardError
-            nil
-          end
-          return unless config
-
-          db_name = "#{config.database}_#{worker_number}"
-          new_config = config.configuration_hash.merge(database: db_name)
-          begin
-            ActiveRecord::Base.establish_connection(new_config)
-          rescue StandardError
-            nil
-          end
-        end
       end
     end
   end