knapsack_pro 0.44.0 → 0.45.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: c8d9401540ef36e33e333a330b03807d83a902de
-  data.tar.gz: 86fe1eeb6c82a2a37ca8c81b0cc7eb465d63ba79
+  metadata.gz: 72e3898daf42db4b3cb0c746ad66f00ecda1048d
+  data.tar.gz: 5ed151190611d41bb925b93d9648385e5c55cbce
 SHA512:
-  metadata.gz: 04e2e06185f9d02bb16c69cdd89e352b5a4e2bdf8c2509ae0f4c22be2381c0e1f4bebcfc29c950df6c6c9ac34fb4bdbbd917da85b90e34ae4cce562b7ea1144f
-  data.tar.gz: bfd28948e2c3cd157b2ee9f51ca40d822fa5e97af13131c42ee67b29ded3f5d8008a42a91dba0b9bc0b7c80287c221d6973ba55d3b506c9b1e721e0aee71f01f
+  metadata.gz: 5eb27e6eb4038c8dd31009a6bee4d46379b68669a79cb6a2e61e7023ed61d390c2cbb13855e221a48f79a1d3c9388c91bc080515e9367a8bc4ef5da648b88a1e
+  data.tar.gz: 99c81386ae4c190c2c0d2bc07185b528f5c8be7c9937d419e2ab33f719c16688a1df321303750267723880fd745ab8d6f7c4bf29a15b43faff746b1602de9a8b

data/CHANGELOG.md

@@ -2,6 +2,14 @@
 
 * TODO
 
+### 0.45.0
+
+* Add before and after queue hooks
+
+    https://github.com/KnapsackPro/knapsack_pro-ruby/pull/46
+
+https://github.com/KnapsackPro/knapsack_pro-ruby/compare/v0.44.0...v0.45.0
+
 ### 0.44.0
 
 * Add ability to set test_dir using an environment variable.

data/README.md

@@ -1188,11 +1188,15 @@ Knapsack Pro Queue Mode runs subset of test files from the work queue many times
 ```ruby
 # spec_helper.rb or rails_helper.rb
 
-# executes before Queue Mode starts work
-Percy::Capybara.initialize_build
+KnapsackPro::Hooks::Queue.before_queue do |queue_id|
+  # executes before Queue Mode starts work
+  Percy::Capybara.initialize_build
+end
 
-# executes after Queue Mode finishes work
-at_exit { Percy::Capybara.finalize_build }
+KnapsackPro::Hooks::Queue.after_queue do |queue_id|
+  # executes after Queue Mode finishes work
+  Percy::Capybara.finalize_build
+end
 ```
 
 #### How to call `before(:suite)` and `after(:suite)` RSpec hooks only once in Queue Mode?
@@ -1202,18 +1206,17 @@ Knapsack Pro Queue Mode runs subset of test files from the work queue many times
 ```ruby
 # spec_helper.rb or rails_helper.rb
 
-RSpec.configure do |config|
-  config.before(:suite) do
-    unless ENV['KNAPSACK_PRO_RSPEC_BEFORE_SUITE_LOADED']
-      ENV['KNAPSACK_PRO_RSPEC_BEFORE_SUITE_LOADED'] = 'true'
-
-      # this will be called only once before the tests started on the CI node
-    end
-  end
+KnapsackPro::Hooks::Queue.before_queue do |queue_id|
+  # This will be called only once before the tests started on the CI node.
+  # It will be run inside of the RSpec before(:suite) block only once.
+  # It means you will have access to whatever RSpec provides in the context of the before(:suite) block.
+end
 
-  at_exit do
-    # this will be called only once at the end when the CI node finished tests
-  end
+KnapsackPro::Hooks::Queue.after_queue do |queue_id|
+  # This will be called only once after test suite is completed.
+  # Note this hook won't be called inside of RSpec after(:suite) block because
+  # we are not able to determine which after(:suite) block will be called as the last one
+  # due to the fact the Knapsack Pro Queue Mode allocates tests in dynamic way.
 end
 ```
 

data/lib/knapsack_pro/adapters/base_adapter.rb

@@ -19,6 +19,7 @@ module KnapsackPro
 
         if KnapsackPro::Config::Env.queue_recording_enabled?
           KnapsackPro.logger.debug('Test suite time execution queue recording enabled.')
+          bind_before_queue_hook
           bind_time_tracker
           bind_save_queue_report
         end
@@ -35,6 +36,10 @@ module KnapsackPro
       def bind_save_queue_report
         raise NotImplementedError
       end
+
+      def bind_before_queue_hook
+        raise NotImplementedError
+      end
     end
   end
 end

data/lib/knapsack_pro/adapters/rspec_adapter.rb

@@ -57,6 +57,17 @@ module KnapsackPro
           end
         end
       end
+
+      def bind_before_queue_hook
+        ::RSpec.configure do |config|
+          config.before(:suite) do
+            unless ENV['KNAPSACK_PRO_BEFORE_QUEUE_HOOK_CALLED']
+              KnapsackPro::Hooks::Queue.call_before_queue
+              ENV['KNAPSACK_PRO_BEFORE_QUEUE_HOOK_CALLED'] = 'true'
+            end
+          end
+        end
+      end
     end
 
     # This is added to provide backwards compatibility

data/lib/knapsack_pro/hooks/queue.rb

@@ -2,16 +2,41 @@ module KnapsackPro
   module Hooks
     class Queue
       class << self
-        attr_reader :after_subset_queue
+        attr_reader :before_queue,
+          :after_subset_queue,
+          :after_queue
+
+        def reset_before_queue
+          @before_queue = nil
+        end
 
         def reset_after_subset_queue
           @after_subset_queue = nil
         end
 
+        def reset_after_queue
+          @after_queue = nil
+        end
+
+        def before_queue(&block)
+          @before_queue ||= block
+        end
+
         def after_subset_queue(&block)
           @after_subset_queue ||= block
         end
 
+        def after_queue(&block)
+          @after_queue ||= block
+        end
+
+        def call_before_queue
+          return unless before_queue
+          before_queue.call(
+            KnapsackPro::Config::Env.queue_id
+          )
+        end
+
         def call_after_subset_queue
           return unless after_subset_queue
           after_subset_queue.call(
@@ -19,6 +44,13 @@ module KnapsackPro
             KnapsackPro::Config::Env.subset_queue_id
           )
         end
+
+        def call_after_queue
+          return unless after_queue
+          after_queue.call(
+            KnapsackPro::Config::Env.queue_id
+          )
+        end
       end
     end
   end

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

@@ -35,6 +35,8 @@ module KnapsackPro
               log_rspec_command(args, all_test_file_paths, :end_of_queue)
             end
 
+            KnapsackPro::Hooks::Queue.call_after_queue
+
             KnapsackPro::Report.save_node_queue_to_api
             exit(exitstatus)
           else

data/lib/knapsack_pro/version.rb

@@ -1,3 +1,3 @@
 module KnapsackPro
-  VERSION = '0.44.0'
+  VERSION = '0.45.0'
 end

data/spec/knapsack_pro/adapters/base_adapter_spec.rb

@@ -48,6 +48,7 @@ describe KnapsackPro::Adapters::BaseAdapter do
       let(:queue_recording_enabled?) { true }
 
       before do
+        allow(subject).to receive(:bind_before_queue_hook)
         allow(subject).to receive(:bind_time_tracker)
         allow(subject).to receive(:bind_save_queue_report)
       end
@@ -57,6 +58,7 @@ describe KnapsackPro::Adapters::BaseAdapter do
         expect(KnapsackPro).to receive(:logger).and_return(logger)
         expect(logger).to receive(:debug).with('Test suite time execution queue recording enabled.')
       end
+      it { expect(subject).to receive(:bind_before_queue_hook) }
       it { expect(subject).to receive(:bind_time_tracker) }
       it { expect(subject).to receive(:bind_save_queue_report) }
     end
@@ -65,6 +67,7 @@ describe KnapsackPro::Adapters::BaseAdapter do
       it { expect(subject).not_to receive(:bind_time_tracker) }
       it { expect(subject).not_to receive(:bind_save_report) }
       it { expect(subject).not_to receive(:bind_save_queue_report) }
+      it { expect(subject).not_to receive(:bind_before_queue_hook) }
     end
   end
 
@@ -91,4 +94,12 @@ describe KnapsackPro::Adapters::BaseAdapter do
       }.to raise_error(NotImplementedError)
     end
   end
+
+  describe '#bind_before_queue_hook' do
+    it do
+      expect {
+        subject.bind_before_queue_hook
+      }.to raise_error(NotImplementedError)
+    end
+  end
 end

data/spec/knapsack_pro/adapters/rspec_adapter_spec.rb

@@ -123,5 +123,16 @@ describe KnapsackPro::Adapters::RSpecAdapter do
         subject.bind_save_queue_report
       end
     end
+
+    describe '#bind_before_queue_hook' do
+      it do
+        expect(config).to receive(:before).with(:suite).and_yield
+        expect(::RSpec).to receive(:configure).and_yield(config)
+
+        expect(KnapsackPro::Hooks::Queue).to receive(:call_before_queue)
+
+        subject.bind_before_queue_hook
+      end
+    end
   end
 end

data/spec/knapsack_pro/hooks/queue_spec.rb

@@ -1,4 +1,30 @@
 describe KnapsackPro::Hooks::Queue do
+  describe '.call_before_queue' do
+    subject { described_class.call_before_queue }
+
+    context 'when callback is not set' do
+      before do
+        described_class.reset_before_queue
+      end
+
+      it { should be_nil }
+    end
+
+    context 'when callback is set' do
+      let(:queue_id) { double }
+
+      before do
+        expect(KnapsackPro::Config::Env).to receive(:queue_id).and_return(queue_id)
+
+        described_class.before_queue do |q_id|
+          [:fake_value, q_id]
+        end
+      end
+
+      it { should eq [:fake_value, queue_id] }
+    end
+  end
+
   describe '.call_after_subset_queue' do
     subject { described_class.call_after_subset_queue }
 
@@ -26,4 +52,30 @@ describe KnapsackPro::Hooks::Queue do
       it { should eq [:fake_value, queue_id, subset_queue_id] }
     end
   end
+
+  describe '.call_after_queue' do
+    subject { described_class.call_after_queue }
+
+    context 'when callback is not set' do
+      before do
+        described_class.reset_after_queue
+      end
+
+      it { should be_nil }
+    end
+
+    context 'when callback is set' do
+      let(:queue_id) { double }
+
+      before do
+        expect(KnapsackPro::Config::Env).to receive(:queue_id).and_return(queue_id)
+
+        described_class.after_queue do |q_id|
+          [:fake_value, q_id]
+        end
+      end
+
+      it { should eq [:fake_value, queue_id] }
+    end
+  end
 end

data/spec/knapsack_pro/runners/queue/rspec_runner_spec.rb

@@ -111,6 +111,7 @@ describe KnapsackPro::Runners::Queue::RSpecRunner do
         it do
           expect(KnapsackPro::Formatters::RSpecQueueSummaryFormatter).to receive(:print_summary)
           expect(KnapsackPro::Formatters::RSpecQueueProfileFormatterExtension).to receive(:print_summary)
+          expect(KnapsackPro::Hooks::Queue).to receive(:call_after_queue)
           expect(KnapsackPro::Report).to receive(:save_node_queue_to_api)
           expect(described_class).to receive(:exit).with(exitstatus)
 
@@ -124,6 +125,7 @@ describe KnapsackPro::Runners::Queue::RSpecRunner do
         it do
           expect(KnapsackPro::Formatters::RSpecQueueSummaryFormatter).to receive(:print_summary)
           expect(KnapsackPro::Formatters::RSpecQueueProfileFormatterExtension).to receive(:print_summary)
+          expect(KnapsackPro::Hooks::Queue).to receive(:call_after_queue)
           expect(KnapsackPro::Report).to receive(:save_node_queue_to_api)
           expect(described_class).to receive(:exit).with(exit_code)
 
@@ -136,6 +138,7 @@ describe KnapsackPro::Runners::Queue::RSpecRunner do
       let(:test_file_paths) { [] }
 
       it do
+        expect(KnapsackPro::Hooks::Queue).to receive(:call_after_queue)
         expect(KnapsackPro::Report).to receive(:save_node_queue_to_api)
         expect(described_class).to receive(:exit).with(exitstatus)
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: knapsack_pro
 version: !ruby/object:Gem::Version
-  version: 0.44.0
+  version: 0.45.0
 platform: ruby
 authors:
 - ArturT
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2017-07-31 00:00:00.000000000 Z
+date: 2017-08-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake