Correctly format multiple function calls in expression

Closes #69.
zeek · Jan 16, 2024 · ef43ea6 · ef43ea6
1 parent f3fd339
commit ef43ea6
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/tests/test_formatting.py b/tests/test_formatting.py
@@ -97,6 +97,16 @@ def test_format_comment_separator(self):
             self._format(code.encode()).decode().splitlines(), expected.splitlines()
         )
 
+    def test_multiple_function_calls(self):
+        """Validates correct line break when a expression with function calls
+        need to be broken. This is a regression test for #69."""
+        code = "f(x) " + "+ f(x)" * 20 + ";"
+        formatted = self._format(code.encode()).decode()
+        self.assertTrue(
+            formatted.splitlines()[0].endswith("("),
+            "expression should not break function calls across lines",
+        )
+
 
 class TestFormattingErrors(unittest.TestCase):
     def _format(self, content):

diff --git a/tree-sitter-zeek b/tree-sitter-zeek