cucumber.rb in test-queue-0.3.0

- old
+ new
@@ -12,35 +12,104 @@
       def to_s
         title
       end
     end
   end
+
+  class Runtime
+    if defined?(::Cucumber::Runtime::FeaturesLoader)
+      # Without this module, Runtime#features would load all features specified
+      # on the command line. We want to avoid that and load only the features
+      # each worker needs ourselves, so we override the default behavior to let
+      # us put our iterator in place without loading any features directly.
+      module InjectableFeatures
+        def features
+          return @features if defined?(@features)
+          super
+        end
+
+        def features=(iterator)
+          @features = ::Cucumber::Ast::Features.new
+          @features.features = iterator
+        end
+      end
+
+      prepend InjectableFeatures
+    else
+      attr_writer :features
+    end
+  end
 end
 
 module TestQueue
   class Runner
     class Cucumber < Runner
       def initialize
-        @cli = ::Cucumber::Cli::Main.new(ARGV.dup)
-        @runtime = ::Cucumber::Runtime.new(@cli.configuration)
-        @features_loader = @runtime.send(:features)
-        features = @features_loader.features.sort_by{ |s| -(stats[s.to_s] || 0) }
-        super(features)
+        super(TestFramework::Cucumber.new)
       end
 
       def run_worker(iterator)
-        @features_loader.features = iterator
-        @cli.execute!(@runtime)
+        runtime = @test_framework.runtime
+        runtime.features = iterator
+
+        @test_framework.cli.execute!(runtime)
+
+        if runtime.respond_to?(:summary_report, true)
+          runtime.send(:summary_report).test_cases.total_failed
+        else
+          runtime.results.scenarios(:failed).size
+        end
       end
 
       def summarize_worker(worker)
-        worker.stats.each do |s, val|
-          stats[s.to_s] = val
+        output                = worker.output.gsub(/\e\[\d+./, '')
+        worker.summary        = output.split("\n").grep(/^\d+ (scenarios?|steps?)/).first
+        worker.failure_output = output.scan(/^Failing Scenarios:\n(.*)\n\d+ scenarios?/m).join("\n")
+      end
+    end
+  end
+
+  class TestFramework
+    class Cucumber < TestFramework
+      class FakeKernel
+        def exit(n)
+          if $!
+            # Let Cucumber exit for raised exceptions.
+            Kernel.exit(n)
+          end
+          # Don't let Cucumber exit to indicate test failures. We want to
+          # return the number of failures from #run_worker instead.
         end
+      end
 
-        output = worker.output.gsub(/\e\[\d+./,'')
-        worker.summary  = output.split("\n").grep(/^\d+ (scenarios?|steps?)/).first
-        worker.failure_output = output.scan(/^Failing Scenarios:\n(.*)\n\d+ scenarios?/m).join("\n")
+      def cli
+        @cli ||= ::Cucumber::Cli::Main.new(ARGV.dup, $stdin, $stdout, $stderr, FakeKernel.new)
+      end
+
+      def runtime
+        @runtime ||= ::Cucumber::Runtime.new(cli.configuration)
+      end
+
+      def all_suite_files
+        if runtime.respond_to?(:feature_files, true)
+          runtime.send(:feature_files)
+        else
+          cli.configuration.feature_files
+        end
+      end
+
+      def suites_from_file(path)
+        if defined?(::Cucumber::Core::Gherkin::Document)
+          source = ::Cucumber::Runtime::NormalisedEncodingFile.read(path)
+          doc = ::Cucumber::Core::Gherkin::Document.new(path, source)
+          [[File.basename(doc.uri), doc]]
+        else
+          loader =
+            ::Cucumber::Runtime::FeaturesLoader.new([path],
+                                                    cli.configuration.filters,
+                                                    cli.configuration.tag_expression)
+          loader.features.map { |feature| [feature.title, feature] }
+        end
       end
     end
   end
 end