Prechádzať zdrojové kódy

fix: github issues/5305 & 5281 新增测试用例

hubin 2 rokov pred
rodič
commit
d1fc2252e7

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

@@ -7,6 +7,7 @@ import com.baomidou.mybatisplus.generator.config.OutputFile;
 import com.baomidou.mybatisplus.generator.config.StrategyConfig;
 import com.baomidou.mybatisplus.generator.config.TemplateType;
 import com.baomidou.mybatisplus.generator.config.builder.CustomFile;
+import com.baomidou.mybatisplus.generator.config.rules.DateType;
 import com.baomidou.mybatisplus.generator.fill.Column;
 import com.baomidou.mybatisplus.generator.fill.Property;
 import com.baomidou.mybatisplus.generator.query.DefaultQuery;
@@ -170,6 +171,26 @@ public class H2CodeGeneratorTest extends BaseGeneratorTest {
         generator.execute();
     }
 
+    /**
+     * 测试日期类型
+     */
+    @Test
+    public void testDateType() {
+        AutoGenerator generator = new AutoGenerator(DATA_SOURCE_CONFIG);
+        generator.strategy(strategyConfig().build());
+        // 关闭其它模块生成只生成实体
+        generator.template(templateConfig().disable(
+            TemplateType.CONTROLLER,
+            TemplateType.SERVICE,
+            TemplateType.SERVICE_IMPL,
+            TemplateType.XML,
+            TemplateType.MAPPER
+        ).build());
+        // 日期类型强制设置为 Date 类型
+        generator.global(globalConfig().dateType(DateType.ONLY_DATE).build());
+        generator.execute();
+    }
+
     /**
      * 注入自定义属性
      */