knapsack_pro 3.8.0 → 7.0.0

data/lib/knapsack_pro/runners/queue/rspec_runner.rb

@@ -1,184 +1,177 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   module Runners
     module Queue
       class RSpecRunner < BaseRunner
-        @@used_seed = nil
-
-        def self.run(args)
+        def self.run(args, stream_error = $stderr, stream_out = $stdout)
           require 'rspec/core'
-          require_relative '../../formatters/rspec_queue_summary_formatter'
-          require_relative '../../formatters/rspec_queue_profile_formatter_extension'
+          require_relative '../../extensions/rspec_extension'
+          require_relative '../../formatters/time_tracker'
+          require_relative '../../formatters/time_tracker_fetcher'
+
+          KnapsackPro::Extensions::RSpecExtension.setup!
 
           ENV['KNAPSACK_PRO_TEST_SUITE_TOKEN'] = KnapsackPro::Config::Env.test_suite_token_rspec
-          ENV['KNAPSACK_PRO_QUEUE_RECORDING_ENABLED'] = 'true'
-          ENV['KNAPSACK_PRO_QUEUE_ID'] = KnapsackPro::Config::EnvGenerator.set_queue_id
 
-          KnapsackPro::Config::Env.set_test_runner_adapter(adapter_class)
-          runner = new(adapter_class)
-
-          cli_args = (args || '').split
-          adapter_class.ensure_no_tag_option_when_rspec_split_by_test_examples_enabled!(cli_args)
-
-          # when format option is not defined by user then use progress formatter to show tests execution progress
-          cli_args += ['--format', 'progress'] unless adapter_class.has_format_option?(cli_args)
-
-          cli_args += [
-            # shows summary of all tests executed in Queue Mode at the very end
-            '--format', KnapsackPro::Formatters::RSpecQueueSummaryFormatter.to_s,
-            '--default-path', runner.test_dir,
-          ]
-
-          accumulator = {
-            status: :next,
-            runner: runner,
-            can_initialize_queue: true,
-            args: cli_args,
-            exitstatus: 0,
-            all_test_file_paths: [],
-          }
-          while accumulator[:status] == :next
-            accumulator = run_tests(accumulator)
-          end
+          rspec_pure = KnapsackPro::Pure::Queue::RSpecPure.new
 
-          Kernel.exit(accumulator[:exitstatus])
+          queue_runner = new(KnapsackPro::Adapters::RSpecAdapter, rspec_pure, args, stream_error, stream_out)
+          queue_runner.run
         end
 
-        def self.run_tests(accumulator)
-          runner = accumulator.fetch(:runner)
-          can_initialize_queue = accumulator.fetch(:can_initialize_queue)
-          args = accumulator.fetch(:args)
-          exitstatus = accumulator.fetch(:exitstatus)
-          all_test_file_paths = accumulator.fetch(:all_test_file_paths)
+        def initialize(adapter_class, rspec_pure, args, stream_error, stream_out)
+          super(adapter_class)
+          @adapter_class = adapter_class
+          @rspec_pure = rspec_pure
+          has_format_option = @adapter_class.has_format_option?((args || '').split)
+          @cli_args = rspec_pure.prepare_cli_args(args, has_format_option, test_dir)
+          @stream_error = stream_error
+          @stream_out = stream_out
+          @node_test_file_paths = []
+          @rspec_runner = nil # RSpec::Core::Runner is lazy initialized
+        end
 
-          test_file_paths = runner.test_file_paths(
-            can_initialize_queue: can_initialize_queue,
-            executed_test_files: all_test_file_paths
-          )
+        # Based on:
+        # https://github.com/rspec/rspec-core/blob/f8c8880dabd8f0544a6f91d8d4c857c1bd8df903/lib/rspec/core/runner.rb#L85
+        #
+        # @return [Fixnum] exit status code.
+        #   0 if all specs passed,
+        #   or the configured failure exit code (1 by default) if specs failed.
+        def run
+          pre_run_setup
+
+          if @rspec_runner.knapsack__wants_to_quit?
+            exit_code = @rspec_runner.knapsack__exit_early
+            Kernel.exit(exit_code)
+          end
+
+          begin
+            exit_code = @rspec_runner.knapsack__run_specs(self)
+          rescue KnapsackPro::Runners::Queue::BaseRunner::TerminationError
+            exit_code = @rspec_pure.error_exit_code(@rspec_runner.knapsack__error_exit_code)
+            Kernel.exit(exit_code)
+          rescue Exception => exception
+            KnapsackPro.logger.error("An unexpected exception happened. RSpec cannot handle it. The exception: #{exception.inspect}")
 
-          if test_file_paths.empty?
-            unless all_test_file_paths.empty?
-              KnapsackPro::Adapters::RSpecAdapter.verify_bind_method_called
+            message = @rspec_pure.exit_summary(unexecuted_test_files)
+            KnapsackPro.logger.warn(message) if message
 
-              KnapsackPro::Formatters::RSpecQueueSummaryFormatter.print_summary
-              KnapsackPro::Formatters::RSpecQueueProfileFormatterExtension.print_summary
+            exit_code = @rspec_pure.error_exit_code(@rspec_runner.knapsack__error_exit_code)
+            Kernel.exit(exit_code)
+          end
 
-              args += ['--seed', @@used_seed] if @@used_seed
+          post_run_tasks(exit_code)
+        end
 
-              log_rspec_command(args, all_test_file_paths, :end_of_queue)
-            end
+        def with_batch
+          can_initialize_queue = true
 
-            KnapsackPro::Hooks::Queue.call_after_queue
+          loop do
+            handle_signal!
+            test_file_paths = pull_tests_from_queue(can_initialize_queue: can_initialize_queue)
+            can_initialize_queue = false
 
-            KnapsackPro::Report.save_node_queue_to_api
+            break if test_file_paths.empty?
 
-            return {
-              status: :completed,
-              exitstatus: exitstatus,
-            }
-          else
             subset_queue_id = KnapsackPro::Config::EnvGenerator.set_subset_queue_id
             ENV['KNAPSACK_PRO_SUBSET_QUEUE_ID'] = subset_queue_id
 
-            KnapsackPro.tracker.reset!
-            KnapsackPro.tracker.set_prerun_tests(test_file_paths)
-
             KnapsackPro::Hooks::Queue.call_before_subset_queue
 
-            all_test_file_paths += test_file_paths
-            cli_args = args + test_file_paths
-
-            options = ::RSpec::Core::ConfigurationOptions.new(cli_args)
-            rspec_runner = ::RSpec::Core::Runner.new(options)
-
-            exit_code = rspec_runner.run($stderr, $stdout)
-            exitstatus = exit_code if exit_code != 0
-
-            printable_args = args_with_seed_option_added_when_viable(args, rspec_runner)
-            log_rspec_command(printable_args, test_file_paths, :subset_queue)
-
-            rspec_clear_examples
+            yield test_file_paths
 
             KnapsackPro::Hooks::Queue.call_after_subset_queue
 
-            KnapsackPro::Report.save_subset_queue_to_file
+            if @rspec_runner.knapsack__wants_to_quit?
+              KnapsackPro.logger.warn('RSpec wants to quit.')
+              set_terminate_process
+            end
+            if @rspec_runner.knapsack__rspec_is_quitting?
+              KnapsackPro.logger.warn('RSpec is quitting.')
+              set_terminate_process
+            end
 
-            return {
-              status: :next,
-              runner: runner,
-              can_initialize_queue: false,
-              args: args,
-              exitstatus: exitstatus,
-              all_test_file_paths: all_test_file_paths,
-            }
+            log_rspec_batch_command(test_file_paths)
           end
         end
 
+        def handle_signal!
+          self.class.handle_signal!
+        end
+
+        def log_fail_fast_limit_met
+          KnapsackPro.logger.warn('Test execution has been canceled because the RSpec --fail-fast option is enabled. It will cause other CI nodes to run tests longer because they need to consume more tests from the Knapsack Pro Queue API.')
+        end
+
         private
 
-        def self.adapter_class
-          KnapsackPro::Adapters::RSpecAdapter
+        def pre_run_setup
+          ENV['KNAPSACK_PRO_QUEUE_RECORDING_ENABLED'] = 'true'
+          ENV['KNAPSACK_PRO_QUEUE_ID'] = KnapsackPro::Config::EnvGenerator.set_queue_id
+
+          KnapsackPro::Config::Env.set_test_runner_adapter(@adapter_class)
+
+          ENV['SPEC_OPTS'] = @rspec_pure.add_knapsack_pro_formatters_to(ENV['SPEC_OPTS'])
+          @adapter_class.ensure_no_tag_option_when_rspec_split_by_test_examples_enabled!(@cli_args)
+
+          rspec_configuration_options = ::RSpec::Core::ConfigurationOptions.new(@cli_args)
+          @rspec_runner = ::RSpec::Core::Runner.new(rspec_configuration_options)
+          @rspec_runner.knapsack__setup(@stream_error, @stream_out)
+
+          ensure_no_deprecated_run_all_when_everything_filtered_option!
         end
 
-        def self.log_rspec_command(cli_args, test_file_paths, type)
-          case type
-          when :subset_queue
-            KnapsackPro.logger.info("To retry the last batch of tests fetched from the API Queue, please run the following command on your machine:")
-          when :end_of_queue
-            KnapsackPro.logger.info("To retry all the tests assigned to this CI node, please run the following command on your machine:")
-          end
+        def post_run_tasks(exit_code)
+          @adapter_class.verify_bind_method_called
 
-          stringified_cli_args = cli_args.join(' ').sub(" --format #{KnapsackPro::Formatters::RSpecQueueSummaryFormatter}", '')
+          log_rspec_queue_command
 
-          KnapsackPro.logger.info(
-            "bundle exec rspec #{stringified_cli_args} " +
-            KnapsackPro::TestFilePresenter.stringify_paths(test_file_paths)
-          )
+          time_tracker = KnapsackPro::Formatters::TimeTrackerFetcher.call
+          KnapsackPro::Report.save_node_queue_to_api(time_tracker&.queue(@node_test_file_paths))
+
+          Kernel.exit(exit_code)
         end
 
-        # Clear rspec examples without the shared examples:
-        # https://github.com/rspec/rspec-core/pull/2379
-        #
-        # Keep formatters and report to accumulate info about failed/pending tests
-        def self.rspec_clear_examples
-          if ::RSpec::ExampleGroups.respond_to?(:remove_all_constants)
-            ::RSpec::ExampleGroups.remove_all_constants
-          else
-            ::RSpec::ExampleGroups.constants.each do |constant|
-              ::RSpec::ExampleGroups.__send__(:remove_const, constant)
-            end
-          end
-          ::RSpec.world.example_groups.clear
-          ::RSpec.configuration.start_time = ::RSpec::Core::Time.now
-
-          if KnapsackPro::Config::Env.rspec_split_by_test_examples?
-            # Reset example group counts to ensure scoped example ids in metadata
-            # have correct index (not increased by each subsequent run).
-            # Solves this problem: https://github.com/rspec/rspec-core/issues/2721
-            ::RSpec.world.instance_variable_set(:@example_group_counts_by_spec_file, Hash.new(0))
-          end
+        def ensure_no_deprecated_run_all_when_everything_filtered_option!
+          return unless @rspec_runner.knapsack__deprecated_run_all_when_everything_filtered_enabled?
 
-          # skip reset filters for old RSpec versions
-          if ::RSpec.configuration.respond_to?(:reset_filters)
-            ::RSpec.configuration.reset_filters
-          end
+          error_message = "The run_all_when_everything_filtered option is deprecated. See: #{KnapsackPro::Urls::RSPEC__DEPRECATED_RUN_ALL_WHEN_EVERYTHING_FILTERED}"
+          KnapsackPro.logger.error(error_message)
+          raise error_message
         end
 
-        def self.args_with_seed_option_added_when_viable(args, rspec_runner)
-          order_option = adapter_class.order_option(args)
+        def pull_tests_from_queue(can_initialize_queue: false)
+          test_file_paths = test_file_paths(
+            can_initialize_queue: can_initialize_queue,
+            executed_test_files: @node_test_file_paths
+          )
+          @node_test_file_paths += test_file_paths
+          test_file_paths
+        end
 
-          if order_option
-            # Don't add the seed option for order other than random, e.g. `defined`
-            return args unless order_option.include?('rand')
-            # Don't add the seed option if the seed is already set in args, e.g. `rand:12345`
-            return args if order_option.to_s.split(':')[1]
-          end
+        def log_rspec_batch_command(test_file_paths)
+          order_option = @adapter_class.order_option(@cli_args)
+          printable_args = @rspec_pure.args_with_seed_option_added_when_viable(order_option, @rspec_runner.knapsack__seed, @cli_args)
+          messages = @rspec_pure.rspec_command(printable_args, test_file_paths, :batch_finished)
+          log_info_messages(messages)
+        end
 
-          # Don't add the seed option if the seed was not used (i.e. a different order is being used, e.g. `defined`)
-          return args unless rspec_runner.configuration.seed_used?
+        def log_rspec_queue_command
+          order_option = @adapter_class.order_option(@cli_args)
+          printable_args = @rspec_pure.args_with_seed_option_added_when_viable(order_option, @rspec_runner.knapsack__seed, @cli_args)
+          messages = @rspec_pure.rspec_command(printable_args, @node_test_file_paths, :queue_finished)
+          log_info_messages(messages)
+        end
 
-          @@used_seed = rspec_runner.configuration.seed.to_s
+        def log_info_messages(messages)
+          messages.each do |message|
+            KnapsackPro.logger.info(message)
+          end
+        end
 
-          args + ['--seed', @@used_seed]
+        def unexecuted_test_files
+          KnapsackPro::Formatters::TimeTrackerFetcher.unexecuted_test_files(@node_test_file_paths)
         end
       end
     end

data/lib/knapsack_pro/runners/rspec_runner.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   module Runners
     class RSpecRunner < BaseRunner
@@ -17,8 +19,6 @@ module KnapsackPro
           cli_args = (args || '').split
           adapter_class.ensure_no_tag_option_when_rspec_split_by_test_examples_enabled!(cli_args)
 
-          KnapsackPro.tracker.set_prerun_tests(runner.test_file_paths)
-
           require 'rspec/core/rake_task'
 
           task_name = 'knapsack_pro:rspec_run'
@@ -31,11 +31,30 @@ module KnapsackPro
             # because pattern does not accept test example path like spec/a_spec.rb[1:2]
             # instead we pass test files and test example paths to t.rspec_opts
             t.pattern = []
-            t.rspec_opts = "#{args} --default-path #{runner.test_dir} #{runner.stringify_test_file_paths}"
+            t.rspec_opts = "#{args} #{self.formatters} --default-path #{runner.test_dir} #{runner.stringify_test_file_paths}"
+            t.verbose = KnapsackPro::Config::Env.log_level < ::Logger::WARN
           end
           Rake::Task[task_name].invoke
         end
       end
+
+      # Use RSpec::Core::ConfigurationOptions to respect external configurations like .rspec
+      def self.formatters
+        require_relative '../formatters/time_tracker'
+
+        formatters = ::RSpec::Core::ConfigurationOptions
+          .new([])
+          .options
+          .fetch(:formatters, [])
+          .map do |formatter, output|
+            arg = "--format #{formatter}"
+            arg += " --out #{output}" if output
+            arg
+          end
+        formatters = ['--format progress'] if formatters.empty?
+        formatters += ["--format #{KnapsackPro::Formatters::TimeTracker}"]
+        formatters.join(' ')
+      end
     end
   end
 end

data/lib/knapsack_pro/runners/spinach_runner.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   module Runners
     class SpinachRunner < BaseRunner

data/lib/knapsack_pro/runners/test_unit_runner.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   module Runners
     class TestUnitRunner < BaseRunner

data/lib/knapsack_pro/slow_test_file_determiner.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class SlowTestFileDeterminer
     TIME_THRESHOLD_PER_CI_NODE = 0.7 # 70%

data/lib/knapsack_pro/slow_test_file_finder.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class SlowTestFileFinder
     # Get recorded test files from API.

data/lib/knapsack_pro/task_loader.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'rake'
 
 module KnapsackPro

data/lib/knapsack_pro/test_case_detectors/rspec_test_example_detector.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   module TestCaseDetectors
     class RSpecTestExampleDetector

data/lib/knapsack_pro/test_case_mergers/base_merger.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   module TestCaseMergers
     class BaseMerger

data/lib/knapsack_pro/test_case_mergers/rspec_merger.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   module TestCaseMergers
     class RSpecMerger < BaseMerger

data/lib/knapsack_pro/test_file_cleaner.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class TestFileCleaner
     def self.clean(test_file_path)

data/lib/knapsack_pro/test_file_finder.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class TestFileFinder
     def self.call(test_file_pattern, test_file_list_enabled: true)

data/lib/knapsack_pro/test_file_pattern.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class TestFilePattern
     def self.call(adapter_class)

data/lib/knapsack_pro/test_file_presenter.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class TestFilePresenter
     def self.stringify_paths(test_file_paths)

data/lib/knapsack_pro/test_files_with_test_cases_composer.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class TestFilesWithTestCasesComposer
     # Args:

data/lib/knapsack_pro/test_flat_distributor.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class TestFlatDistributor
     DIR_TYPES = [

data/lib/knapsack_pro/tracker.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class Tracker
     include Singleton
@@ -6,11 +8,10 @@ module KnapsackPro
     # to better allocate it in Queue Mode for future CI build runs
     DEFAULT_TEST_FILE_TIME = 0.0 # seconds
 
-    attr_reader :global_time_since_beginning, :global_time, :test_files_with_time, :prerun_tests_loaded
+    attr_reader :global_time, :test_files_with_time, :prerun_tests_loaded
     attr_writer :current_test_path
 
     def initialize
-      @global_time_since_beginning = 0
       KnapsackPro::Config::TempFiles.ensure_temp_directory_exists!
       FileUtils.mkdir_p(tracker_dir_path)
       set_defaults
@@ -133,7 +134,6 @@ module KnapsackPro
 
     def update_global_time(execution_time)
       @global_time += execution_time
-      @global_time_since_beginning += execution_time
     end
 
     def update_test_file_time(execution_time)

data/lib/knapsack_pro/urls.rb

@@ -18,6 +18,8 @@ module KnapsackPro
 
     INSTALLATION_GUIDE = "#{HOST}/perma/ruby/installation-guide"
 
+    KNAPSACK_PRO_CI_NODE_BUILD_ID= "#{HOST}/perma/ruby/knapsack-pro-ci-node-build-id"
+
     QUEUE_MODE__CONNECTION_ERROR_WITH_FALLBACK_ENABLED_FALSE = "#{HOST}/perma/ruby/queue-mode-connection-error-with-fallback-enabled-false"
 
     QUEUE_MODE__CONNECTION_ERROR_WITH_FALLBACK_ENABLED_TRUE_AND_POSITIVE_RETRY_COUNT = "#{HOST}/perma/ruby/queue-mode-connection-error-with-fallback-enabled-true-and-positive-retry-count"
@@ -26,6 +28,8 @@ module KnapsackPro
 
     REGULAR_MODE__CONNECTION_ERROR_WITH_FALLBACK_ENABLED_TRUE_AND_POSITIVE_RETRY_COUNT = "#{HOST}/perma/ruby/regular-mode-connection-error-with-fallback-enabled-true-and-positive-retry-count"
 
+    RSPEC__DEPRECATED_RUN_ALL_WHEN_EVERYTHING_FILTERED = "#{HOST}/perma/ruby/rspec-deprecated-run-all-when-everything-filtered"
+
     RSPEC__SKIPS_TESTS = "#{HOST}/perma/ruby/rspec-skips-tests"
 
     RSPEC__SPLIT_BY_TEST_EXAMPLES__TAG = "#{HOST}/perma/ruby/rspec-split-by-test-examples-tag"

data/lib/knapsack_pro/utils.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
   class Utils
     def self.unsymbolize(obj)

data/lib/knapsack_pro/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module KnapsackPro
-  VERSION = '3.8.0'
+  VERSION = '7.0.0'
 end

data/lib/knapsack_pro.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'logger'
 require 'singleton'
 require 'net/http'
@@ -6,9 +8,9 @@ require 'uri'
 require 'rake/testtask'
 require 'digest'
 require 'securerandom'
+require 'timeout'
 require_relative 'knapsack_pro/urls'
 require_relative 'knapsack_pro/version'
-require_relative 'knapsack_pro/extensions/time'
 require_relative 'knapsack_pro/hooks/queue'
 require_relative 'knapsack_pro/utils'
 require_relative 'knapsack_pro/config/ci/base'
@@ -21,8 +23,6 @@ require_relative 'knapsack_pro/config/ci/semaphore'
 require_relative 'knapsack_pro/config/ci/semaphore2'
 require_relative 'knapsack_pro/config/ci/buildkite'
 require_relative 'knapsack_pro/config/ci/travis'
-require_relative 'knapsack_pro/config/ci/snap_ci'
-require_relative 'knapsack_pro/config/ci/solano_ci'
 require_relative 'knapsack_pro/config/ci/codeship'
 require_relative 'knapsack_pro/config/ci/github_actions'
 require_relative 'knapsack_pro/config/ci/heroku'
@@ -57,6 +57,7 @@ require_relative 'knapsack_pro/adapters/test_unit_adapter'
 require_relative 'knapsack_pro/adapters/spinach_adapter'
 require_relative 'knapsack_pro/allocator'
 require_relative 'knapsack_pro/queue_allocator'
+require_relative 'knapsack_pro/mask_string'
 require_relative 'knapsack_pro/test_case_mergers/base_merger'
 require_relative 'knapsack_pro/test_case_mergers/rspec_merger'
 require_relative 'knapsack_pro/build_distribution_fetcher'
@@ -81,6 +82,7 @@ require_relative 'knapsack_pro/crypto/encryptor'
 require_relative 'knapsack_pro/crypto/branch_encryptor'
 require_relative 'knapsack_pro/crypto/decryptor'
 require_relative 'knapsack_pro/crypto/digestor'
+require_relative 'knapsack_pro/pure/queue/rspec_pure'
 
 require 'knapsack_pro/railtie' if defined?(Rails::Railtie)
 

data/lib/tasks/cucumber.rake

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'knapsack_pro'
 
 namespace :knapsack_pro do

data/lib/tasks/encrypted_branch_names.rake

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'knapsack_pro'
 
 namespace :knapsack_pro do

data/lib/tasks/encrypted_test_file_names.rake

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'knapsack_pro'
 
 namespace :knapsack_pro do

data/lib/tasks/minitest.rake

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require 'knapsack_pro'
 
 namespace :knapsack_pro do

data/lib/tasks/queue/cucumber.rake

@@ -1,8 +1,21 @@
+# frozen_string_literal: true
+
 require 'knapsack_pro'
 
 namespace :knapsack_pro do
   namespace :queue do
     task :cucumber, [:cucumber_args] do |_, args|
+      Kernel.system("RAILS_ENV=test RACK_ENV=test #{$PROGRAM_NAME} 'knapsack_pro:queue:cucumber_go[#{args[:cucumber_args]}]'")
+      exitstatus = $?.exitstatus
+      if exitstatus.nil?
+        puts 'Something went wrong. Most likely, the process has been killed. Knapsack Pro has been terminated.'
+        Kernel.exit(1)
+      else
+        Kernel.exit(exitstatus)
+      end
+    end
+
+    task :cucumber_go, [:cucumber_args] do |_, args|
       KnapsackPro::Runners::Queue::CucumberRunner.run(args[:cucumber_args])
     end
   end

data/lib/tasks/queue/minitest.rake

@@ -1,8 +1,21 @@
+# frozen_string_literal: true
+
 require 'knapsack_pro'
 
 namespace :knapsack_pro do
   namespace :queue do
     task :minitest, [:minitest_args] do |_, args|
+      Kernel.system("RAILS_ENV=test RACK_ENV=test #{$PROGRAM_NAME} 'knapsack_pro:queue:minitest_go[#{args[:minitest_args]}]'")
+      exitstatus = $?.exitstatus
+      if exitstatus.nil?
+        puts 'Something went wrong. Most likely, the process has been killed. Knapsack Pro has been terminated.'
+        Kernel.exit(1)
+      else
+        Kernel.exit(exitstatus)
+      end
+    end
+
+    task :minitest_go, [:minitest_args] do |_, args|
       KnapsackPro::Runners::Queue::MinitestRunner.run(args[:minitest_args])
     end
   end

data/lib/tasks/queue/rspec.rake

@@ -1,8 +1,21 @@
+# frozen_string_literal: true
+
 require 'knapsack_pro'
 
 namespace :knapsack_pro do
   namespace :queue do
     task :rspec, [:rspec_args] do |_, args|
+      Kernel.system("RAILS_ENV=test RACK_ENV=test #{$PROGRAM_NAME} 'knapsack_pro:queue:rspec_go[#{args[:rspec_args]}]'")
+      exitstatus = $?.exitstatus
+      if exitstatus.nil?
+        puts 'Something went wrong. Most likely, the process has been killed. Knapsack Pro has been terminated.'
+        Kernel.exit(1)
+      else
+        Kernel.exit(exitstatus)
+      end
+    end
+
+    task :rspec_go, [:rspec_args] do |_, args|
       KnapsackPro::Runners::Queue::RSpecRunner.run(args[:rspec_args])
     end
   end