From efdfbbf9bfe4b581b9d7a96ab53f3e49d1e8499c Mon Sep 17 00:00:00 2001
From: Andrey Breslav <abreslav@abreslav-mac.labs.intellij.net>
Date: Mon, 28 Feb 2011 15:24:04 +0300
Subject: [PATCH] TopDown resolve for Int.plus works.
 FunctionDescriptorUtil.getOriginal() added

---
 .../lang/types/FunctionDescriptorUtil.java    |  9 ++++++++
 idea/testData/resolve/Basic.jet               |  2 +-
 .../jetbrains/jet/resolve/JetResolveTest.java | 21 ++++++++++++++++++-
 3 files changed, 30 insertions(+), 2 deletions(-)

diff --git a/idea/src/org/jetbrains/jet/lang/types/FunctionDescriptorUtil.java b/idea/src/org/jetbrains/jet/lang/types/FunctionDescriptorUtil.java
index 1c5540acbb6..9f1a0fa43e6 100644
--- a/idea/src/org/jetbrains/jet/lang/types/FunctionDescriptorUtil.java
+++ b/idea/src/org/jetbrains/jet/lang/types/FunctionDescriptorUtil.java
@@ -77,6 +77,9 @@ public class FunctionDescriptorUtil {
 
     @NotNull
     public static FunctionDescriptor substituteFunctionDescriptor(@NotNull List<Type> typeArguments, @NotNull FunctionDescriptor functionDescriptor) {
+        if (functionDescriptor.getTypeParameters().isEmpty()) {
+            return functionDescriptor;
+        }
         return new FunctionDescriptorImpl(
                 ResolveUtil.getJetFunction(functionDescriptor),
                 functionDescriptor,
@@ -89,4 +92,10 @@ public class FunctionDescriptorUtil {
         );
     }
 
+    public static FunctionDescriptor getOriginal(FunctionDescriptor descriptor) {
+        while (descriptor.getOriginal() != null) {
+            descriptor = descriptor.getOriginal();
+        }
+        return descriptor;
+    }
 }
diff --git a/idea/testData/resolve/Basic.jet b/idea/testData/resolve/Basic.jet
index 8bd26115014..0190967038f 100644
--- a/idea/testData/resolve/Basic.jet
+++ b/idea/testData/resolve/Basic.jet
@@ -5,7 +5,7 @@ class A {
 
   fun foo(a : Int) = a
   fun fooB() = foo(1)
-  fun foo() : Int = 1
+  fun foo() : Int = 1.plus(1)
   fun foo1() : B = new B()
 }
 
diff --git a/idea/tests/org/jetbrains/jet/resolve/JetResolveTest.java b/idea/tests/org/jetbrains/jet/resolve/JetResolveTest.java
index 6d71ff5e1f3..69a8fa65637 100644
--- a/idea/tests/org/jetbrains/jet/resolve/JetResolveTest.java
+++ b/idea/tests/org/jetbrains/jet/resolve/JetResolveTest.java
@@ -10,6 +10,7 @@ import org.jetbrains.jet.parsing.JetParsingTest;
 
 import java.io.File;
 import java.io.FileReader;
+import java.util.Collection;
 import java.util.Collections;
 import java.util.List;
 
@@ -54,11 +55,29 @@ public class JetResolveTest extends LightDaemonAnalyzerTestCase {
         DeclarationDescriptor resolve = bindingContext.resolve((JetReferenceExpression) fooDecl.getBodyExpression());
         assertSame(bindingContext.getFunctionDescriptor(fooDecl).getUnsubstitutedValueParameters().get(0), resolve);
 
+        {
         JetFunction fooBDecl = (JetFunction) classADecl.getDeclarations().get(2);
         JetCallExpression fooBBody = (JetCallExpression) fooBDecl.getBodyExpression();
         JetReferenceExpression refToFoo = (JetReferenceExpression) fooBBody.getCalleeExpression();
         FunctionDescriptor mustBeFoo = (FunctionDescriptor) bindingContext.resolve(refToFoo);
-        assertSame(bindingContext.getFunctionDescriptor(fooDecl), mustBeFoo.getOriginal());
+        assertSame(bindingContext.getFunctionDescriptor(fooDecl), FunctionDescriptorUtil.getOriginal(mustBeFoo));
+        }
+
+        {
+            JetFunction fooIntDecl = (JetFunction) classADecl.getDeclarations().get(3);
+            JetCallExpression fooIntBody = (JetCallExpression) fooIntDecl.getBodyExpression();
+            JetDotQualifiedExpression qualifiedPlus = (JetDotQualifiedExpression) fooIntBody.getCalleeExpression();
+            JetReferenceExpression refToPlus = (JetReferenceExpression) qualifiedPlus.getSelectorExpression();
+            FunctionDescriptor mustBePlus = (FunctionDescriptor) bindingContext.resolve(refToPlus);
+            FunctionGroup plusGroup = JetStandardClasses.getInt().getMemberScope(Collections.<TypeProjection>emptyList()).getFunctionGroup("plus");
+            Collection<FunctionDescriptor> pluses = plusGroup.getPossiblyApplicableFunctions(Collections.<Type>emptyList(), Collections.singletonList(JetStandardClasses.getIntType()));
+            FunctionDescriptor intPlus = null;
+            for (FunctionDescriptor plus : pluses) {
+                intPlus = plus;
+            }
+            assertSame(intPlus, FunctionDescriptorUtil.getOriginal(mustBePlus));
+        }
+
 
         JetClass classCDecl = (JetClass) declarations.get(1);
         ClassDescriptor classC = bindingContext.getClassDescriptor(classCDecl);