add missing tests

Author: giacomocavalieri

compiler-core/src/format.rs

@@ -738,7 +738,9 @@ impl<'comments> Formatter<'comments> {
             [first_line, lines @ ..] => {
                 let mut doc = docvec!("\"", first_line);
                 for line in lines {
-                    doc = doc.append(pretty::line().set_nest(0)).append(line.to_doc())
+                    doc = doc
+                        .append(pretty::line().set_nesting(0))
+                        .append(line.to_doc())
                 }
                 doc.append("\"".to_doc()).group()
             }

compiler-core/src/format/tests.rs

@@ -5390,6 +5390,25 @@ fn multiline_string_are_not_broken_with_string_concatenation_if_they_fit() {
     );
 }
 
+#[test]
+fn nesting_goes_back_to_normal_after_multiline_string() {
+    assert_format!(
+        r#"pub fn main() {
+  let x = {
+    "
+1
+2
+" <> long_name_function_call(
+      1_111_111_111_111_111,
+      222_222_222_222,
+      3_333_333_333_333_333,
+    )
+  }
+}
+"#
+    );
+}
+
 #[test]
 fn multiline_string_get_broken_on_newlines_as_function_arguments() {
     assert_format!(

compiler-core/src/pretty.rs

@@ -584,7 +584,7 @@ impl<'a> Document<'a> {
         Self::Group(Box::new(self))
     }
 
-    pub fn set_nest(self, indent: isize) -> Self {
+    pub fn set_nesting(self, indent: isize) -> Self {
         Self::Nest(indent, NestMode::Set, NestCondition::Always, Box::new(self))
     }
 

compiler-core/src/pretty/tests.rs

@@ -309,3 +309,9 @@ fn empty_documents() {
     assert!(!"foo".to_doc().append("".to_doc()).is_empty());
     assert!(!"".to_doc().append("foo".to_doc()).is_empty());
 }
+
+#[test]
+fn set_nesting() {
+    let doc = Vec(vec!["foo".to_doc(), break_("", " "), "bar".to_doc()]).group();
+    assert_eq!("foo\nbar", doc.set_nesting(0).nest(2).to_pretty_string(1));
+}