浏览代码

fix: github issues/5307 新增文件覆盖测试用例

hubin 2 年之前
父节点
当前提交
7f3a9e4760

+ 19 - 0
mybatis-plus-generator/src/test/java/com/baomidou/mybatisplus/generator/samples/H2CodeGeneratorTest.java

@@ -171,6 +171,25 @@ public class H2CodeGeneratorTest extends BaseGeneratorTest {
         generator.execute();
     }
 
+    /**
+     * 测试文件覆盖
+     */
+    @Test
+    public void testFileOverride() {
+        AutoGenerator generator = new AutoGenerator(DATA_SOURCE_CONFIG);
+        generator.strategy(strategyConfig()
+            // 实体文件覆盖
+            .entityBuilder().enableFileOverride()
+            // Mapper文件覆盖
+            .mapperBuilder().enableFileOverride()
+            // Service文件覆盖
+            .serviceBuilder().enableFileOverride()
+            // Controller文件覆盖
+            .controllerBuilder().enableFileOverride()
+            .build());
+        generator.execute();
+    }
+
     /**
      * 测试日期类型
      */