Unit Test

Author: moshowgame

pom.xml

@@ -190,6 +190,20 @@
 				<artifactId>spring-boot-maven-plugin</artifactId>
 			</plugin>
 
+			<!-- Maven Surefire Plugin for JUnit 5 -->
+			<plugin>
+				<groupId>org.apache.maven.plugins</groupId>
+				<artifactId>maven-surefire-plugin</artifactId>
+				<version>3.5.2</version>
+				<configuration>
+					<useSystemClassLoader>false</useSystemClassLoader>
+					<includes>
+						<include>**/*Test.java</include>
+						<include>**/*Tests.java</include>
+					</includes>
+				</configuration>
+			</plugin>
+			
 			<!-- JaCoCo Maven Plugin for Code Coverage -->
 			<plugin>
 				<groupId>org.jacoco</groupId>

src/main/java/com/softdev/system/generator/entity/enums/ParserTypeEnum.java

@@ -27,11 +27,33 @@ public enum ParserTypeEnum {
     }
 
     public static ParserTypeEnum fromValue(String value) {
+        if (value == null || value.trim().isEmpty()) {
+            return SQL;
+        }
+        
+        String trimmedValue = value.trim();
+        
+        // 首先尝试精确匹配枚举值
+        for (ParserTypeEnum type : ParserTypeEnum.values()) {
+            if (type.getValue().equals(trimmedValue)) {
+                return type;
+            }
+        }
+        
+        // 如果精确匹配失败，尝试忽略大小写匹配
+        for (ParserTypeEnum type : ParserTypeEnum.values()) {
+            if (type.getValue().equalsIgnoreCase(trimmedValue)) {
+                return type;
+            }
+        }
+        
+        // 尝试匹配枚举名称
         for (ParserTypeEnum type : ParserTypeEnum.values()) {
-            if (type.getValue().equals(value)) {
+            if (type.name().equalsIgnoreCase(trimmedValue)) {
                 return type;
             }
         }
+        
         // 默认返回SQL类型
         return SQL;
     }

src/main/java/com/softdev/system/generator/service/impl/CodeGenServiceImpl.java

@@ -92,6 +92,9 @@ public Map<String, String> getResultByParams(Map<String, Object> params) throws
     private ClassInfo parseTableStructure(ParamInfo paramInfo) throws Exception {
         String dataType = MapUtil.getString(paramInfo.getOptions(), "dataType");
         ParserTypeEnum parserType = ParserTypeEnum.fromValue(dataType);
+        
+        // 添加调试信息
+        log.debug("解析数据类型: {}, 解析结果: {}", dataType, parserType);
 
         switch (parserType) {
             case SQL:

src/test/java/com/softdev/system/generator/controller/CodeGenControllerTest.java

@@ -47,7 +47,15 @@ class CodeGenControllerTest {
     void setUp() {
         // 初始化测试数据
         paramInfo = new ParamInfo();
-        paramInfo.setTableSql("CREATE TABLE test (id INT PRIMARY KEY, name VARCHAR(50));");
+        paramInfo.setTableSql("""
+                CREATE TABLE 'sys_user_info' (
+                  'user_id' int(11) NOT NULL AUTO_INCREMENT COMMENT '用户编号',
+                  'user_name' varchar(255) NOT NULL COMMENT '用户名',
+                  'status' tinyint(1) NOT NULL COMMENT '状态',
+                  'create_time' datetime NOT NULL COMMENT '创建时间',
+                  PRIMARY KEY ('user_id')
+                ) ENGINE=InnoDB DEFAULT CHARSET=utf8 COMMENT='用户信息'
+                """);
 
         Map<String, Object> options = new HashMap<>();
         options.put("dataType", "SQL");

src/test/java/com/softdev/system/generator/entity/ParamInfoTest.java

@@ -48,8 +48,8 @@ void testSetOptions() {
 
     @Test
     void testNameCaseTypeConstants() {
-        assertEquals("camelCase", ParamInfo.NAME_CASE_TYPE.CAMEL_CASE);
-        assertEquals("underScoreCase", ParamInfo.NAME_CASE_TYPE.UNDER_SCORE_CASE);
-        assertEquals("upperUnderScoreCase", ParamInfo.NAME_CASE_TYPE.UPPER_UNDER_SCORE_CASE);
+        assertEquals("CamelCase", ParamInfo.NAME_CASE_TYPE.CAMEL_CASE);
+        assertEquals("UnderScoreCase", ParamInfo.NAME_CASE_TYPE.UNDER_SCORE_CASE);
+        assertEquals("UpperUnderScoreCase", ParamInfo.NAME_CASE_TYPE.UPPER_UNDER_SCORE_CASE);
     }
 }