From eedce2f2ceec5c27b96a9b4f75a799b5b0f01483 Mon Sep 17 00:00:00 2001
From: xiangtianyu <tjjxiang@gmail.com>
Date: Wed, 4 Sep 2024 18:19:47 +0800
Subject: [PATCH] fix test case call wrong method

---
 src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java b/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java
index a7801cf..c023976 100644
--- a/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java
+++ b/src/main/java/com/zhongan/devpilot/util/PsiElementUtils.java
@@ -59,7 +59,7 @@ private static Set<PsiClass> getClassRelatedClass(@NotNull PsiElement element) {
             var fields = psiClass.getFields();
 
             for (PsiMethod psiMethod : methods) {
-                result.addAll(getMethodParameterTypeClass(psiMethod));
+                result.addAll(getMethodRelatedClass(psiMethod));
             }
 
             for (PsiField psiField : fields) {