Add EnsureNode#rescue_node.

Author: dvandersluis

lib/rubocop/ast/node/ensure_node.rb

@@ -13,6 +13,13 @@ def body
         node_parts[1]
       end
 
+      # Returns the `rescue` node of the `ensure`, if present.
+      #
+      # @return [Node, nil] The `rescue` node.
+      def rescue_node
+        node_parts[0] if node_parts[0].rescue_type?
+      end
+
       # Checks whether this node body is a void context.
       # Always `true` for `ensure`.
       #

spec/rubocop/ast/ensure_node_spec.rb

@@ -15,6 +15,40 @@
     it { expect(ensure_node.body).to be_sym_type }
   end
 
+  describe '#rescue_node' do
+    subject(:rescue_node) { ensure_node.rescue_node }
+
+    context 'when there is no `rescue` node' do
+      let(:source) do
+        <<~RUBY
+          begin
+            beginbody
+          ensure
+            ensurebody
+          end
+        RUBY
+      end
+
+      it { is_expected.to be_nil }
+    end
+
+    context 'when there is a `rescue` node' do
+      let(:source) do
+        <<~RUBY
+          begin
+            beginbody
+          rescue
+            rescuebody
+          ensure
+            ensurebody
+          end
+        RUBY
+      end
+
+      it { is_expected.to be_a(RuboCop::AST::RescueNode) }
+    end
+  end
+
   describe '#void_context?' do
     let(:source) { 'begin; beginbody; ensure; ensurebody; end' }
 

spec/rubocop/ast/rescue_node_spec.rb

@@ -24,6 +24,19 @@
     RUBY
 
     it { expect(rescue_node.body).to be_send_type }
+
+    context 'with multiple lines in body' do
+      let(:source) { <<~RUBY }
+        begin
+          foo
+          bar
+        rescue => e
+          baz
+        end
+      RUBY
+
+      it { expect(rescue_node.body).to be_begin_type }
+    end
   end
 
   describe '#resbody_branches' do