Prepare for decoupling analyzer ASTs from element model.

pkg/analysis_server/test/src/utilities/change_builder_dart_test.dart

 // Copyright (c) 2015, 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 analysis_server.test.src.utilities.change_builder_dart_test;
 
 import 'package:analysis_server/plugin/protocol/protocol.dart';
 import 'package:analysis_server/src/provisional/edit/utilities/change_builder_core.dart';
 import 'package:analysis_server/src/provisional/edit/utilities/change_builder_dart.dart';
 import 'package:analysis_server/src/utilities/change_builder_dart.dart';
 import 'package:analyzer/dart/ast/ast.dart';
+import 'package:analyzer/dart/ast/resolution_accessors.dart';
 import 'package:analyzer/dart/element/element.dart';
 import 'package:analyzer/src/generated/source.dart';
 import 'package:test/test.dart';
 import 'package:test_reflective_loader/test_reflective_loader.dart';
 
 import '../../abstract_context.dart';
 
 main() {
   defineReflectiveSuite(() {
     defineReflectiveTests(DartChangeBuilderImplTest);
@@ skipping 33 matching lines @@
   }
 
   void test_writeClassDeclaration_interfaces() {
     Source source = addSource('/test.dart', 'class A {}');
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration declaration = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(0, (EditBuilder builder) {
-        (builder as DartEditBuilder)
-            .writeClassDeclaration('C', interfaces: [declaration.element.type]);
+        (builder as DartEditBuilder).writeClassDeclaration('C',
+            interfaces: [elementForClassDeclaration(declaration).type]);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(
         edit.replacement, equalsIgnoringWhitespace('class C implements A { }'));
   }
 
   void test_writeClassDeclaration_isAbstract() {
     Source source = addSource('/test.dart', '');
     resolveLibraryUnit(source);
@@ skipping 27 matching lines @@
   }
 
   void test_writeClassDeclaration_mixins_noSuperclass() {
     Source source = addSource('/test.dart', 'class A {}');
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(0, (EditBuilder builder) {
-        (builder as DartEditBuilder)
-            .writeClassDeclaration('C', mixins: [classA.element.type]);
+        (builder as DartEditBuilder).writeClassDeclaration('C',
+            mixins: [elementForClassDeclaration(classA).type]);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement,
         equalsIgnoringWhitespace('class C extends Object with A { }'));
   }
 
   void test_writeClassDeclaration_mixins_superclass() {
     Source source = addSource('/test.dart', 'class A {} class B {}');
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
     ClassDeclaration classB = unit.declarations[1];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(0, (EditBuilder builder) {
         (builder as DartEditBuilder).writeClassDeclaration('C',
-            mixins: [classB.element.type], superclass: classA.element.type);
+            mixins: [elementForClassDeclaration(classB).type],
+            superclass: elementForClassDeclaration(classA).type);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement,
         equalsIgnoringWhitespace('class C extends A with B { }'));
   }
 
   void test_writeClassDeclaration_nameGroupName() {
     Source source = addSource('/test.dart', '');
     resolveLibraryUnit(source);
@@ skipping 17 matching lines @@
   }
 
   void test_writeClassDeclaration_superclass() {
     Source source = addSource('/test.dart', 'class B {}');
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration declaration = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(0, (EditBuilder builder) {
-        (builder as DartEditBuilder)
-            .writeClassDeclaration('C', superclass: declaration.element.type);
+        (builder as DartEditBuilder).writeClassDeclaration('C',
+            superclass: elementForClassDeclaration(declaration).type);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('class C extends B { }'));
   }
 
   void test_writeFieldDeclaration_initializerWriter() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     resolveLibraryUnit(source);
@@ skipping 100 matching lines @@
   void test_writeFieldDeclaration_type_typeGroupName() {
     String content = 'class A {} class B {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration declaration = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder).writeFieldDeclaration('f',
-            type: declaration.element.type, typeGroupName: 'type');
+            type: elementForClassDeclaration(declaration).type,
+            typeGroupName: 'type');
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('A f;'));
 
     List<LinkedEditGroup> linkedEditGroups =
         builder.sourceChange.linkedEditGroups;
     expect(linkedEditGroups, hasLength(1));
     LinkedEditGroup group = linkedEditGroups[0];
     expect(group.length, 1);
@@ skipping 64 matching lines @@
   void test_writeGetterDeclaration_returnType() {
     String content = 'class A {} class B {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder).writeGetterDeclaration('g',
-            returnType: classA.element.type, returnTypeGroupName: 'returnType');
+            returnType: elementForClassDeclaration(classA).type,
+            returnTypeGroupName: 'returnType');
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('A get g => null;'));
 
     List<LinkedEditGroup> linkedEditGroups =
         builder.sourceChange.linkedEditGroups;
     expect(linkedEditGroups, hasLength(1));
     LinkedEditGroup group = linkedEditGroups[0];
     expect(group.length, 1);
     expect(group.positions, hasLength(1));
     Position position = group.positions[0];
     expect(position.offset, equals(20));
   }
 
   void test_writeOverrideOfInheritedMember() {
     String content = '''
 class A {
   A add(A a) => null;
 }
 class B extends A {
 }''';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration declaration = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilder)
-            .writeOverrideOfInheritedMember(declaration.element.methods[0]);
+        (builder as DartEditBuilder).writeOverrideOfInheritedMember(
+            elementForClassDeclaration(declaration).methods[0]);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('''
 @override
 A add(A a) {
   // TODO: implement add
   return null;
 }'''));
   }
 
   void test_writeParameters_named() {
     String content = 'f(int i, {String s}) {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     FunctionDeclaration f = unit.declarations[0];
     FormalParameterList parameters = f.functionExpression.parameters;
-    Iterable<ParameterElement> elements = parameters.parameters
-        .map((FormalParameter parameter) => parameter.element);
+    Iterable<ParameterElement> elements =
+        parameters.parameters.map(elementForFormalParameter);
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder).writeParameters(elements);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('(int i, {String s})'));
   }
 
   void test_writeParameters_positional() {
     String content = 'f(int i, [String s]) {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     FunctionDeclaration f = unit.declarations[0];
     FormalParameterList parameters = f.functionExpression.parameters;
-    Iterable<ParameterElement> elements = parameters.parameters
-        .map((FormalParameter parameter) => parameter.element);
+    Iterable<ParameterElement> elements =
+        parameters.parameters.map(elementForFormalParameter);
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder).writeParameters(elements);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('(int i, [String s])'));
   }
 
   void test_writeParameters_required() {
     String content = 'f(int i, String s) {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     FunctionDeclaration f = unit.declarations[0];
     FormalParameterList parameters = f.functionExpression.parameters;
-    Iterable<ParameterElement> elements = parameters.parameters
-        .map((FormalParameter parameter) => parameter.element);
+    Iterable<ParameterElement> elements =
+        parameters.parameters.map(elementForFormalParameter);
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder).writeParameters(elements);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('(int i, String s)'));
   }
@@ skipping 48 matching lines @@
   void test_writeParameterSource() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder)
-            .writeParameterSource(classA.element.type, 'a');
+            .writeParameterSource(elementForClassDeclaration(classA).type, 'a');
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('A a'));
   }
 
   void test_writeType_dymanic() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilder)
-            .writeType(unit.element.context.typeProvider.dynamicType);
+        (builder as DartEditBuilder).writeType(
+            elementForCompilationUnit(unit).context.typeProvider.dynamicType);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace(''));
   }
 
   void test_writeType_genericType() {
     String content = 'class A {} class B<E> {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
     ClassDeclaration classB = unit.declarations[1];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilder)
-            .writeType(classB.element.type.instantiate([classA.element.type]));
+        (builder as DartEditBuilder).writeType(
+            elementForClassDeclaration(classB)
+                .type
+                .instantiate([elementForClassDeclaration(classA).type]));
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('B<A>'));
   }
 
   void test_writeType_groupName() {
     String content = 'class A {} class B extends A {} class C extends B {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classC = unit.declarations[2];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilder)
-            .writeType(classC.element.type, groupName: 'type');
+        (builder as DartEditBuilder).writeType(
+            elementForClassDeclaration(classC).type,
+            groupName: 'type');
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('C'));
 
     List<LinkedEditGroup> linkedEditGroups =
         builder.sourceChange.linkedEditGroups;
     expect(linkedEditGroups, hasLength(1));
     LinkedEditGroup group = linkedEditGroups[0];
     expect(group, isNotNull);
   }
 
   void test_writeType_groupName_addSupertypeProposals() {
     String content = 'class A {} class B extends A {} class C extends B {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classC = unit.declarations[2];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilder).writeType(classC.element.type,
-            addSupertypeProposals: true, groupName: 'type');
+        (builder as DartEditBuilder).writeType(
+            elementForClassDeclaration(classC).type,
+            addSupertypeProposals: true,
+            groupName: 'type');
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('C'));
 
     List<LinkedEditGroup> linkedEditGroups =
         builder.sourceChange.linkedEditGroups;
     expect(linkedEditGroups, hasLength(1));
     LinkedEditGroup group = linkedEditGroups[0];
     List<LinkedEditSuggestion> suggestions = group.suggestions;
@@ skipping 26 matching lines @@
 
   void test_writeType_required_dymanic() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder).writeType(
-            unit.element.context.typeProvider.dynamicType,
+            elementForCompilationUnit(unit).context.typeProvider.dynamicType,
             required: true);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('var'));
   }
 
   void test_writeType_required_notNull() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
         (builder as DartEditBuilder)
-            .writeType(classA.element.type, required: true);
+            .writeType(elementForClassDeclaration(classA).type, required: true);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('A'));
   }
 
   void test_writeType_required_null() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     resolveLibraryUnit(source);
@@ skipping 10 matching lines @@
 
   void test_writeType_simpleType() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilder).writeType(classA.element.type);
+        (builder as DartEditBuilder)
+            .writeType(elementForClassDeclaration(classA).type);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('A'));
   }
 
   void test_writeTypes_empty() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     resolveLibraryUnit(source);
@@ skipping 11 matching lines @@
   void test_writeTypes_noPrefix() {
     String content = 'class A {} class B {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
     ClassDeclaration classB = unit.declarations[1];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilderImpl)
-            .writeTypes([classA.element.type, classB.element.type]);
+        (builder as DartEditBuilderImpl).writeTypes([
+          elementForClassDeclaration(classA).type,
+          elementForClassDeclaration(classB).type
+        ]);
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('A, B'));
   }
 
   void test_writeTypes_null() {
     String content = 'class A {}';
     Source source = addSource('/test.dart', content);
     resolveLibraryUnit(source);
@@ skipping 11 matching lines @@
   void test_writeTypes_prefix() {
     String content = 'class A {} class B {}';
     Source source = addSource('/test.dart', content);
     CompilationUnit unit = resolveLibraryUnit(source);
     ClassDeclaration classA = unit.declarations[0];
     ClassDeclaration classB = unit.declarations[1];
 
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, 1, (FileEditBuilder builder) {
       builder.addInsertion(content.length - 1, (EditBuilder builder) {
-        (builder as DartEditBuilderImpl).writeTypes(
-            [classA.element.type, classB.element.type],
-            prefix: 'implements ');
+        (builder as DartEditBuilderImpl).writeTypes([
+          elementForClassDeclaration(classA).type,
+          elementForClassDeclaration(classB).type
+        ], prefix: 'implements ');
       });
     });
     SourceEdit edit = getEdit(builder);
     expect(edit.replacement, equalsIgnoringWhitespace('implements A, B'));
   }
 }
 
 @reflectiveTest
 class DartFileEditBuilderImplTest extends AbstractContextTest {
   void test_createEditBuilder() {
     Source source = addSource('/test.dart', 'library test;');
     resolveLibraryUnit(source);
     int timeStamp = 65;
     DartChangeBuilderImpl builder = new DartChangeBuilderImpl(context);
     builder.addFileEdit(source, timeStamp, (FileEditBuilder builder) {
       int offset = 4;
       int length = 5;
       DartEditBuilderImpl editBuilder = (builder as DartFileEditBuilderImpl)
           .createEditBuilder(offset, length);
       expect(editBuilder, new isInstanceOf<DartEditBuilder>());
       SourceEdit sourceEdit = editBuilder.sourceEdit;
       expect(sourceEdit.length, length);
       expect(sourceEdit.offset, offset);
       expect(sourceEdit.replacement, isEmpty);
     });
   }
 }