Issue 4238. Infer Element subclass from query() parameter

compiler/javatests/com/google/dart/compiler/end2end/inc/IncrementalCompilation2Test.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.end2end.inc;
 
 import static com.google.dart.compiler.DartCompiler.EXTENSION_DEPS;
 import static com.google.dart.compiler.DartCompiler.EXTENSION_TIMESTAMP;
 import static com.google.dart.compiler.common.ErrorExpectation.assertErrors;
 import static com.google.dart.compiler.common.ErrorExpectation.errEx;
 
 import com.google.common.collect.Lists;
+import com.google.common.collect.Maps;
 import com.google.common.collect.Sets;
 import com.google.dart.compiler.CompilerTestCase;
 import com.google.dart.compiler.DartCompilationError;
 import com.google.dart.compiler.DartCompiler;
 import com.google.dart.compiler.DartCompilerErrorCode;
 import com.google.dart.compiler.DartCompilerListener;
 import com.google.dart.compiler.DartSource;
 import com.google.dart.compiler.DefaultCompilerConfiguration;
 import com.google.dart.compiler.LibrarySource;
 import com.google.dart.compiler.MockArtifactProvider;
 import com.google.dart.compiler.Source;
 import com.google.dart.compiler.SystemLibraryManager;
 import com.google.dart.compiler.UrlSource;
 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.resolver.ResolverErrorCode;
 import com.google.dart.compiler.resolver.TypeErrorCode;
 
 import junit.framework.AssertionFailedError;
 
 import java.io.IOException;
 import java.io.Reader;
 import java.io.Writer;
 import java.net.URI;
 import java.util.List;
+import java.util.Map;
 import java.util.Set;
 import java.util.concurrent.ConcurrentSkipListSet;
 
 // TODO(zundel): update this test not to rely on code generation
 public class IncrementalCompilation2Test extends CompilerTestCase {
   private static final String APP = "Application.dart";
 
   private static class IncMockArtifactProvider extends MockArtifactProvider {
     Set<String> reads = new ConcurrentSkipListSet<String>();
     Set<String> writes = new ConcurrentSkipListSet<String>();
@@ skipping 13 matching lines @@
     void resetReadsAndWrites() {
       reads.clear();
       writes.clear();
     }
   }
 
   private DefaultCompilerConfiguration config;
   private IncMockArtifactProvider provider;
   private MemoryLibrarySource appSource;
   private final List<DartCompilationError> errors = Lists.newArrayList();
+  private final Map<String, DartUnit> units = Maps.newHashMap();
 
   @Override
   protected void setUp() throws Exception {
     config = new DefaultCompilerConfiguration() {
       @Override
       public boolean incremental() {
         return true;
       }
     };
     provider = new IncMockArtifactProvider();
@@ skipping 10 matching lines @@
     appSource.setContent("A.dart", "");
     appSource.setContent("B.dart", "");
     appSource.setContent("C.dart", "");
   }
 
   @Override
   protected void tearDown() {
     config = null;
     provider = null;
     appSource = null;
+    errors.clear();
+    units.clear();
   }
 
   /**
    * "not_hole" is referenced using "super" qualifier, so is not affected by declaring top-level
    * field with same name.
    */
   public void test_useQualifiedFieldReference_ignoreTopLevelDeclaration() {
     appSource.setContent(
         "B.dart",
         makeCode(
@@ skipping 762 matching lines @@
             "#library('application');",
             "#import('A.dart', prefix: 'prf');",
             "main() {",
             "  var prf;",
             "}",
             ""));
     // do compile, no errors expected
     compile();
     assertErrors(errors, errEx(ResolverErrorCode.CANNOT_HIDE_IMPORT_PREFIX, 5, 7, 3));
   }
+  
+  /**
+   * <p>
+   * http://code.google.com/p/dart/issues/detail?id=4238
+   */
+  public void test_typesPropagation_html_query() throws Exception {
+    appSource.setContent(
+        APP,
+        makeCode(
+            "// filler filler filler filler filler filler filler filler filler filler filler",
+            "#library('application');",
+            "#import('dart:html');",
+            "main() {",
+            "  var v1 = query('a');",
+            "  var v2 = query('A');",
+            "  var v3 = query('body:active');",
+            "  var v4 = query('button[foo=\"bar\"]');",
+            "  var v5 = query('div.class');",
+            "  var v6 = query('input#id');",
+            "  // invocation of method",
+            "  var m1 = document.query('div');",
+            "  // unsupported currently",
+            "  var b1 = query('noSuchTag');",
+            "  var b2 = query('DART_EDITOR_NO_SUCH_TYPE');",
+            "  var b3 = query('body div');",
+            "}",
+            ""));
+    // do compile, no errors expected
+    compile();
+    assertErrors(errors);
+    // validate types
+    DartUnit unit = units.get(APP);
+    assertNotNull(unit);
+    assertInferredElementTypeString(unit, "v1", "AnchorElement");
+    assertInferredElementTypeString(unit, "v2", "AnchorElement");
+    assertInferredElementTypeString(unit, "v3", "BodyElement");
+    assertInferredElementTypeString(unit, "v4", "ButtonElement");
+    assertInferredElementTypeString(unit, "v5", "DivElement");
+    assertInferredElementTypeString(unit, "v6", "InputElement");
+    // invocation of method
+    assertInferredElementTypeString(unit, "m1", "DivElement");
+    // bad cases, or unsupported now
+    assertInferredElementTypeString(unit, "b1", "Element");
+    assertInferredElementTypeString(unit, "b2", "Element");
+    assertInferredElementTypeString(unit, "b3", "Element");
+  }
 
   /**
    * Libraries "dart:io" and "dart:html" can not be used together in single application.
    * <p>
    * http://code.google.com/p/dart/issues/detail?id=3839
    */
   public void test_consoleWebMix() throws Exception {
     appSource.setContent(
         APP,
         makeCode(
@@ skipping 93 matching lines @@
           // Remember errors only between unitAboutToCompile/unitCompiled.
           Source source = event.getSource();
           if (source != null && compilingUris.contains(source.getUri())) {
             errors.add(event);
           }
         }
 
         @Override
         public void unitCompiled(DartUnit unit) {
           compilingUris.remove(unit.getSourceInfo().getSource().getUri());
+          units.put(unit.getSourceName(), unit);
         }
       };
       DartCompiler.compileLib(lib, config, provider, listener);
     } catch (IOException e) {
       throw new AssertionFailedError("Unexpected IOException: " + e.getMessage());
     }
   }
 
   private void didWrite(String sourceName, String extension) {
     String spec = sourceName + "/" + extension;
     assertTrue("Expected write: " + spec, provider.writes.contains(spec));
   }
 
   private void didNotWrite(String sourceName, String extension) {
     String spec = sourceName + "/" + extension;
     assertFalse("Didn't expect write: " + spec, provider.writes.contains(spec));
   }
 }