1.10-unit test

starcwang · Apr 2, 2021 · 8c3f0d4 · 8c3f0d4
1 parent e141452
commit 8c3f0d4
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/build.gradle b/build.gradle
@@ -20,6 +20,7 @@ dependencies {
 }
 
 // See https://github.com/JetBrains/gradle-intellij-plugin/
+// all versions https://data.services.jetbrains.com/products?fields=code,name,releases.downloads,releases.version,releases.build,releases.type&code=DG,PS,DL,MPS,MPSIIP,RDCPPP,WS,AC,IIC,IIE,GO,CL,PCD,PCC,RD,PCE,IIU,RM,PCP
 intellij {
 //    plugins 'java'
     version '2019.1'

diff --git a/src/main/java/com/star/easygenerate/action/GenerateUnitTestTemplateAction.java b/src/main/java/com/star/easygenerate/action/GenerateUnitTestTemplateAction.java
@@ -9,6 +9,7 @@
 import java.util.stream.Collectors;
 
 import com.google.common.collect.Maps;
+import com.intellij.codeInsight.navigation.NavigationUtil;
 import com.intellij.ide.fileTemplates.FileTemplateUtil;
 import com.intellij.ide.fileTemplates.impl.CustomFileTemplate;
 import com.intellij.openapi.actionSystem.AnAction;
@@ -21,13 +22,17 @@
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.JavaDirectoryService;
+import com.intellij.psi.JavaPsiFacade;
+import com.intellij.psi.PsiClass;
 import com.intellij.psi.PsiDirectory;
 import com.intellij.psi.PsiField;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.PsiJavaFile;
 import com.intellij.psi.PsiMethod;
 import com.intellij.psi.PsiModifier;
 import com.intellij.psi.PsiPackage;
+import com.intellij.psi.impl.source.PsiJavaFileImpl;
+import com.intellij.psi.search.GlobalSearchScope;
 import com.intellij.refactoring.util.classMembers.MemberInfo;
 import com.intellij.util.ResourceUtil;
 import com.star.easygenerate.dialog.CreateUnitTestDialog;
@@ -83,6 +88,14 @@ public void actionPerformed(AnActionEvent e) {
         } catch (Exception exception) {
             LOGGER.error("create template error:", exception);
         }
+
+        // 跳转到类上
+        String tgtClassName = ((PsiJavaFileImpl)dialog.getTargetClass().getParent()).getPackageName() + "." + dialog.getClassName();
+        PsiClass testClass = JavaPsiFacade.getInstance(project).findClass(tgtClassName, GlobalSearchScope.allScope(project));
+        if (testClass == null) {
+            return;
+        }
+        NavigationUtil.activateFileWithPsiElement(testClass, true);
     }
 
     /**
@@ -162,7 +175,7 @@ private Map<String, Object> getParamMap(PsiJavaFile psiJavaFile, CreateUnitTestD
         params.put("user", System.getProperty("user.name"));
         params.put("date", LocalDate.now().format(DateTimeFormatter.ISO_LOCAL_DATE));
         params.put("className", className);
-        params.put("package", dialog.getTargetDirectory().getName());
+        params.put("package", ((PsiJavaFileImpl)dialog.getTargetClass().getParent()).getPackageName());
         params.put("instanceName", StringUtils.substring(className, 0, 1).toLowerCase() + StringUtils.substring(className, 1));
         params.put("testClassName", dialog.getClassName());
         params.put("fieldList", fields);

diff --git a/src/main/resources/template/unit-test.vm b/src/main/resources/template/unit-test.vm
@@ -10,7 +10,7 @@ import ${field.packageName};
 #end
 
 /**
- * {@link ${package}.${className}}的单元测试类
+ * {@link ${className}}的单元测试类
  *
  * @author ${user}
  * @date ${date}