Test clean-up

editor/tools/plugins/com.google.dart.engine_test/src/com/google/dart/engine/context/AnalysisContextFactory.java

 /*
  * Copyright (c) 2013, the Dart project authors.
  * 
  * Licensed under the Eclipse Public License v1.0 (the "License"); you may not use this file except
  * in compliance with the License. You may obtain a copy of the License at
  * 
  * http://www.eclipse.org/legal/epl-v10.html
  * 
  * Unless required by applicable law or agreed to in writing, software distributed under the License
  * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
  * or implied. See the License for the specific language governing permissions and limitations under
  * the License.
  */
 package com.google.dart.engine.context;
 
 import com.google.dart.engine.element.ClassElement;
+import com.google.dart.engine.element.FunctionElement;
 import com.google.dart.engine.element.LibraryElement;
+import com.google.dart.engine.element.MethodElement;
+import com.google.dart.engine.element.TopLevelVariableElement;
 import com.google.dart.engine.internal.context.AnalysisContextImpl;
+import com.google.dart.engine.internal.element.ClassElementImpl;
 import com.google.dart.engine.internal.element.CompilationUnitElementImpl;
 import com.google.dart.engine.internal.element.LibraryElementImpl;
+import com.google.dart.engine.internal.element.TopLevelVariableElementImpl;
 import com.google.dart.engine.internal.resolver.TestTypeProvider;
 import com.google.dart.engine.sdk.DartSdk;
 import com.google.dart.engine.sdk.DirectoryBasedDartSdk;
 import com.google.dart.engine.source.DartUriResolver;
 import com.google.dart.engine.source.FileUriResolver;
 import com.google.dart.engine.source.Source;
 import com.google.dart.engine.source.SourceFactory;
+import com.google.dart.engine.type.InterfaceType;
+import com.google.dart.engine.type.Type;
 
 import static com.google.dart.engine.ast.ASTFactory.libraryIdentifier;
+import static com.google.dart.engine.element.ElementFactory.classElement;
+import static com.google.dart.engine.element.ElementFactory.functionElement;
+import static com.google.dart.engine.element.ElementFactory.methodElement;
+import static com.google.dart.engine.element.ElementFactory.topLevelVariableElement;
 
 import java.util.HashMap;
 
 /**
  * The class {@code AnalysisContextFactory} defines utility methods used to create analysis contexts
  * for testing purposes.
  */
 public final class AnalysisContextFactory {
   /**
    * Create an analysis context that has a fake core library already resolved.
@@ skipping 22 matching lines @@
     LibraryElementImpl coreLibrary = new LibraryElementImpl(context, libraryIdentifier(
         "dart",
         "core"));
     coreLibrary.setDefiningCompilationUnit(coreUnit);
     //
     // dart:html
     //
     CompilationUnitElementImpl htmlUnit = new CompilationUnitElementImpl("html_dartium.dart");
     Source htmlSource = sourceFactory.forUri(DartSdk.DART_HTML);
     htmlUnit.setSource(htmlSource);
+    ClassElementImpl elementElement = classElement("Element");
+    InterfaceType elementType = elementElement.getType();
+    ClassElementImpl documentElement = classElement("Document", elementType);
+    ClassElementImpl htmlDocumentElement = classElement("HtmlDocument", documentElement.getType());
+    htmlDocumentElement.setMethods(new MethodElement[] {methodElement(
+        "query",
+        elementType,
+        new Type[] {provider.getStringType()})});
+    htmlUnit.setTypes(new ClassElement[] {
+        classElement("AnchorElement", elementType), classElement("BodyElement", elementType),
+        classElement("ButtonElement", elementType), classElement("DivElement", elementType),
+        documentElement, elementElement, htmlDocumentElement,
+        classElement("InputElement", elementType), classElement("SelectElement", elementType),});
+    htmlUnit.setFunctions(new FunctionElement[] {functionElement(
+        "query",
+        elementElement,
+        new ClassElement[] {provider.getStringType().getElement()},
+        ClassElementImpl.EMPTY_ARRAY)});
+    TopLevelVariableElementImpl document = topLevelVariableElement("document");
+    document.setType(htmlDocumentElement.getType());
+    htmlUnit.setTopLevelVariables(new TopLevelVariableElement[] {document});
     LibraryElementImpl htmlLibrary = new LibraryElementImpl(context, libraryIdentifier(
         "dart",
         "dom",
         "html"));
     htmlLibrary.setDefiningCompilationUnit(htmlUnit);
 
     HashMap<Source, LibraryElement> elementMap = new HashMap<Source, LibraryElement>();
     elementMap.put(coreSource, coreLibrary);
     elementMap.put(htmlSource, htmlLibrary);
 
     context.setContents(coreSource, "");
     context.setContents(htmlSource, "");
     context.recordLibraryElements(elementMap);
     return context;
   }
 
   /**
    * Prevent the creation of instances of this class.
    */
   private AnalysisContextFactory() {
   }
 }