Allow flip-flop operators to be formatted correctly within IfMod and UnlessMod nodes.

Author: kddnewton

CHANGELOG.md

@@ -42,6 +42,7 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/) a
 - Make `Command` and `CommandCall` aware that their arguments could exceed their normal expected bounds because of heredocs.
 - Only unescape forward slashes in regular expressions if converting from slash bounds to `%r` bounds.
 - Allow `When` nodes to grab trailing comments away from their statements lists.
+- Allow flip-flop operators to be formatted correctly within `IfMod` and `UnlessMod` nodes.
 
 ### Removed
 

lib/syntax_tree.rb

@@ -4708,8 +4708,7 @@ def initialize(operator, node)
     end
 
     def format(q)
-      parent = q.parent
-      space = parent.is_a?(If) || parent.is_a?(Unless)
+      space = [If, IfMod, Unless, UnlessMod].include?(q.parent.class)
 
       left = node.left
       right = node.right