Fix for comment references tests.

pkg/analyzer/test/generated/non_error_resolver_test.dart

 // Copyright (c) 2014, 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 engine.non_error_resolver_test;
 
 import 'package:analyzer/src/generated/ast.dart';
 import 'package:analyzer/src/generated/element.dart';
 import 'package:analyzer/src/generated/engine.dart';
 import 'package:analyzer/src/generated/error.dart';
@@ skipping 777 matching lines @@
   void test_commentReference_beforeConstructor() {
     String code = r'''
 abstract class A {
   /// [p]
   A(int p) {}
 }''';
     Source source = addSource(code);
     resolve(source);
     assertNoErrors(source);
     verify([source]);
-    CompilationUnit unit = analysisContext.parseCompilationUnit(source);
+    CompilationUnit unit = _getResolvedLibraryUnit(source);
     {
       SimpleIdentifier ref = EngineTestCase.findNode(
           unit, code, "p]", (node) => node is SimpleIdentifier);
       EngineTestCase.assertInstanceOf((obj) => obj is ParameterElement,
           ParameterElement, ref.staticElement);
     }
   }
 
   void test_commentReference_beforeFunction_blockBody() {
     String code = r'''
 /// [p]
 foo(int p) {
 }''';
     Source source = addSource(code);
     resolve(source);
     assertNoErrors(source);
     verify([source]);
-    CompilationUnit unit = analysisContext.parseCompilationUnit(source);
+    CompilationUnit unit = _getResolvedLibraryUnit(source);
     SimpleIdentifier ref = EngineTestCase.findNode(
         unit, code, "p]", (node) => node is SimpleIdentifier);
     EngineTestCase.assertInstanceOf(
         (obj) => obj is ParameterElement, ParameterElement, ref.staticElement);
   }
 
   void test_commentReference_beforeFunction_expressionBody() {
     String code = r'''
 /// [p]
 foo(int p) => null;''';
     Source source = addSource(code);
     resolve(source);
     assertNoErrors(source);
     verify([source]);
-    CompilationUnit unit = analysisContext.parseCompilationUnit(source);
+    CompilationUnit unit = _getResolvedLibraryUnit(source);
     SimpleIdentifier ref = EngineTestCase.findNode(
         unit, code, "p]", (node) => node is SimpleIdentifier);
     EngineTestCase.assertInstanceOf(
         (obj) => obj is ParameterElement, ParameterElement, ref.staticElement);
   }
 
   void test_commentReference_beforeMethod() {
     String code = r'''
 abstract class A {
   /// [p1]
   ma(int p1) {}
   /// [p2]
   mb(int p2);
 }''';
     Source source = addSource(code);
     resolve(source);
     assertNoErrors(source);
     verify([source]);
-    CompilationUnit unit = analysisContext.parseCompilationUnit(source);
+    CompilationUnit unit = _getResolvedLibraryUnit(source);
     {
       SimpleIdentifier ref = EngineTestCase.findNode(
           unit, code, "p1]", (node) => node is SimpleIdentifier);
       EngineTestCase.assertInstanceOf((obj) => obj is ParameterElement,
           ParameterElement, ref.staticElement);
     }
     {
       SimpleIdentifier ref = EngineTestCase.findNode(
           unit, code, "p2]", (node) => node is SimpleIdentifier);
       EngineTestCase.assertInstanceOf((obj) => obj is ParameterElement,
           ParameterElement, ref.staticElement);
     }
   }
 
   void test_commentReference_class() {
     String code = r'''
 /// [foo]
 class A {
   foo() {}
 }''';
     Source source = addSource(code);
     resolve(source);
     assertNoErrors(source);
     verify([source]);
-    CompilationUnit unit = analysisContext.parseCompilationUnit(source);
+    CompilationUnit unit = _getResolvedLibraryUnit(source);
     SimpleIdentifier ref = EngineTestCase.findNode(
         unit, code, "foo]", (node) => node is SimpleIdentifier);
     EngineTestCase.assertInstanceOf(
         (obj) => obj is MethodElement, MethodElement, ref.staticElement);
   }
 
   void test_commentReference_setter() {
     String code = r'''
 class A {
   /// [x] in A
   mA() {}
   set x(value) {}
 }
 class B extends A {
   /// [x] in B
   mB() {}
 }
 ''';
     Source source = addSource(code);
     resolve(source);
     assertNoErrors(source);
     verify([source]);
-    CompilationUnit unit = analysisContext.parseCompilationUnit(source);
+    CompilationUnit unit = _getResolvedLibraryUnit(source);
     {
       SimpleIdentifier ref = EngineTestCase.findNode(
           unit, code, "x] in A", (node) => node is SimpleIdentifier);
       EngineTestCase.assertInstanceOf((obj) => obj is PropertyAccessorElement,
           PropertyAccessorElement, ref.staticElement);
     }
     {
       SimpleIdentifier ref = EngineTestCase.findNode(
           unit, code, "x] in B", (node) => node is SimpleIdentifier);
       EngineTestCase.assertInstanceOf((obj) => obj is PropertyAccessorElement,
@@ skipping 4593 matching lines @@
 }""");
     resolve(source);
     assertNoErrors(source);
     verify([source]);
     reset();
   }
 
   void _check_wrongNumberOfParametersForOperator1(String name) {
     _check_wrongNumberOfParametersForOperator(name, "a");
   }
+
+  CompilationUnit _getResolvedLibraryUnit(Source source) =>
+      analysisContext.getResolvedCompilationUnit2(source, source);
 }