lib/rubocop/rspec/example_group.rb in rubocop-rspec-1.9.1 vs lib/rubocop/rspec/example_group.rb in rubocop-rspec-1.10.0

@@ -1,39 +1,55 @@
 # frozen_string_literal: true
 
 module RuboCop
   module RSpec
     # Wrapper for RSpec example groups
-    class ExampleGroup
-      include Language
-      extend NodePattern::Macros
-
-      def_node_matcher :example?, Examples::ALL.block_pattern
-
+    class ExampleGroup < Concept
       # @!method scope_change?(node)
       #
       #   Detect if the node is an example group or shared example
       #
       #   Selectors which indicate that we should stop searching
       #
       def_node_matcher :scope_change?,
                        (ExampleGroups::ALL + SharedGroups::ALL).block_pattern
 
-      def initialize(node)
-        @node = node
-      end
+      # @!method hook(node)
+      #
+      #   Detect if node is `before`, `after`, `around`
+      def_node_matcher :hook, <<-PATTERN
+      (block {$(send nil #{Hooks::ALL.node_pattern_union} ...)} ...)
+      PATTERN
 
       def examples
         examples_in_scope(node).map(&Example.public_method(:new))
       end
 
+      def hooks
+        hooks_in_scope(node).map(&Hook.public_method(:new))
+      end
+
       private
 
-      attr_reader :node
+      def hooks_in_scope(node)
+        node.each_child_node.flat_map do |child|
+          find_hooks(child)
+        end
+      end
 
-      def examples_in_scope(node)
+      def find_hooks(node)
+        return [] if scope_change?(node) || example?(node)
+
+        if hook(node)
+          [node]
+        else
+          hooks_in_scope(node)
+        end
+      end
+
+      def examples_in_scope(node, &blk)
         node.each_child_node.flat_map do |child|
-          find_examples(child)
+          find_examples(child, &blk)
         end
       end
 
       # Recursively search for examples within the current scope
       #