Incorrect auto-insertion of } on Enter #216 : more tests for comple…

…x expressions
xonixx · Jul 5, 2024 · b8c6caa · b8c6caa
1 parent deb61e3
commit b8c6caa
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/test/java/intellij_awk/AwkBracesAndQuoteAutoCloseTests.java b/src/test/java/intellij_awk/AwkBracesAndQuoteAutoCloseTests.java
@@ -112,6 +112,22 @@ public void testEnterCurlyBrace7() {
   public void testEnterCurlyBrace8() {
     doTest('\n', "function f()\n{<caret>print 123", "function f()\n{\n    <caret>print 123\n}");
   }
+  public void testEnterCurlyBrace9() {
+    doTestEnterCurlyBraceComplex("if (2)");
+  }
+  public void testEnterCurlyBrace10() {
+    doTestEnterCurlyBraceComplex("while (2)");
+  }
+  public void testEnterCurlyBrace11() {
+    doTestEnterCurlyBraceComplex("for(;;)");
+  }
+
+  private void doTestEnterCurlyBraceComplex(String opener) {
+    doTest(
+        '\n',
+        "function f() {\n    if (1) {<caret>" + opener + " {\n        print 123}\n}",
+        "function f() {\n    if (1) {\n        "+opener+" {\n            print 123}\n    }\n}");
+  }
 
   private void doTest(char brace, String code, String expectedCode) {
     myFixture.configureByText("a.awk", code);