Issue 4238. Infer Element subclass from query() parameter

compiler/javatests/com/google/dart/compiler/CompilerTestCase.java

 // Copyright (c) 2012, 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.
 
 package com.google.dart.compiler;
 
 import static com.google.dart.compiler.common.ErrorExpectation.assertErrors;
 
 import com.google.common.collect.Lists;
 import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
 import com.google.dart.compiler.CommandLineOptions.CompilerOptions;
 import com.google.dart.compiler.ast.ASTVisitor;
 import com.google.dart.compiler.ast.DartExpression;
 import com.google.dart.compiler.ast.DartFunctionTypeAlias;
+import com.google.dart.compiler.ast.DartIdentifier;
 import com.google.dart.compiler.ast.DartNode;
 import com.google.dart.compiler.ast.DartUnit;
 import com.google.dart.compiler.ast.LibraryUnit;
 import com.google.dart.compiler.common.ErrorExpectation;
 import com.google.dart.compiler.common.SourceInfo;
 import com.google.dart.compiler.parser.DartParser;
 import com.google.dart.compiler.parser.DartParserRunner;
+import com.google.dart.compiler.resolver.Element;
+import com.google.dart.compiler.type.Type;
+import com.google.dart.compiler.type.TypeKind;
 
 import junit.framework.TestCase;
 
 import java.io.IOException;
 import java.io.InputStreamReader;
 import java.io.Reader;
 import java.net.URI;
 import java.net.URL;
 import java.util.List;
 import java.util.Map;
 
 /**
  * Base class for compiler tests, with helpful utility methods.
  */
 public abstract class CompilerTestCase extends TestCase {
 
   private static final String UTF8 = "UTF-8";
   protected CompilerConfiguration compilerConfiguration;
+  protected DartUnit testUnit;
 
   /**
    * Instance of {@link CompilerConfiguration} for incremental check-only compilation.
    */
   protected static final CompilerConfiguration CHECK_ONLY_CONFIGURATION =
       new DefaultCompilerConfiguration(new CompilerOptions()) {
         @Override
         public boolean incremental() {
           return true;
         }
@@ skipping 113 matching lines @@
     }
     return buf.toString();
   }
   
   @Override
   protected void setUp() throws Exception {
     super.setUp();
     compilerConfiguration = CHECK_ONLY_CONFIGURATION;
   }
 
+  @Override
+  protected void tearDown() throws Exception {
+    compilerConfiguration = null;
+    testUnit = null;
+    super.tearDown();
+  }
+
   protected AnalyzeLibraryResult analyzeLibrary(String... lines) throws Exception {
-    return analyzeLibrary(getName(), makeCode(lines));
+    String name = getName();
+    AnalyzeLibraryResult libraryResult = analyzeLibrary(name, makeCode(lines));
+    testUnit = libraryResult.getLibraryUnitResult().getUnit(name);
+    return libraryResult;
   }
 
   /**
    * Simulate running {@code analyzeLibrary} the way the IDE will.
    * <p>
    * <b>Note:</b> if the IDE changes how it calls analyzeLibrary, this should
    * be changed to match.
    *
    * @param name the name to use for the source file
    * @param code the Dart code to parse/analyze
@@ skipping 208 matching lines @@
         return null;
       }
     });
     return result[0];
   }
 
   public static String getNodeSource(String code, DartNode node) {
     SourceInfo sourceInfo = node.getSourceInfo();
     return code.substring(sourceInfo.getOffset(), sourceInfo.getEnd());
   }
+
+
+  /**
+   * Asserts that {@link Element} with given name has expected type.
+   */
+  protected static void assertInferredElementTypeString(
+      DartUnit unit,
+      String variableName,
+      String expectedType) {
+    // find element
+    Element element = getNamedElement(unit, variableName);
+    assertNotNull(element);
+    // check type
+    Type actualType = element.getType();
+    assertEquals(element.getName(), expectedType, actualType.toString());
+    // should be inferred
+    if (TypeKind.of(actualType) != TypeKind.DYNAMIC) {
+      assertTrue("Should be marked as inferred", actualType.isInferred());
+    }
+  }
+
+  /**
+   * @return the {@link Element} with given name, may be <code>null</code>.
+   */
+  private static Element getNamedElement(DartUnit unit, final String name) {
+    final Element[] result = {null};
+    unit.accept(new ASTVisitor<Void>() {
+      @Override
+      public Void visitIdentifier(DartIdentifier node) {
+        Element element = node.getElement();
+        if (element != null && element.getName().equals(name)) {
+          result[0] = element;
+        }
+        return super.visitIdentifier(node);
+      }
+    });
+    return result[0];
+  }
 }