Fix formatting WITH as a single line

Author: lu-zero

src/formatter.rs

@@ -244,6 +244,8 @@ impl<'a> Formatter<'a> {
         query.push_str(&self.equalize_whitespace(&self.format_reserved_word(token.value)));
         if newline_after {
             self.add_new_line(query);
+        } else {
+            query.push(' ');
         }
     }
 

src/lib.rs

@@ -2053,6 +2053,26 @@ mod tests {
         assert_eq!(format(input, &QueryParams::None, &options), expected);
     }
 
+    #[test]
+    fn format_short_with() {
+        let input = "WITH a AS ( SELECT a, b, c FROM t WHERE a > 100 ) SELECT b, field FROM a, aa;";
+        let max_line = 80;
+        let options = FormatOptions {
+            max_inline_block: max_line,
+            max_inline_arguments: Some(max_line),
+            max_inline_top_level: Some(max_line),
+            joins_as_top_level: true,
+            ..Default::default()
+        };
+        let expected = indoc! {
+            "
+            WITH a AS (SELECT a, b, c FROM t WHERE a > 100)
+            SELECT b, field
+            FROM a, aa;"
+        };
+        assert_eq!(format(input, &QueryParams::None, &options), expected);
+    }
+
     #[test]
     fn format_nested_select_nested_blocks() {
         let input =