Complete for-each and switch statements

pkg/analysis_server/test/services/completion/statement/statement_completion_test.dart

 // Copyright (c) 2017, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library test.services.completion.statement;
 
 import 'package:analysis_server/src/protocol_server.dart';
 import 'package:analysis_server/src/services/completion/statement/statement_completion.dart';
 import 'package:analyzer/src/dart/analysis/driver.dart';
 import 'package:test/test.dart';
 import 'package:test_reflective_loader/test_reflective_loader.dart';
 
 import '../../../abstract_single_unit.dart';
 
 main() {
   defineReflectiveSuite(() {
     defineReflectiveTests(_DoCompletionTest);
     defineReflectiveTests(_ForCompletionTest);
+    defineReflectiveTests(_ForEachCompletionTest);
     defineReflectiveTests(_IfCompletionTest);
     defineReflectiveTests(_SimpleCompletionTest);
+    defineReflectiveTests(_SwitchCompletionTest);
     defineReflectiveTests(_WhileCompletionTest);
   });
 }
 
 class StatementCompletionTest extends AbstractSingleUnitTest {
   SourceChange change;
 
   bool get enableNewAnalysisDriver => true;
 
   int _after(String source, String match) =>
@@ skipping 311 matching lines @@
 main() {
   for (int i = 0; ) {
   }
 }
 ''',
         (s) => _after(s, '0; '));
   }
 }
 
 @reflectiveTest
+class _ForEachCompletionTest extends StatementCompletionTest {
+  test_emptyIdentifier() async {
+    await _prepareCompletion(
+        'in xs)',
+        '''
+main() {
+  for (in xs)
+}
+''',
+        atEnd: true);
+    _assertHasChange(
+        'Complete for-each-statement',
+        '''
+main() {
+  for ( in xs) {
+    ////
+  }
+}
+''',
+        (s) => _after(s, 'for ('));
+  }
+
+  test_emptyIdentifierAndIterable() async {
+    await _prepareCompletion(
+        'in)',
+        '''
+main() {
+  for (in)
+}
+''',
+        atEnd: true);
+    _assertHasChange(
+        'Complete for-each-statement',
+        '''
+main() {
+  for ( in ) {
+    ////
+  }
+}
+''',
+        (s) => _after(s, 'for ('));
+  }
+
+  test_emptyIterable() async {
+    await _prepareCompletion(
+        'in)',
+        '''
+main() {
+  for (var x in)
+}
+''',
+        atEnd: true);
+    _assertHasChange(
+        'Complete for-each-statement',
+        '''
+main() {
+  for (var x in ) {
+    ////
+  }
+}
+''',
+        (s) => _after(s, 'in '));
+  }
+}
+
+@reflectiveTest
 class _IfCompletionTest extends StatementCompletionTest {
   test_afterCondition_BAD() async {
     // TODO(messick): Fix the code to make this like test_completeIfWithCondition.
     // Recap: Finding the node at the selectionOffset returns the block, not the
     // if-statement. Need to understand if that only happens when the if-statement
     // is the only statement in the block, or perhaps first or last? And what
     // happens when it is in the middle of other statements?
     await _prepareCompletion(
         'if (true) ', // Trigger completion after space.
         '''
@@ skipping 158 matching lines @@
 main() {
   int v = 1;
   ////
 }
 ''',
         (s) => _afterLast(s, '  '));
   }
 }
 
 @reflectiveTest
+class _SwitchCompletionTest extends StatementCompletionTest {
+  test_emptyCondition() async {
+    await _prepareCompletion(
+        'switch',
+        '''
+main() {
+  switch ()
+}
+''',
+        atEnd: true);
+    _assertHasChange(
+        'Complete switch-statement',
+        '''
+main() {
+  switch () {
+    ////
+  }
+}
+''',
+        (s) => _after(s, 'switch ('));
+  }
+
+  test_keywordOnly() async {
+    await _prepareCompletion(
+        'switch',
+        '''
+main() {
+  switch////
+}
+''',
+        atEnd: true);
+    _assertHasChange(
+        'Complete switch-statement',
+        '''
+main() {
+  switch () {
+    ////
+  }
+}
+''',
+        (s) => _after(s, 'switch ('));
+  }
+
+  test_keywordSpace() async {
+    await _prepareCompletion(
+        'switch',
+        '''
+main() {
+  switch ////
+}
+''',
+        atEnd: true);
+    _assertHasChange(
+        'Complete switch-statement',
+        '''
+main() {
+  switch () {
+    ////
+  }
+}
+''',
+        (s) => _after(s, 'switch ('));
+  }
+}
+
+@reflectiveTest
 class _WhileCompletionTest extends StatementCompletionTest {
   /*
      The implementation of completion for while-statements is shared with
      if-statements. Here we check that the wrapper for while-statements
      functions as expected. The individual test cases are covered by the
      _IfCompletionTest tests. If the implementation changes then the same
      set of tests defined for if-statements should be duplicated here.
    */
   test_keywordOnly() async {
     await _prepareCompletion(
         'while',
         '''
 main() {
   while ////
 }
 ''',
         atEnd: true);
     _assertHasChange(
         'Complete while-statement',
         '''
 main() {
   while () {
     ////
   }
 }
 ''',
         (s) => _after(s, 'while ('));
   }
 }