1: 添加MedicalAiAnswerSetter测试类

pull/1/head
xueqingkun 1 year ago
parent 626e609031
commit 05db4e0771

@ -51,7 +51,8 @@ public class MaterialLibraryManageServiceImpl implements MaterialLibraryManageSe
for (MaterialLibrary materialLibrary : materialLibraryList) { for (MaterialLibrary materialLibrary : materialLibraryList) {
FileResource fileResource = fileResourceMap.get(materialLibrary.getFileResourceId()); FileResource fileResource = fileResourceMap.get(materialLibrary.getFileResourceId());
Assert.notNull(fileResource, "文件资源不存在"); Assert.notNull(fileResource, "文件资源不存在");
materialLibrary.setMaterialName(renameIfNecessary(fileResource.getFileName())); String fileName = StrUtil.isEmpty(materialLibrary.getMaterialName()) ? fileResource.getFileName() : materialLibrary.getMaterialName();
materialLibrary.setMaterialName(renameIfNecessary(fileName));
// 保存素材 // 保存素材
materialLibrary.setFileResourceId(materialLibrary.getFileResourceId()); materialLibrary.setFileResourceId(materialLibrary.getFileResourceId());
materialLibraryService.save(materialLibrary); materialLibraryService.save(materialLibrary);

@ -12,7 +12,9 @@ import lombok.extern.slf4j.Slf4j;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest; import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.transaction.annotation.Transactional;
import java.io.File;
import java.io.FileInputStream; import java.io.FileInputStream;
import java.util.List; import java.util.List;
@ -35,14 +37,18 @@ public class MedicalAiAnswerSetter {
private final String directoryId = "1796467642475745282"; private final String directoryId = "1796467642475745282";
@Test @Test
@Transactional
public void setByFileAndMedical() throws Exception { public void setByFileAndMedical() throws Exception {
String medicalId = ""; // 董婉平
String medicalId = "1768548145559506946";
// 动态视频 // 动态视频
String dynamicFile = "D:\\virtual-patient-manage\\src\\main\\resources\\static\\medical\\" + medicalId + ".txt"; String dynamicFilePath = "F:\\tmp\\数字人\\董婉萍-动态视频-1.mp4";
// 静态视频 // 静态视频
String staticFile = "D:\\virtual-patient-manage\\src\\main\\resources\\static\\medical\\" + medicalId + ".silence.txt"; String staticFilePath = "F:\\tmp\\数字人\\董婉萍静态视频-1.mp4";
File staticFile = new File(staticFilePath);
File dynamicFile = new File(dynamicFilePath);
FileResource staticFileResource = fileManageService.uploadFile(new FileInputStream(staticFile)); FileResource staticFileResource = fileManageService.uploadFile(new FileInputStream(staticFile));
FileResource dynamicFileResource = fileManageService.uploadFile(new FileInputStream(dynamicFile)); FileResource dynamicFileResource = fileManageService.uploadFile(new FileInputStream(dynamicFile));
@ -50,13 +56,13 @@ public class MedicalAiAnswerSetter {
MaterialLibrary staticMaterial = new MaterialLibrary(); MaterialLibrary staticMaterial = new MaterialLibrary();
staticMaterial.setDirectoryId(directoryId); staticMaterial.setDirectoryId(directoryId);
staticMaterial.setMaterialType(1); staticMaterial.setMaterialType(1);
staticMaterial.setMaterialName(staticFileResource.getFileName()); staticMaterial.setMaterialName(staticFile.getName());
staticMaterial.setFileResourceId(staticFileResource.getId()); staticMaterial.setFileResourceId(staticFileResource.getId());
MaterialLibrary dynamicMaterial = new MaterialLibrary(); MaterialLibrary dynamicMaterial = new MaterialLibrary();
dynamicMaterial.setDirectoryId(directoryId); dynamicMaterial.setDirectoryId(directoryId);
dynamicMaterial.setMaterialType(1); dynamicMaterial.setMaterialType(1);
dynamicMaterial.setMaterialName(dynamicFileResource.getFileName()); dynamicMaterial.setMaterialName(dynamicFile.getName());
dynamicMaterial.setFileResourceId(dynamicFileResource.getId()); dynamicMaterial.setFileResourceId(dynamicFileResource.getId());
materialLibraryManageService.uploadMaterial(List.of(staticMaterial,dynamicMaterial)); materialLibraryManageService.uploadMaterial(List.of(staticMaterial,dynamicMaterial));
@ -86,6 +92,6 @@ public class MedicalAiAnswerSetter {
askPatientAnswer.setAnswerResourceId(fileId); askPatientAnswer.setAnswerResourceId(fileId);
askPatientAnswer.setAnswerType(1); askPatientAnswer.setAnswerType(1);
} }
askPatientAnswerService.save(askPatientAnswer); askPatientAnswerService.saveOrUpdate(askPatientAnswer);
} }
} }

Loading…
Cancel
Save