Implement RubyVMSpecWalker#on_blocks_in_examples

Last hook!

lib/solargraph/rspec/ruby_vm_spec_walker.rb

@@ -263,12 +263,12 @@ def walk!
             handler.call(block_ast)
           end
 
-          # # @param blocks_in_examples [Parser::AST::Node]
-          # each_block(block_ast.children[2]) do |blocks_in_examples|
-          #   @handlers[:on_blocks_in_examples].each do |handler|
-          #     handler.call(blocks_in_examples)
-          #   end
-          # end
+          # @param blocks_in_examples [RubyVM::AbstractSyntaxTree::Node]
+          each_block(block_ast.children[1]) do |blocks_in_examples|
+            @handlers[:on_blocks_in_examples].each do |handler|
+              handler.call(blocks_in_examples)
+            end
+          end
         end
 
         walker.on :ITER do |block_ast|
@@ -278,12 +278,12 @@ def walk!
             handler.call(block_ast)
           end
 
-          # # @param blocks_in_examples [Parser::AST::Node]
-          # each_block(block_ast.children[2]) do |blocks_in_examples|
-          #   @handlers[:on_blocks_in_examples].each do |handler|
-          #     handler.call(blocks_in_examples)
-          #   end
-          # end
+          # @param blocks_in_examples [RubyVM::AbstractSyntaxTree::Node]
+          each_block(block_ast.children[1]) do |blocks_in_examples|
+            @handlers[:on_blocks_in_examples].each do |handler|
+              handler.call(blocks_in_examples)
+            end
+          end
         end
 
         walker.walk

spec/solargraph/rspec/ruby_vm_spec_walker_spec.rb

@@ -271,31 +271,35 @@ def walk_code(code)
     end
   end
 
-  # describe '#on_blocks_in_examples' do
-  #   it 'yields each context block' do
-  #     code = <<~RUBY
-  #       RSpec.describe SomeClass, type: :model do
-  #         it 'does something' do
-  #           expect { subject }.to change { SomeClass.count }.by(1)
-  #         end
-
-  #         context 'when something' do
-  #           it 'does something' do
-  #             do_something { subject }
-  #           end
-  #         end
-  #       end
-  #     RUBY
+  describe '#on_blocks_in_examples' do
+    it 'yields each context block' do
+      code = <<~RUBY
+        RSpec.describe SomeClass, type: :model do
+          it 'does something' do
+            expect { subject }.to change { SomeClass.count }.by(1)
+          end
 
-  #     called = 0
-      # @param walker [Solargraph::Rspec::RubyVMSpecWalker]
-  #     walk_code(code) do |walker|
-  #       walker.on_blocks_in_examples do |_|
-  #         called += 1
-  #       end
-  #     end
+          context 'when something' do
+            around do
+              do_something { subject }
+            end
 
-  #     expect(called).to eq(3)
-  #   end
-  # end
+            it 'does something' do
+              do_something_too { subject }
+            end
+          end
+        end
+      RUBY
+
+      called = 0
+      # @param walker [Solargraph::Rspec::SpecWalker]
+      walk_code(code) do |walker|
+        walker.on_blocks_in_examples do |_|
+          called += 1
+        end
+      end
+
+      expect(called).to eq(4)
+    end
+  end
 end

spec/solargraph/rspec/spec_walker_spec.rb

@@ -173,9 +173,13 @@ def walk_code(code)
           end
 
           context 'when something' do
-            it 'does something' do
+            around do
               do_something { subject }
             end
+
+            it 'does something' do
+              do_something_too { subject }
+            end
           end
         end
       RUBY
@@ -188,7 +192,7 @@ def walk_code(code)
         end
       end
 
-      expect(called).to eq(3)
+      expect(called).to eq(4)
     end
   end
 end