(belated) Tests for CompilationUnitElement.getElementAt().

pkg/analyzer/test/generated/element_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.
 
 // This code was auto-generated, is not intended to be edited, and is subject to
 // significant change. Please see the README file for more information.
 
 library engine.element_test;
 
 import 'package:analyzer/src/generated/ast.dart';
@@ skipping 898 matching lines @@
       ClassTypeAlias nodeA = elementA.node;
       expect(nodeA, isNotNull);
       expect(nodeA.name.name, "A");
       expect(nodeA.element, same(elementA));
     }
   }
 }
 
 @reflectiveTest
 class CompilationUnitElementImplTest extends EngineTestCase {
+  void test_getElementAt() {
+    AnalysisContextHelper contextHelper = new AnalysisContextHelper();
+    AnalysisContext context = contextHelper.context;
+    String code = r'''
+class A {
+  int field;
+}
+main() {
+  int localVar = 42;
+}
+''';
+    Source libSource = contextHelper.addSource("/my_lib.dart", code);
+    // prepare library/unit elements
+    LibraryElement libraryElement = context.computeLibraryElement(libSource);
+    CompilationUnitElement unitElement = libraryElement.definingCompilationUnit;
+    // A
+    ClassElement elementA;
+    {
+      int offset = code.indexOf('A {');
+      elementA = unitElement.getElementAt(offset);
+      expect(elementA, isNotNull);
+      expect(elementA.enclosingElement, unitElement);
+      expect(elementA.name, 'A');
+    }
+    // A.field
+    {
+      int offset = code.indexOf('field;');
+      FieldElement element = unitElement.getElementAt(offset);
+      expect(element, isNotNull);
+      expect(element.enclosingElement, elementA);
+      expect(element.name, 'field');
+    }
+    // main
+    FunctionElement mainElement;
+    {
+      int offset = code.indexOf('main() {');
+      mainElement = unitElement.getElementAt(offset);
+      expect(mainElement, isNotNull);
+      expect(mainElement.enclosingElement, unitElement);
+      expect(mainElement.name, 'main');
+    }
+    // main.localVar
+    {
+      int offset = code.indexOf('localVar');
+      LocalVariableElement element = unitElement.getElementAt(offset);
+      expect(element, isNotNull);
+      expect(element.enclosingElement, mainElement);
+      expect(element.name, 'localVar');
+    }
+    // null
+    expect(unitElement.getElementAt(1000), isNull);
+  }
+
+  void test_getElementAt_multipleUnitsInLibrary() {
+    AnalysisContextHelper contextHelper = new AnalysisContextHelper();
+    AnalysisContext context = contextHelper.context;
+    Source libSource = contextHelper.addSource("/my_lib.dart", r'''
+library my_lib;
+part 'unit_a.dart';
+part 'unit_b.dart';
+''');
+    Source unitSourceA =
+        contextHelper.addSource("/unit_a.dart", 'part of my_lib;class A {}');
+    Source unitSourceB =
+        contextHelper.addSource("/unit_b.dart", 'part of my_lib;class B {}');
+    int offset = 'part of my_lib;class A {}'.indexOf('A {}');
+    // prepare library/unit elements
+    context.computeLibraryElement(libSource);
+    CompilationUnitElement unitElementA =
+        context.getCompilationUnitElement(unitSourceA, libSource);
+    CompilationUnitElement unitElementB =
+        context.getCompilationUnitElement(unitSourceB, libSource);
+    // A
+    {
+      ClassElement element = unitElementA.getElementAt(offset);
+      expect(element, isNotNull);
+      expect(element.enclosingElement, unitElementA);
+      expect(element.name, 'A');
+    }
+    // B
+    {
+      ClassElement element = unitElementB.getElementAt(offset);
+      expect(element, isNotNull);
+      expect(element.enclosingElement, unitElementB);
+      expect(element.name, 'B');
+    }
+  }
+
   void test_getEnum_declared() {
     TestTypeProvider typeProvider = new TestTypeProvider();
     CompilationUnitElementImpl unit =
         ElementFactory.compilationUnit("/lib.dart");
     String enumName = "E";
     ClassElement enumElement =
         ElementFactory.enumElement(typeProvider, enumName);
     unit.enums = <ClassElement>[enumElement];
     expect(unit.getEnum(enumName), same(enumElement));
   }
@@ skipping 2943 matching lines @@
 }
 
 class _FunctionTypeImplTest_isSubtypeOf_baseCase_classFunction
     extends InterfaceTypeImpl {
   _FunctionTypeImplTest_isSubtypeOf_baseCase_classFunction(ClassElement arg0)
       : super.con1(arg0);
 
   @override
   bool get isDartCoreFunction => true;
 }