test-queue 0.10.0 → 0.11.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cf82719e0483ccba7f40a8a66ffdd0d262bd2ff79cae203055e730988eed1219
-  data.tar.gz: 584b68033ee5540cc1c2aaeeb1c391e04ff67b158e6c170975151397ea287c66
+  metadata.gz: 63e449555d271631ef49690393447ae267535a8c8506874ae3217d1992c51e47
+  data.tar.gz: 4f434b3bb3a7ca9040836751a0f1c08e198552aeda35956adb9353450b8b01dd
 SHA512:
-  metadata.gz: 041b553cf8d565683e9bf304e043ca0e44346a14243969834d4dc5e3d98f8ada5c96ada0e42ad0da4d4ccc488322013b632b6f2cde4633908d7b526915c7af74
-  data.tar.gz: ac21ecddc2d3236e5b7de2bc0d138ac791d420efea33a43286294169fba4082329de1540c614153273cc0783e8897b9700bd24469fbad660168c22ec86c9ae65
+  metadata.gz: 65cb68eb9661a6c6cfe813a41aed60f07962d7ba7b30a2b5ee178cc6c10fce481cfbb82ed83c7e9741dae71dc5d5be954b57f8fe1997c182497e332b5e342102
+  data.tar.gz: 7ab7dc0e7b758966cec5cb900e7e27d6538aee8639838401bedd781ae5457051085fa587d96c30d8c2e1b0c2a4a005f42af70533fe4464fb8e1956e4f8002271

data/lib/test_queue/runner/testunit.rb

@@ -4,57 +4,65 @@ require_relative '../runner'
 
 gem 'test-unit'
 require 'test/unit'
-require 'test/unit/collector/descendant'
-require 'test/unit/testresult'
-require 'test/unit/testsuite'
+require 'test/unit/collector/load'
 require 'test/unit/ui/console/testrunner'
 
-class Test::Unit::TestSuite
-  attr_accessor :iterator
-
-  def run(result, &progress_block)
-    @start_time = Time.now
-    yield(STARTED, name)
-    yield(STARTED_OBJECT, self)
-    run_startup(result)
-    (@iterator || @tests).each do |test|
-      @n_tests += test.size
-      run_test(test, result, &progress_block)
-      @passed = false unless test.passed?
-    end
-    run_shutdown(result)
-  ensure
-    @elapsed_time = Time.now - @start_time
-    yield(FINISHED, name)
-    yield(FINISHED_OBJECT, self)
-  end
-
-  def failure_count
-    (@iterator || @tests).sum { |t| t.instance_variable_get(:@_result).failure_count }
-  end
-end
-
 module TestQueue
   class Runner
     class TestUnit < Runner
-      def initialize
-        if Test::Unit::Collector::Descendant.new.collect.tests.any?
-          raise 'Do not `require` test files. Pass them via ARGV instead and they will be required as needed.'
+      class TestSuite < ::Test::Unit::TestSuite
+        def initialize(name, iterator)
+          super(name)
+          @tests = IteratorWrapper.new(iterator)
+        end
+
+        def run(*)
+          @started = true
+          super
+        end
+
+        def size
+          return 0 unless @started
+
+          super
+        end
+      end
+
+      class IteratorWrapper
+        def initialize(iterator)
+          @generator = Fiber.new do
+            iterator.each do |test|
+              Fiber.yield(test)
+            end
+          end
+        end
+
+        def shift
+          @generator.resume
+        rescue FiberError
+          nil
         end
 
+        def each
+          while (test = shift)
+            yield(test)
+          end
+        end
+      end
+
+      def initialize
         super(TestFramework::TestUnit.new)
       end
 
       def run_worker(iterator)
-        @suite = Test::Unit::TestSuite.new('specified by test-queue master')
-        @suite.iterator = iterator
+        @suite = TestSuite.new('specified by test-queue master', iterator)
         res = Test::Unit::UI::Console::TestRunner.new(@suite).start
         res.run_count - res.pass_count
       end
 
       def summarize_worker(worker)
         worker.summary = worker.output.split("\n").grep(/^\d+ tests?/).first
-        worker.failure_output = worker.output.scan(/^Failure:\n(.*)\n=======================*/m).join("\n")
+        worker.failure_output = worker.output.scan(/^Failure:[^\n]*\n(.*?)\n=======================*/m).join("\n")
       end
     end
   end
@@ -66,11 +74,39 @@ module TestQueue
       end
 
       def suites_from_file(path)
-        Test::Unit::TestCase::DESCENDANTS.clear
-        require File.absolute_path(path)
-        Test::Unit::Collector::Descendant.new.collect.tests.map { |suite|
-          [suite.name, suite]
-        }
+        test_suite = Test::Unit::Collector::Load.new.collect(path)
+        return [] unless test_suite
+        return test_suite.tests.map { [_1.name, _1] } unless split_groups?
+
+        split_groups(test_suite)
+      end
+
+      def split_groups?
+        return @split_groups if defined?(@split_groups)
+
+        @split_groups = ENV['TEST_QUEUE_SPLIT_GROUPS'] && ENV['TEST_QUEUE_SPLIT_GROUPS'].strip.downcase == 'true'
+      end
+
+      def split_groups(test_suite, groups = [])
+        unless splittable?(test_suite)
+          groups << [test_suite.name, test_suite]
+          return groups
+        end
+
+        test_suite.tests.each do |suite|
+          if suite.is_a?(Test::Unit::TestSuite)
+            split_groups(suite, groups)
+          else
+            groups << [suite.name, suite]
+          end
+        end
+        groups
+      end
+
+      def splittable?(test_suite)
+        test_suite.tests.none? do |test|
+          test.is_a?(Test::Unit::TestCase) && test[:no_split]
+        end
       end
     end
   end

data/lib/test_queue/runner.rb

@@ -282,7 +282,7 @@ module TestQueue
       @concurrency.times do |i|
         num = i + 1
 
-        pid = fork do
+        pid = Process.fork do
           @server&.close
 
           iterator = Iterator.new(@test_framework, relay? ? @relay : @socket, method(:around_filter), early_failure_limit: @early_failure_limit, run_token: @run_token)

data/lib/test_queue/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module TestQueue
-  VERSION = '0.10.0'
+  VERSION = '0.11.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: test-queue
 version: !ruby/object:Gem::Version
-  version: 0.10.0
+  version: 0.11.1
 platform: ruby
 authors:
 - Aman Gupta
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-26 00:00:00.000000000 Z
+date: 2023-12-20 00:00:00.000000000 Z
 dependencies: []
 description: minitest/rspec parallel test runner for CI environments
 email: ruby@tmm1.net
@@ -61,7 +61,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.10
+rubygems_version: 3.4.19
 signing_key:
 specification_version: 4
 summary: parallel test runner