Add support for config.around(:all) and config.around(:all_nested)

README.md

@@ -7,7 +7,7 @@ Provides around(:all) hooks for RSpec.
 In your Gemfile:
 
 ``` ruby
-gem 'rspec_around_all', git: 'git://gist.github.com/2005175.git'
+gem 'rspec_around_all', require: false
 ```
 
 In a spec:

lib/rspec_around_all.rb

@@ -10,25 +10,48 @@ def run_examples
     def to_proc
       proc { run_examples }
     end
+
+    def class
+      __getobj__.class
+    end
   end
 
   def around(scope = :each, &block)
+    return around_all &block if scope == :all
     # let RSpec handle around(:each) hooks...
-    return super(scope, &block) unless scope == :all
+    return super(scope, &block) unless scope == :all_nested
+    around_all do |group|
+      group.children.each {|c| c.around(:all_nested, &block) }
+      block[group]
+    end
+  end
 
-    group, fiber = self, nil
-    before(:all) do
+  private
+
+  FIBERS_STACK = []
+
+  def around_all(&block)
+    prepend_before(:all) do |group|
       fiber = Fiber.new(&block)
-      fiber.resume(FiberAwareGroup.new(group))
+      FIBERS_STACK << fiber
+      fiber.resume(FiberAwareGroup.new(group.class))
     end
 
-    after(:all) do
+    prepend_after(:all) do
+      fiber = FIBERS_STACK.pop
       fiber.resume
     end
   end
 end
 
 RSpec.configure do |c|
   c.extend RSpecAroundAll
+
+  # Add config.around(:all):
+  # c.extend overrides the original Object#extend method.
+  # See discussion in https://github.com/rspec/rspec-core/issues/1031#issuecomment-22264638
+  Object.instance_method(:extend).bind(c).call RSpecAroundAll
+  # Ruby 2 alternative:
+  # RSpec::Core::Configuration.send :prepend, RSpecAroundAll
 end
 

spec/rspec_around_all_spec.rb

@@ -1,5 +1,17 @@
 require_relative '../lib/rspec_around_all'
 
+RSpec.configure do |c|
+  [:all, :all_nested].each do |all_type|
+    c.around(all_type) do |group|
+      (group.run_examples; next) unless order = group.metadata[:order]
+      count = group.metadata[:count][all_type] += 1
+      order << "config.before(:#{all_type}) #{count}"
+      group.run_examples
+      order << "config.after(:#{all_type}) #{count}"
+    end
+  end
+end
+
 module RSpec
   module Core
     describe "around(:all) hook" do
@@ -89,6 +101,44 @@ def self.transactionally
         order.should eq([:before, :e1, :after, :before, :e2, :after])
       end
     end
+
+    describe "config.around(:all) hook", order: [], count: {all_nested: 0, all: 0} do
+      order = nil
+      context "part 1" do
+        around(:all) do |g|
+          order = g.metadata[:order]
+          order << 'inner.before(:all)'
+          g.run_examples
+          order << 'inner.after(:all)'
+        end
+
+        specify { order << 'first' }
+        specify { order << 'second' }
+
+        example "blocks are executed in the right order" do
+          expect(order).to eq [
+            'config.before(:all_nested) 1',
+            'config.before(:all) 1',
+            'config.before(:all_nested) 2',
+            'inner.before(:all)',
+            'first',
+            'second',
+          ]
+        end
+      end
+
+      # this context won't pass if run alone (rspec -e "part 2") since it depends on part 1 to run before it
+      context "part 2" do
+        example "before and after hooks order is correct" do
+          expect(order[-3..-1]).to eq [
+            'inner.after(:all)', # part 1
+            'config.after(:all_nested) 2', # inner config.around
+            'config.before(:all_nested) 3', # config.before part 2
+          ]
+        end
+      end
+
+    end
   end
 end