Renamed some generated tests to follow conventions.

2013-12-18 20:47:35 +04:00
parent 7c4675a9c0
commit cd8f5fae54
23 changed files with 7158 additions and 152 deletions
@@ -0,0 +1,134 @@
+/*
+ * Copyright 2010-2013 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.resolve;
+
+import com.google.common.collect.Lists;
+import com.google.common.collect.Ordering;
+import com.intellij.ide.startup.impl.StartupManagerImpl;
+import com.intellij.openapi.startup.StartupManager;
+import com.intellij.psi.PsiElement;
+import com.intellij.psi.PsiPolyVariantReference;
+import com.intellij.psi.PsiReference;
+import com.intellij.psi.ResolveResult;
+import com.intellij.testFramework.LightProjectDescriptor;
+import com.intellij.testFramework.fixtures.LightPlatformCodeInsightFixtureTestCase;
+import junit.framework.Assert;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.InTextDirectivesUtils;
+import org.jetbrains.jet.plugin.JetWithJdkAndRuntimeLightProjectDescriptor;
+import org.jetbrains.jet.testing.ReferenceUtils;
+
+import java.util.List;
+
+public abstract class AbstractReferenceResolveTest extends LightPlatformCodeInsightFixtureTestCase {
+
+    public static final String MULTIRESOLVE = "MULTIRESOLVE";
+    public static final String REF_EMPTY = "REF_EMPTY";
+
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+        ((StartupManagerImpl) StartupManager.getInstance(getProject())).runPostStartupActivities();
+    }
+
+    protected void doTest(String path) {
+        myFixture.configureByFile(path);
+
+        if (InTextDirectivesUtils.isDirectiveDefined(myFixture.getFile().getText(), MULTIRESOLVE)) {
+            doMultiResolveTest();
+        }
+        else {
+            doSingleResolveTest();
+        }
+    }
+
+    protected void doSingleResolveTest() {
+        boolean shouldBeUnresolved = InTextDirectivesUtils.isDirectiveDefined(myFixture.getFile().getText(), REF_EMPTY);
+        List<String> refs = InTextDirectivesUtils.findLinesWithPrefixesRemoved(myFixture.getFile().getText(), "REF:");
+
+        String referenceToString;
+        if (shouldBeUnresolved) {
+            Assert.assertTrue("REF: directives will be ignored for " + REF_EMPTY + " test: " + refs, refs.isEmpty());
+            referenceToString = "<empty>";
+        }
+        else {
+            assertTrue("Must be a single ref: " + refs + ".\n" +
+                       "Use " + MULTIRESOLVE + " if you need multiple refs\n" +
+                       "Use "+ REF_EMPTY + " for an unresolved reference",
+                       refs.size() == 1);
+            referenceToString = refs.get(0);
+            Assert.assertNotNull("Test data wasn't found, use \"// REF: \" directive", referenceToString);
+        }
+
+        int offset = myFixture.getEditor().getCaretModel().getOffset();
+        PsiReference psiReference = myFixture.getFile().findReferenceAt(offset);
+        if (psiReference != null) {
+            PsiElement resolvedTo = psiReference.resolve();
+            if (resolvedTo != null) {
+                String resolvedToElementStr = ReferenceUtils.renderAsGotoImplementation(resolvedTo);
+                String notEqualMessage = String.format("Found reference to '%s', but '%s' was expected",
+                                                       resolvedToElementStr, referenceToString);
+                assertEquals(notEqualMessage, referenceToString, resolvedToElementStr);
+            }
+            else {
+                if (!shouldBeUnresolved) {
+                    Assert.assertNull(
+                            String.format("Element %s wasn't resolved to anything, but %s was expected", psiReference, referenceToString),
+                            referenceToString);
+                }
+            }
+        }
+        else {
+            Assert.assertNull(String.format("No reference found at offset: %s, but one resolved to %s was expected", offset, referenceToString),
+                              referenceToString);
+        }
+    }
+
+    protected void doMultiResolveTest() {
+        List<String> expectedReferences = InTextDirectivesUtils.findLinesWithPrefixesRemoved(myFixture.getFile().getText(), "REF:");
+
+        PsiReference psiReference =
+                myFixture.getFile().findReferenceAt(myFixture.getEditor().getCaretModel().getOffset());
+
+        assertTrue(psiReference instanceof PsiPolyVariantReference);
+
+        PsiPolyVariantReference variantReference = (PsiPolyVariantReference) psiReference;
+
+        ResolveResult[] results = variantReference.multiResolve(true);
+
+        List<String> actualResolvedTo = Lists.newArrayList();
+        for (ResolveResult result : results) {
+            PsiElement resolvedToElement = result.getElement();
+            assertNotNull(resolvedToElement);
+
+            actualResolvedTo.add(ReferenceUtils.renderAsGotoImplementation(resolvedToElement));
+        }
+
+        assertOrderedEquals(Ordering.natural().sortedCopy(actualResolvedTo), Ordering.natural().sortedCopy(expectedReferences));
+    }
+
+    @Override
+    protected LightProjectDescriptor getProjectDescriptor() {
+        return JetWithJdkAndRuntimeLightProjectDescriptor.INSTANCE;
+    }
+
+    @NotNull
+    @Override
+    protected String getTestDataPath() {
+        return "./";
+    }
+}