图像透明处理
This commit is contained in:
spllzh
234
src/test/java/com/rj/service/FaceDetectImageCountTest.java
Normal file
234
src/test/java/com/rj/service/FaceDetectImageCountTest.java
Normal file
@@ -0,0 +1,234 @@
|
||||
package com.rj.service;
|
||||
|
||||
import com.rj.entity.FaceDetectLog;
|
||||
import com.rj.service.impl.FaceDetectLogServiceImpl;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
import org.springframework.test.context.ActiveProfiles;
|
||||
|
||||
/**
|
||||
* 图像检测图片数量字段测试
|
||||
*/
|
||||
@SpringBootTest
|
||||
@ActiveProfiles("test")
|
||||
public class FaceDetectImageCountTest {
|
||||
|
||||
@Autowired
|
||||
private FaceDetectLogServiceImpl faceDetectLogService;
|
||||
|
||||
/**
|
||||
* 测试设置图片数量
|
||||
*/
|
||||
@Test
|
||||
public void testSetImageCount() {
|
||||
System.out.println("=== 测试设置图片数量 ===");
|
||||
|
||||
// 创建测试数据
|
||||
FaceDetectLog faceDetectLog = new FaceDetectLog();
|
||||
faceDetectLog.setId(java.util.UUID.randomUUID().toString());
|
||||
faceDetectLog.setRequestId(java.util.UUID.randomUUID().toString());
|
||||
faceDetectLog.setModel("liveportrait-detect");
|
||||
faceDetectLog.setImageUrl("https://example.com/test-image.jpg");
|
||||
faceDetectLog.setOwnerName("张三");
|
||||
faceDetectLog.setOwnerPhone("13800138000");
|
||||
faceDetectLog.setAvatarName("测试头像");
|
||||
faceDetectLog.setRequestTime(java.time.LocalDateTime.now());
|
||||
faceDetectLog.setResponseTime(java.time.LocalDateTime.now());
|
||||
faceDetectLog.setStatusCode(200);
|
||||
faceDetectLog.setSuccess(true);
|
||||
faceDetectLog.setFaceCount(1);
|
||||
|
||||
// 设置图片数量
|
||||
Integer imageCount = 1;
|
||||
|
||||
// 保存到数据库
|
||||
boolean saved = faceDetectLogService.saveFaceDetectLog(faceDetectLog);
|
||||
|
||||
if (saved) {
|
||||
System.out.println("✅ 图像检测日志保存成功");
|
||||
} else {
|
||||
System.out.println("❌ 图像检测日志保存失败");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试解析JSON中的image_count
|
||||
*/
|
||||
@Test
|
||||
public void testParseImageCountFromJson() {
|
||||
System.out.println("=== 测试解析JSON中的image_count ===");
|
||||
|
||||
// 模拟API响应JSON
|
||||
String jsonResponse = """
|
||||
{
|
||||
"output": {
|
||||
"message": "",
|
||||
"pass": true
|
||||
},
|
||||
"usage": {
|
||||
"image_count": 1
|
||||
},
|
||||
"request_id": "a4529f04-2fbb-4818-8e6c-3627945c95fc"
|
||||
}
|
||||
""";
|
||||
|
||||
System.out.println("模拟API响应: " + jsonResponse);
|
||||
|
||||
// 验证JSON解析逻辑
|
||||
try {
|
||||
com.fasterxml.jackson.databind.ObjectMapper objectMapper = new com.fasterxml.jackson.databind.ObjectMapper();
|
||||
com.fasterxml.jackson.databind.JsonNode jsonNode = objectMapper.readTree(jsonResponse);
|
||||
|
||||
if (jsonNode.has("usage")) {
|
||||
com.fasterxml.jackson.databind.JsonNode usage = jsonNode.get("usage");
|
||||
if (usage.has("image_count")) {
|
||||
int imageCount = usage.get("image_count").asInt();
|
||||
System.out.println("✅ 成功解析image_count: " + imageCount);
|
||||
|
||||
// 验证值是否正确
|
||||
if (imageCount == 1) {
|
||||
System.out.println("✅ image_count值正确");
|
||||
} else {
|
||||
System.out.println("❌ image_count值错误: " + imageCount);
|
||||
}
|
||||
} else {
|
||||
System.out.println("❌ usage中没有image_count字段");
|
||||
}
|
||||
} else {
|
||||
System.out.println("❌ JSON中没有usage字段");
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("❌ JSON解析异常: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试不同image_count值
|
||||
*/
|
||||
@Test
|
||||
public void testDifferentImageCountValues() {
|
||||
System.out.println("=== 测试不同image_count值 ===");
|
||||
|
||||
Integer[] testValues = {1, 2, 3, 0, null};
|
||||
|
||||
for (Integer imageCount : testValues) {
|
||||
System.out.println("测试image_count值: " + imageCount);
|
||||
|
||||
FaceDetectLog faceDetectLog = new FaceDetectLog();
|
||||
faceDetectLog.setId(java.util.UUID.randomUUID().toString());
|
||||
faceDetectLog.setRequestId(java.util.UUID.randomUUID().toString());
|
||||
faceDetectLog.setModel("liveportrait-detect");
|
||||
faceDetectLog.setImageUrl("https://example.com/test-image.jpg");
|
||||
faceDetectLog.setRequestTime(java.time.LocalDateTime.now());
|
||||
faceDetectLog.setResponseTime(java.time.LocalDateTime.now());
|
||||
faceDetectLog.setStatusCode(200);
|
||||
faceDetectLog.setSuccess(true);
|
||||
faceDetectLog.setFaceCount(1);
|
||||
|
||||
|
||||
System.out.println("---");
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 测试完整的API响应解析流程
|
||||
*/
|
||||
@Test
|
||||
public void testCompleteApiResponseParsing() {
|
||||
System.out.println("=== 测试完整的API响应解析流程 ===");
|
||||
|
||||
// 模拟完整的API响应
|
||||
String jsonResponse = """
|
||||
{
|
||||
"output": {
|
||||
"message": "检测成功",
|
||||
"pass": true
|
||||
},
|
||||
"usage": {
|
||||
"image_count": 1
|
||||
},
|
||||
"request_id": "a4529f04-2fbb-4818-8e6c-3627945c95fc"
|
||||
}
|
||||
""";
|
||||
|
||||
System.out.println("完整API响应: " + jsonResponse);
|
||||
|
||||
try {
|
||||
com.fasterxml.jackson.databind.ObjectMapper objectMapper = new com.fasterxml.jackson.databind.ObjectMapper();
|
||||
com.fasterxml.jackson.databind.JsonNode jsonNode = objectMapper.readTree(jsonResponse);
|
||||
|
||||
// 模拟解析逻辑
|
||||
FaceDetectLog faceDetectLog = new FaceDetectLog();
|
||||
|
||||
// 解析output
|
||||
if (jsonNode.has("output")) {
|
||||
com.fasterxml.jackson.databind.JsonNode output = jsonNode.get("output");
|
||||
|
||||
if (output.has("pass")) {
|
||||
boolean pass = output.get("pass").asBoolean();
|
||||
int faceCount = pass ? 1 : 0;
|
||||
faceDetectLog.setFaceCount(faceCount);
|
||||
System.out.println("✅ 解析pass字段: " + pass + ", 图像数量: " + faceCount);
|
||||
}
|
||||
|
||||
if (output.has("message")) {
|
||||
String message = output.get("message").asText();
|
||||
System.out.println("✅ 解析message字段: " + message);
|
||||
}
|
||||
}
|
||||
|
||||
// 解析usage
|
||||
if (jsonNode.has("usage")) {
|
||||
com.fasterxml.jackson.databind.JsonNode usage = jsonNode.get("usage");
|
||||
|
||||
if (usage.has("image_count")) {
|
||||
int imageCount = usage.get("image_count").asInt();
|
||||
System.out.println("✅ 解析image_count字段: " + imageCount);
|
||||
}
|
||||
}
|
||||
|
||||
// 解析request_id
|
||||
if (jsonNode.has("request_id")) {
|
||||
String requestId = jsonNode.get("request_id").asText();
|
||||
System.out.println("✅ 解析request_id字段: " + requestId);
|
||||
}
|
||||
|
||||
System.out.println("✅ 完整API响应解析成功");
|
||||
System.out.println("最终结果:");
|
||||
System.out.println("- 图像数量: " + faceDetectLog.getFaceCount());
|
||||
|
||||
} catch (Exception e) {
|
||||
System.out.println("❌ 解析异常: " + e.getMessage());
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user