增加单元测试覆盖.

mybatis-plus-generator/src/main/java/com/baomidou/mybatisplus/generator/config/PackageConfig.java

@@ -129,6 +129,7 @@ public class PackageConfig {
      * @deprecated 3.5.10
      */
     @NotNull
+    @Deprecated
     public Map<String, String> getPackageInfo() {
         return getPackageInfo((InjectionConfig) null);
     }
@@ -174,6 +175,7 @@ public class PackageConfig {
      *
      * @param module 模块
      * @return 配置信息
+     * @see #getPackageInfo(InjectionConfig, String)
      * @since 3.5.0
      */
     @Deprecated

mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/generator/config/PackageConfigTest.java

@@ -1,10 +1,12 @@
 package com.baomidou.mybatisplus.generator.config;
 
+import com.baomidou.mybatisplus.generator.config.builder.CustomFile;
 import com.baomidou.mybatisplus.generator.config.builder.GeneratorBuilder;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
 
 import java.util.Collections;
+import java.util.Map;
 
 /**
  * @author nieqiurong 2020/10/6.
@@ -38,4 +40,27 @@ public class PackageConfigTest {
             .mapper("dao").service("iservice").serviceImpl("serviceIm")
             .pathInfo(Collections.singletonMap(OutputFile.controller,"bbbb")).build());
     }
+
+    @Test
+    void testCustomFile() {
+        var packageConfig = GeneratorBuilder.packageConfigBuilder().build();
+        var injectionConfig = GeneratorBuilder.injectionConfigBuilder()
+            .customFile(new CustomFile.Builder()
+                .fileName("Dto.java").packageName("dto").templatePath("dto.vm")
+                .build())
+            .customFile(new CustomFile.Builder()
+                .fileName("Vo.java").templatePath("vo.vm")
+                .build())
+            .customFile(new CustomFile.Builder()
+                .fileName("Bo.java").packageName("com.baomidou.bo").templatePath("Bo.vm")
+                .build())
+            .build();
+        Map<String, String> packageInfo = packageConfig.getPackageInfo(injectionConfig);
+        Assertions.assertNotNull(packageInfo.get("Dto"));
+        Assertions.assertEquals("com.baomidou.dto", packageInfo.get("Dto"));
+        Assertions.assertNull(packageInfo.get("Vo"));
+        Assertions.assertNotNull(packageInfo.get("Bo"));
+        Assertions.assertEquals("com.baomidou.bo", packageInfo.get("Bo"));
+    }
+
 }