package cn.iocoder.yudao.module.qcreport.service.aiimport.llm;
|
|
import cn.iocoder.yudao.framework.common.exception.ServiceException;
|
import cn.iocoder.yudao.module.qcreport.controller.admin.aiimport.vo.QcReportAiDraftRespVO;
|
import org.junit.jupiter.api.DisplayName;
|
import org.junit.jupiter.api.Test;
|
|
import java.util.List;
|
import java.util.Map;
|
|
import static org.junit.jupiter.api.Assertions.assertEquals;
|
import static org.junit.jupiter.api.Assertions.assertNotNull;
|
import static org.junit.jupiter.api.Assertions.assertNull;
|
import static org.junit.jupiter.api.Assertions.assertThrows;
|
import static org.junit.jupiter.api.Assertions.assertTrue;
|
|
/**
|
* 模型回复的解析。
|
* <p>
|
* 全部喂<b>罐头字符串</b>，不 mock {@code AiChatApi}——这里是「字符串 → 草稿」的纯函数，
|
* mock 掉模型再断言解析正确，测的是 mock 的返回值而不是解析逻辑。
|
* <p>
|
* 罐头用例挑的都是真实会遇到的回复形状：模型习惯把 JSON 包进 markdown 围栏、习惯在前后加一句
|
* 「好的，这是识别结果」，也习惯在输出被截断时留下半个对象。这三种必须在切片这一步就被吃掉，
|
* 否则用户看到的是「AI 返回的内容无法解析」——而其实回复里内容是全的。
|
*/
|
class QcReportAiDraftParserTest {
|
|
@Test
|
@DisplayName("裸 JSON 原样解析")
|
void plainJson() {
|
QcReportAiDraftRespVO resp = QcReportAiDraftParser.parse("""
|
{"summary":"来料检验报告","page":{"size":"A4","orientation":"portrait"},
|
"components":[{"type":"ReportHeader","props":{"title":"来料检验报告"}}]}""");
|
|
assertEquals("来料检验报告", resp.getSummary());
|
assertEquals("A4", resp.getPage().getSize());
|
assertEquals("portrait", resp.getPage().getOrientation());
|
assertEquals(1, resp.getComponents().size());
|
assertEquals("ReportHeader", resp.getComponents().get(0).getType());
|
assertEquals("来料检验报告", resp.getComponents().get(0).getProps().get("title"));
|
}
|
|
@Test
|
@DisplayName("markdown 围栏包裹的 JSON 照样解析（模型最常见的输出形状）")
|
void markdownFenced() {
|
QcReportAiDraftRespVO resp = QcReportAiDraftParser.parse("""
|
```json
|
{"summary":"扫描件","components":[{"type":"Heading","props":{"text":"检验项目"}}]}
|
```""");
|
|
assertEquals(1, resp.getComponents().size());
|
assertEquals("Heading", resp.getComponents().get(0).getType());
|
}
|
|
@Test
|
@DisplayName("前后带寒暄的 JSON 照样解析")
|
void withChatterAround() {
|
QcReportAiDraftRespVO resp = QcReportAiDraftParser.parse("""
|
好的，我已经分析了这份文件。识别结果如下：
|
{"summary":"识别完成","components":[{"type":"Result","props":{"conclusion":"合格"}}]}
|
如果需要调整请告诉我。""");
|
|
assertEquals("识别完成", resp.getSummary());
|
assertEquals("Result", resp.getComponents().get(0).getType());
|
}
|
|
@Test
|
@DisplayName("props 里有嵌套的数组与对象也不炸——props 的合法 key 由前端清单决定，后端不该强类型化")
|
void nestedPropValues() {
|
QcReportAiDraftRespVO resp = QcReportAiDraftParser.parse("""
|
{"components":[{"type":"QualityTable","props":{
|
"itemsPath":"inspectionItems",
|
"columns":[{"key":"size","label":"尺寸"}],
|
"meta":{"source":"iqc"}}}]}""");
|
|
Map<String, Object> props = resp.getComponents().get(0).getProps();
|
assertTrue(props.get("columns") instanceof List, "数组型属性应当原样保留为集合");
|
assertTrue(props.get("meta") instanceof Map, "对象型属性应当原样保留为 Map");
|
}
|
|
@Test
|
@DisplayName("components 被二次编码成字符串时也能读出来")
|
void doubleEncodedComponents() {
|
QcReportAiDraftRespVO resp = QcReportAiDraftParser.parse("""
|
{"summary":"x","components":"[{\\"type\\":\\"Heading\\",\\"props\\":{\\"text\\":\\"标题\\"}}]"}""");
|
|
assertEquals(1, resp.getComponents().size());
|
assertEquals("Heading", resp.getComponents().get(0).getType());
|
}
|
|
@Test
|
@DisplayName("components 为空数组不算解析失败：这是「解析成功但内容为空」，由服务层报更贴切的错")
|
void emptyComponentsIsNotAParseFailure() {
|
QcReportAiDraftRespVO resp = QcReportAiDraftParser.parse("{\"summary\":\"没识别出来\",\"components\":[]}");
|
|
assertNotNull(resp.getComponents());
|
assertTrue(resp.getComponents().isEmpty());
|
}
|
|
@Test
|
@DisplayName("没有 page 字段时为 null，交给设计器沿用当前模板纸张")
|
void missingPageIsNull() {
|
assertNull(QcReportAiDraftParser.parse("{\"components\":[{\"type\":\"X\",\"props\":{}}]}").getPage());
|
// 只给了 size 也算有效，缺的部分让设计器补
|
assertEquals("A3", QcReportAiDraftParser.parse(
|
"{\"page\":{\"size\":\"A3\"},\"components\":[]}").getPage().getSize());
|
}
|
|
@Test
|
@DisplayName("完全不含 JSON 对象、空串、截断成半个对象时统一抛错")
|
void unparseableThrows() {
|
assertThrows(ServiceException.class, () -> QcReportAiDraftParser.parse("我不太明白你的意思"));
|
assertThrows(ServiceException.class, () -> QcReportAiDraftParser.parse(""));
|
assertThrows(ServiceException.class, () -> QcReportAiDraftParser.parse(null));
|
assertThrows(ServiceException.class, () -> QcReportAiDraftParser.parse("{\"components\": ["),
|
"被 max_tokens 截断的回复会在切片后拿到畸形 JSON，必须抛错而不是给出半份草稿");
|
}
|
|
@Test
|
@DisplayName("回填原文按 4000 字符截断，够排查又不灌爆响应体")
|
void rawTextTruncation() {
|
assertNull(QcReportAiDraftParser.truncate(null));
|
assertEquals("abc", QcReportAiDraftParser.truncate("abc"));
|
|
String truncated = QcReportAiDraftParser.truncate("x".repeat(5000));
|
assertTrue(truncated.startsWith("x".repeat(100)));
|
assertTrue(truncated.contains("已截断"));
|
assertTrue(truncated.length() < 4100);
|
}
|
|
}
|
