added for* statements to isCompletedNode

vladima · vladima · commit 4b955ee91a66 · 2015-03-11T14:03:32.000-07:00
diff --git a/src/services/formatting/smartIndenter.ts b/src/services/formatting/smartIndenter.ts
@@ -461,6 +461,12 @@ module ts.formatting {
                 case SyntaxKind.DefaultClause:
                     // there is no such thing as terminator token for CaseClause\DefaultClause so for simplicitly always consider them non-completed
                     return false;
+                case SyntaxKind.ForStatement:
+                    return isCompletedNode((<ForStatement>n).statement, sourceFile);
+                case SyntaxKind.ForInStatement:
+                    return isCompletedNode((<ForInStatement>n).statement, sourceFile);
+                case SyntaxKind.ForOfStatement:
+                    return isCompletedNode((<ForOfStatement>n).statement, sourceFile);
                 case SyntaxKind.WhileStatement:
                     return isCompletedNode((<WhileStatement>n).statement, sourceFile);
                 case SyntaxKind.DoStatement:
diff --git a/tests/cases/fourslash/smartIndentStatementFor.ts b/tests/cases/fourslash/smartIndentStatementFor.ts
@@ -5,10 +5,15 @@
 ////        /*insideStatement*/
 ////    }
 ////    /*afterStatement*/
+////    for (var i = 0;;) 
+////        /*insideStatement2*/
 ////}
 
 goTo.marker('insideStatement');
 verify.indentationIs(8);
 
 goTo.marker('afterStatement');
 verify.indentationIs(4);
+
+goTo.marker('insideStatement2');
+verify.indentationIs(8);
diff --git a/tests/cases/fourslash/smartIndentStatementForIn.ts b/tests/cases/fourslash/smartIndentStatementForIn.ts
@@ -6,10 +6,15 @@
 ////        /*insideStatement*/
 ////    }
 ////    /*afterStatement*/
+////    for (var i in [])
+////        /*insideStatement2*/
 ////}
 
 goTo.marker('insideStatement');
 verify.indentationIs(8);
 
 goTo.marker('afterStatement');
 verify.indentationIs(4);
+
+goTo.marker('insideStatement2');
+verify.indentationIs(8);
diff --git a/tests/cases/fourslash/smartIndentStatementForOf.ts b/tests/cases/fourslash/smartIndentStatementForOf.ts
@@ -0,0 +1,20 @@
+/// <reference path='fourslash.ts'/>
+
+////function Foo() {
+////    for (var i of [])
+////    {
+////        /*insideStatement*/
+////    }
+////    /*afterStatement*/
+////    for (var i of [])
+////        /*insideStatement2*/
+////}
+
+goTo.marker('insideStatement');
+verify.indentationIs(8);
+
+goTo.marker('afterStatement');
+verify.indentationIs(4);
+
+goTo.marker('insideStatement2');
+verify.indentationIs(8);
