feat: update task routing and coze handling

2026-06-03 17:14:50 +08:00
parent ea37d82d73
commit e98a1a1207
37 changed files with 1883 additions and 254 deletions
@@ -0,0 +1,64 @@
+package com.nanri.aiimage.modules.appearancepatent.client;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.nanri.aiimage.config.AppearancePatentProperties;
+import com.nanri.aiimage.config.BrandCheckProperties;
+import com.nanri.aiimage.modules.brand.client.BrandCheckClient;
+import com.nanri.aiimage.modules.appearancepatent.model.dto.AppearancePatentResultRowDto;
+import org.junit.jupiter.api.Test;
+
+import java.util.List;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.never;
+import static org.mockito.Mockito.verify;
+
+class AppearancePatentCozeClientTest {
+
+    private final AppearancePatentCozeClient client = new AppearancePatentCozeClient(
+            new AppearancePatentProperties(),
+            new ObjectMapper(),
+            null,
+            new BrandCheckClient(new BrandCheckProperties())
+    );
+
+    @Test
+    void titleRiskIsInfringementWhenBrandCheckHasFailedData() {
+        BrandCheckClient.BrandCheckBatchResult result =
+                new BrandCheckClient.BrandCheckBatchResult(List.of("阿凡达"), List.of("阿凡达"), List.of());
+
+        assertThat(client.buildTitleRisk("阿凡达", result)).isEqualTo("侵权");
+    }
+
+    @Test
+    void conclusionIsInfringementWhenTitleOrAppearanceIsInfringement() {
+        assertThat(client.buildConclusion("侵权", "无侵权", "")).isEqualTo("侵权");
+        assertThat(client.buildConclusion("无侵权", "侵权", "")).isEqualTo("侵权");
+        assertThat(client.buildConclusion("无侵权", "无侵权", "")).isEqualTo("无侵权");
+    }
+
+    @Test
+    void skipBrandCheckWhenCozeTitleAndReasonAreNone() throws Exception {
+        BrandCheckClient brandCheckClient = mock(BrandCheckClient.class);
+        AppearancePatentCozeClient cozeClient = new AppearancePatentCozeClient(
+                new AppearancePatentProperties(),
+                new ObjectMapper(),
+                null,
+                brandCheckClient
+        );
+        AppearancePatentResultRowDto row = new AppearancePatentResultRowDto();
+        row.setId("1");
+        String dataText = """
+                {"data":[{"row_id":"1","title":"无","title_reason":"无","appearance":"无侵权","result":"无侵权"}]}
+                """;
+
+        List<AppearancePatentResultRowDto> merged = cozeClient.mergeRowsFromDataText(List.of(row), dataText);
+
+        assertThat(merged).hasSize(1);
+        assertThat(merged.get(0).getTitleRisk()).isEqualTo("无侵权");
+        assertThat(merged.get(0).getConclusion()).isEqualTo("无侵权");
+        verify(brandCheckClient, never()).checkTitleText(anyString(), anyString());
+    }
+}