fix: fix super class lost issue

Author: phodal

java/src/main/kotlin/cc/unitmesh/idea/service/JavaRelatedContext.kt

@@ -65,10 +65,10 @@ object JavaRelatedContext {
     private fun findSuperClasses(psiClass: PsiClass): List<PsiClass> {
         val superClass = psiClass.superClass ?: return emptyList()
         if (isProjectContent(superClass)) {
-            return listOf(psiClass.superClass!!)
+            return listOf(psiClass.superClass!!, psiClass)
         }
 
-        return emptyList()
+        return listOf(psiClass)
     }
 
     private fun canBeRemoved(member: PsiMember): Boolean {

java/src/main/kotlin/cc/unitmesh/idea/service/JavaWriteTestService.kt

@@ -105,7 +105,8 @@ class JavaWriteTestService : WriteTestService() {
                 }
 
                 is PsiMethod -> {
-                    JavaRelatedContext.findRelatedClassesAndCleanUp(element).forEach { psiClass ->
+                    val findRelatedClassesAndCleanUp = JavaRelatedContext.findRelatedClassesAndCleanUp(element)
+                    findRelatedClassesAndCleanUp.forEach { psiClass ->
                         resolvedClasses[psiClass.name!!] = psiClass
                     }
                 }