Enable Type Hierarchy for Kotlin classes referenced by non-Kotlin code #KT-4414 Fixed

idea/src/org/jetbrains/jet/plugin/hierarchy/KotlinTypeHierarchyProvider.java

@@ -32,7 +32,6 @@ import com.intellij.psi.util.PsiTreeUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.jet.lang.descriptors.FunctionDescriptor;
 import org.jetbrains.jet.lang.psi.JetClassOrObject;
-import org.jetbrains.jet.lang.psi.JetFile;
 import org.jetbrains.jet.lang.psi.JetNamedFunction;
 import org.jetbrains.jet.lang.resolve.BindingContext;
 import org.jetbrains.jet.lang.types.JetType;
@@ -45,7 +44,6 @@ import org.jetbrains.jet.renderer.DescriptorRenderer;
 import java.util.Collection;
 
 public class KotlinTypeHierarchyProvider extends JavaTypeHierarchyProvider {
-
     @Override
     public PsiElement getTarget(@NotNull DataContext dataContext) {
         Project project = PlatformDataKeys.PROJECT.getData(dataContext);
@@ -54,7 +52,7 @@ public class KotlinTypeHierarchyProvider extends JavaTypeHierarchyProvider {
         Editor editor = PlatformDataKeys.EDITOR.getData(dataContext);
         if (editor != null) {
             PsiFile file = PsiDocumentManager.getInstance(project).getPsiFile(editor.getDocument());
-            if (!(file instanceof JetFile)) return null;
+            if (file == null) return null;
 
             if (!JetPluginUtil.isInSource(file)) return null;
             if (JetPluginUtil.isKtFileInGradleProjectInWrongFolder(file)) return null;

idea/testData/hierarchy/class/type/JavaFromKotlinByRef/1_MyClass.java

@@ -0,0 +1,3 @@
+public class MyClass extends <caret>KotlinClass {
+
+}

idea/testData/hierarchy/class/type/JavaFromKotlinByRef/2_main.kt

@@ -0,0 +1 @@
+public open class KotlinClass

idea/testData/hierarchy/class/type/JavaFromKotlinByRef/JavaFromKotlinByRef_verification.xml

@@ -0,0 +1,5 @@
+<node text="Object (java.lang)">
+  <node text="KotlinClass ()" base="true">
+    <node text="MyClass ()"/>
+  </node>
+</node>

idea/testData/hierarchy/class/type/KotlinFromJavaByRef/1_JavaClass.java

@@ -0,0 +1,3 @@
+public class JavaClass {
+
+}

idea/testData/hierarchy/class/type/KotlinFromJavaByRef/2_main.kt

@@ -0,0 +1 @@
+public open class KotlinClass: <caret>JavaClass()

idea/testData/hierarchy/class/type/KotlinFromJavaByRef/KotlinFromJavaByRef_verification.xml

@@ -0,0 +1,5 @@
+<node text="Object (java.lang)">
+  <node text="JavaClass ()" base="true">
+    <node text="KotlinClass ()"/>
+  </node>
+</node>

idea/testData/hierarchy/class/type/KotlinFromKotlinByRef/1_main.kt

@@ -0,0 +1 @@
+public open class BaseClass

idea/testData/hierarchy/class/type/KotlinFromKotlinByRef/2_main.kt

@@ -0,0 +1 @@
+public open class DerivedClass: <caret>BaseClass()

idea/testData/hierarchy/class/type/KotlinFromKotlinByRef/KotlinFromKotlinByRef_verification.xml

@@ -0,0 +1,5 @@
+<node text="Object (java.lang)">
+  <node text="BaseClass ()" base="true">
+    <node text="DerivedClass ()"/>
+  </node>
+</node>

idea/tests/org/jetbrains/jet/plugin/hierarchy/HierarchyTestGenerated.java

@@ -128,6 +128,11 @@ public class HierarchyTestGenerated extends AbstractHierarchyTest {
             doTypeClassHierarchyTest("idea/testData/hierarchy/class/type/JavaFromKotlin");
         }
         
+        @TestMetadata("JavaFromKotlinByRef")
+        public void testJavaFromKotlinByRef() throws Exception {
+            doTypeClassHierarchyTest("idea/testData/hierarchy/class/type/JavaFromKotlinByRef");
+        }
+        
         @TestMetadata("JavaFromKotlinForKotlinClass")
         public void testJavaFromKotlinForKotlinClass() throws Exception {
             doTypeClassHierarchyTest("idea/testData/hierarchy/class/type/JavaFromKotlinForKotlinClass");
@@ -148,6 +153,16 @@ public class HierarchyTestGenerated extends AbstractHierarchyTest {
             doTypeClassHierarchyTest("idea/testData/hierarchy/class/type/KotlinFromJava");
         }
         
+        @TestMetadata("KotlinFromJavaByRef")
+        public void testKotlinFromJavaByRef() throws Exception {
+            doTypeClassHierarchyTest("idea/testData/hierarchy/class/type/KotlinFromJavaByRef");
+        }
+        
+        @TestMetadata("KotlinFromKotlinByRef")
+        public void testKotlinFromKotlinByRef() throws Exception {
+            doTypeClassHierarchyTest("idea/testData/hierarchy/class/type/KotlinFromKotlinByRef");
+        }
+        
         @TestMetadata("Object")
         public void testObject() throws Exception {
             doTypeClassHierarchyTest("idea/testData/hierarchy/class/type/Object");