Supporting (predefined) classifier aliases in Lazy Resolve

2012-06-26 23:23:06 +02:00
parent ceafb5ddaf
commit 0c910dba8b
4 changed files with 1522 additions and 1 deletions
@@ -0,0 +1,111 @@
+/*
+ * Copyright 2010-2012 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
+ *
+ * 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 org.jetbrains.jet.lang.resolve.lazy;
+
+import com.google.common.base.Predicates;
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.Lists;
+import com.google.common.collect.Sets;
+import com.intellij.openapi.util.io.FileUtil;
+import com.intellij.util.Function;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.jvm.compiler.NamespaceComparator;
+import org.jetbrains.jet.lang.DefaultModuleConfiguration;
+import org.jetbrains.jet.lang.ModuleConfiguration;
+import org.jetbrains.jet.lang.descriptors.ModuleDescriptor;
+import org.jetbrains.jet.lang.descriptors.NamespaceDescriptor;
+import org.jetbrains.jet.lang.psi.JetFile;
+import org.jetbrains.jet.lang.psi.JetImportDirective;
+import org.jetbrains.jet.lang.psi.JetPsiFactory;
+import org.jetbrains.jet.lang.resolve.BindingTrace;
+import org.jetbrains.jet.lang.resolve.ImportPath;
+import org.jetbrains.jet.lang.resolve.name.FqName;
+import org.jetbrains.jet.lang.resolve.name.FqNameUnsafe;
+import org.jetbrains.jet.lang.resolve.name.Name;
+import org.jetbrains.jet.lang.resolve.scopes.WritableScope;
+import org.jetbrains.jet.lang.types.lang.JetStandardLibrary;
+import org.junit.Test;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Collection;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @author abreslav
+ */
+public class LazyResolveBuiltinClassesTest extends AbstractLazyResolveTest {
+    @Test
+    public void testJetStandardLibrary() throws Exception {
+        List<JetFile> files = Lists.newArrayList();
+        addJetFilesFromDir(files, new File("compiler/frontend/src/jet"));
+        addJetFilesFromDir(files, new File("compiler/frontend/src/jet.src"));
+
+        final Map<FqName, Name> aliases = ImmutableMap.<FqName, Name>builder()
+                .put(new FqName("jet.Unit"), Name.identifier("Tuple0"))
+                .build();
+
+        ModuleDescriptor lazyModule = new ModuleDescriptor(Name.special("<lazy module>"));
+        ResolveSession session = new ResolveSession(
+                project,
+                lazyModule,
+                new SpecialModuleConfiguration(),
+                new FileBasedDeclarationProviderFactory(files),
+                new Function<FqName, Name>() {
+                    @Override
+                    public Name fun(FqName name) {
+                        return aliases.get(name);
+                    }
+                },
+                Predicates.in(Sets.newHashSet(new FqNameUnsafe("jet.Any"), new FqNameUnsafe("jet.Nothing"))));
+
+        NamespaceDescriptor jetNamespaceFromJSL =
+                (NamespaceDescriptor) JetStandardLibrary.getInstance().getInt().getContainingDeclaration();
+        NamespaceDescriptor jetNamespaceFromLazy = lazyModule.getRootNamespace().getMemberScope().getNamespace(
+                jetNamespaceFromJSL.getName());
+
+        NamespaceComparator.compareNamespaces(jetNamespaceFromJSL, jetNamespaceFromLazy,
+                                              true, Predicates.<NamespaceDescriptor>alwaysTrue(),
+                                              new File("compiler/testData/builtin-classes.txt"));
+
+    }
+
+    private void addJetFilesFromDir(List<JetFile> files, File jetDir) throws IOException {
+        for (File file : jetDir.listFiles()) {
+            if (FileUtil.getExtension(file.getName()).equals("jet")) {
+                files.add(JetPsiFactory.createFile(project, file.getName(), FileUtil.loadFile(file, true)));
+            }
+        }
+    }
+
+    private class SpecialModuleConfiguration implements ModuleConfiguration {
+        @Override
+        public void addDefaultImports(@NotNull Collection<JetImportDirective> directives) {
+            for (ImportPath defaultJetImport : DefaultModuleConfiguration.DEFAULT_JET_IMPORTS) {
+                directives.add(JetPsiFactory.createImportDirective(project, defaultJetImport));
+            }
+        }
+
+        @Override
+        public void extendNamespaceScope(@NotNull BindingTrace trace,
+                @NotNull NamespaceDescriptor namespaceDescriptor,
+                @NotNull WritableScope namespaceMemberScope) {
+            // DO nothing
+        }
+    }
+}