Special handling for RSpec matchers when nesting CommandCall nodes.

Author: kddnewton

CHANGELOG.md

@@ -9,6 +9,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/) a
 ### Added
 
 - [#7](https://github.com/kddnewton/syntax_tree/issues/7) Better formatting for hashes and arrays that are values in hashes.
+- [#9](https://github.com/kddnewton/syntax_tree/issues/9) Special handling for RSpec matchers when nesting `CommandCall` nodes.
 
 ## [1.1.0] - 2021-12-08
 

lib/syntax_tree.rb

@@ -3771,14 +3771,8 @@ def format(q)
           end
 
         if arguments
-          width = doc_width(doc) + 1
           q.text(" ")
-
-          if width > (q.maxwidth / 2)
-            q.format(arguments)
-          else
-            q.nest(width) { q.format(arguments) }
-          end
+          q.nest(argument_alignment(q, doc)) { q.format(arguments) }
         end
       end
     end
@@ -3843,6 +3837,28 @@ def doc_width(parent)
 
       width
     end
+
+    def argument_alignment(q, doc)
+      # Very special handling case for rspec matchers. In general with rspec
+      # matchers you expect to see something like:
+      #
+      #     expect(foo).to receive(:bar).with(
+      #       'one',
+      #       'two',
+      #       'three',
+      #       'four',
+      #       'five'
+      #     )
+      #
+      # In this case the arguments are aligned to the left side as opposed to
+      # being aligned with the `receive` call.
+      if ["to", "not_to", "to_not"].include?(message.value)
+        0
+      else
+        width = doc_width(doc) + 1
+        width > (q.maxwidth / 2) ? 0 : width
+      end
+    end
   end
 
   # :call-seq:

test/fixtures/command_call.rb

@@ -5,3 +5,21 @@
 -
 foo.bar barrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr,
         bazzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzzz
+%
+expect(foo).to receive(fooooooooooooooooooooooooooooooooooooooooooooooooooooooooo)
+-
+expect(foo).to receive(
+  fooooooooooooooooooooooooooooooooooooooooooooooooooooooooo
+)
+%
+expect(foo).not_to receive(fooooooooooooooooooooooooooooooooooooooooooooooooooooooooo)
+-
+expect(foo).not_to receive(
+  fooooooooooooooooooooooooooooooooooooooooooooooooooooooooo
+)
+%
+expect(foo).to_not receive(fooooooooooooooooooooooooooooooooooooooooooooooooooooooooo)
+-
+expect(foo).to_not receive(
+  fooooooooooooooooooooooooooooooooooooooooooooooooooooooooo
+)