DigitalImgController测试

main
liushujing 3 months ago
parent 32c17f13c8
commit 36e493dbb5

@ -53,7 +53,7 @@ public class ChunksController {
@GetMapping("{id}")
@ApiOperation("信息")
public Result<ChunksDTO> get(@PathVariable("id") Long id){
public Result<ChunksDTO> get(@PathVariable("id") String id){
ChunksDTO data = chunksService.get(id);
return new Result<ChunksDTO>().ok(data);
@ -69,7 +69,7 @@ public class ChunksController {
public Result save(@RequestBody ChunksDTO dto){
chunksService.save(dto);
return new Result();
return new Result().ok(dto);
}
@PutMapping
@ -78,16 +78,16 @@ public class ChunksController {
chunksService.update(dto);
return new Result();
return new Result().ok(dto);
}
@DeleteMapping
@ApiOperation("删除")
public Result delete(@RequestBody Long[] ids){
public Result delete(@RequestBody String[] ids){
//效验数据
chunksService.delete(ids);
return new Result();
return new Result().ok(ids);
}
}

@ -52,7 +52,7 @@ public class FileChunksController {
@GetMapping("{id}")
@ApiOperation("信息")
public Result<FileChunksDTO> get(@PathVariable("id") Long id){
public Result<FileChunksDTO> get(@PathVariable("id") String id){
FileChunksDTO data = fileChunksService.get(id);
return new Result<FileChunksDTO>().ok(data);
@ -78,7 +78,7 @@ public class FileChunksController {
@DeleteMapping
@ApiOperation("删除")
public Result delete(@RequestBody Long[] ids){
public Result delete(@RequestBody String[] ids){
fileChunksService.delete(ids);

@ -99,7 +99,7 @@ public class FilesController {
public Result delete(@RequestBody String[] ids){
filesService.delete(ids);
return new Result();
return new Result().ok(ids);
}
}

@ -52,7 +52,7 @@ public class GlobalFilesController {
@GetMapping("{id}")
@ApiOperation("信息")
public Result<GlobalFilesDTO> get(@PathVariable("id") Long id){
public Result<GlobalFilesDTO> get(@PathVariable("id") String id){
GlobalFilesDTO data = globalFilesService.get(id);
return new Result<GlobalFilesDTO>().ok(data);
@ -64,7 +64,7 @@ public class GlobalFilesController {
globalFilesService.save(dto);
return new Result();
return new Result().ok(dto);
}
@PutMapping
@ -73,16 +73,16 @@ public class GlobalFilesController {
globalFilesService.update(dto);
return new Result();
return new Result().ok(dto);
}
@DeleteMapping
@ApiOperation("删除")
public Result delete(@RequestBody Long[] ids){
public Result delete(@RequestBody String[] ids){
globalFilesService.delete(ids);
return new Result();
return new Result().ok(ids);
}
}

@ -86,7 +86,7 @@ public class KnowledgeBasesController {
public Result delete(@RequestBody String[] ids){
knowledgeBasesService.delete(ids);
return new Result();
return new Result().ok(ids);
}
}

@ -52,7 +52,7 @@ public class MyDigitalHumanController {
@GetMapping("{id}")
@ApiOperation("信息")
public Result<MyDigitalHumanDTO> get(@PathVariable("id") Long id){
public Result<MyDigitalHumanDTO> get(@PathVariable("id") String id){
MyDigitalHumanDTO data = myDigitalHumanService.get(id);
return new Result<MyDigitalHumanDTO>().ok(data);
@ -62,23 +62,26 @@ public class MyDigitalHumanController {
@ApiOperation("保存")
public Result save(@RequestBody MyDigitalHumanDTO dto){
myDigitalHumanService.save(dto);
//
MyDigitalHumanDTO ret = myDigitalHumanService.genVideo(dto);
//myDigitalHumanService.save(dto);
return new Result();
return new Result<MyDigitalHumanDTO>().ok(ret);
}
@PutMapping
@ApiOperation("修改")
public Result update(@RequestBody MyDigitalHumanDTO dto){
myDigitalHumanService.update(dto);
MyDigitalHumanDTO ret = myDigitalHumanService.genVideo(dto);
//myDigitalHumanService.update(dto);
return new Result();
return new Result<MyDigitalHumanDTO>().ok(ret);
}
@DeleteMapping
@ApiOperation("删除")
public Result delete(@RequestBody Long[] ids){
public Result delete(@RequestBody String[] ids){
myDigitalHumanService.delete(ids);

@ -58,7 +58,7 @@ public class VoiceRefFilesController {
}
@GetMapping("{id}")
@ApiOperation("信息")
public Result<VoiceRefFilesDTO> get(@PathVariable("id") Long id){
public Result<VoiceRefFilesDTO> get(@PathVariable("id") String id){
VoiceRefFilesDTO data = voiceRefFilesService.get(id);
return new Result<VoiceRefFilesDTO>().ok(data);

@ -31,5 +31,6 @@ public class DigitalImgDTO implements Serializable {
@ApiModelProperty(value = "图片类型1数字人形象2数字人场景3数字人背景4音色形象")
private String imgType;
@ApiModelProperty(value = "音频在服务器的目录")
private String voiceServerDir;
}

@ -1,6 +1,7 @@
package com.pjilisense.flxai.entity;
import com.baomidou.mybatisplus.annotation.TableName;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import java.util.Date;
@ -31,4 +32,10 @@ public class DigitalImgEntity {
* 1234
*/
private String imgType;
/**
*
*/
private String voiceServerDir;
}

@ -12,4 +12,5 @@ import com.pjilisense.flxai.entity.MyDigitalHumanEntity;
*/
public interface MyDigitalHumanService extends CrudService<MyDigitalHumanEntity, MyDigitalHumanDTO> {
MyDigitalHumanDTO genVideo(MyDigitalHumanDTO dto);
}

@ -37,7 +37,7 @@ public class ChunksServiceImpl extends CrudServiceImpl<ChunksDao, ChunksEntity,
@Override
public List<Map<String, Object>> search(String userid,String keyText) {
return search1( userid, keyText);
return search2( userid, keyText);
}
private List<Map<String, Object>> search1(String userid,String keyText) {

@ -31,4 +31,15 @@ public class MyDigitalHumanServiceImpl extends CrudServiceImpl<MyDigitalHumanDao
}
@Override
public MyDigitalHumanDTO genVideo(MyDigitalHumanDTO dto) {
//TODO:gen video code here
if(dto.getId()!=null&&dto.getId().length()>0){
update(dto);
} else {
save(dto);
}
return dto;
}
}

@ -39,6 +39,7 @@ public class VoiceRefFilesServiceImpl extends CrudServiceImpl<VoiceRefFilesDao,
QueryWrapper<VoiceRefFilesEntity> wrapper = new QueryWrapper<>();
wrapper.eq(StrUtil.isNotBlank(dto.getLanguage()), "language", dto.getLanguage());
wrapper.eq(StrUtil.isNotBlank(dto.getId()), "id", dto.getId());
wrapper.like(StrUtil.isNotBlank(dto.getName()), "name", dto.getName());
wrapper.orderByAsc("id");
List<VoiceRefFilesEntity> entityList =baseDao.selectList(wrapper);
return ConvertUtils.sourceToTarget(entityList, VoiceRefFilesDTO.class);

@ -8,6 +8,7 @@
<result property="name" column="name"/>
<result property="imgPath" column="img_path"/>
<result property="imgType" column="img_type"/>
<result property="voiceServerDir" column="voice_server_dir"/>
</resultMap>

@ -28,7 +28,7 @@ public class ChunksControllerTest {
}
@Test
public void testGetAllAiTools() {
public void testsearch() {
String url = TestUtils.WebconnectURL+"/api/robot/chunks/search";
String charset = TestUtils.UTF8;
try {
@ -40,6 +40,12 @@ public class ChunksControllerTest {
} catch (Exception e) {
e.printStackTrace();
}
//param ={"text":"开源项目","userId":"2f3c5337-cf1a-40ce-bbe3-ec45455d82f1"}
//{"code":0,"msg":"success","data":[{"fid":"file_wb9Pk6GBUq0z","fname":"问答.txt","question":"开源项目","answer":"HanLP是由一系列NLP工具组成的开源项目其中包含了中文分词、词性标注、命名实体识别等功能。","similarity":"0.9999999999999997779553950749686919152736663818359375","id":"2bcbd9ae-053e-45e5-a3a9-5498915bd749","text":"开源项目,HanLP是由一系列NLP工具组成的开源项目其中包含了中文分词、词性标注、命名实体识别等功能。"},{"fid":"file_wb9Pk6GBUq0z","fname":"问答.txt","question":"开源项目HanLP","answer":"HanLP是由一系列NLP工具组成的开源项目其中包含了中文分词、词性标注、命名实体识别等功能。","similarity":"0.81649658092772592343777660062187351286411285400390625","id":"2bcbd9ae-053e-45e5-a3a9-5498915bd748","text":"开源项目HanLP,HanLP是由一系列NLP工具组成的开源项目其中包含了中文分词、词性标注、命名实体识别等功能。"}]}
//2
//param ={"text":"开源项目","userId":"2f3c5337-cf1a-40ce-bbe3-ec45455d82f1"}
//{"code":0,"msg":"success","data":[{"fid":"file_wb9Pk6GBUq0z","fname":"问答.txt","similarity":"0.452267016866645388262213600683026015758514404296875","id":"2bcbd9ae-053e-45e5-a3a9-5498915bd749","text":"开源项目,HanLP是由一系列NLP工具组成的开源项目其中包含了中文分词、词性标注、命名实体识别等功能。"},{"fid":"file_wb9Pk6GBUq0z","fname":"问答.txt","similarity":"0.424264068711928510335695818866952322423458099365234375","id":"2bcbd9ae-053e-45e5-a3a9-5498915bd748","text":"开源项目HanLP,HanLP是由一系列NLP工具组成的开源项目其中包含了中文分词、词性标注、命名实体识别等功能。"}]}
}
}

@ -113,9 +113,12 @@ public class DigitalImgControllerTest {
// params.put("name","开源项目");
// params.put("imgPath","uploads/202503/5c799cffa5864619b817063f77fe734d.png");
// params.put("imgType","1");
params.put("name","开源项目");
params.put("imgPath","uploads/202503/d1f8398f9bb844079f774ef496347e12.png");
params.put("imgType","1");
// params.put("name","开源项目");
// params.put("imgPath","uploads/202503/d1f8398f9bb844079f774ef496347e12.png");
// params.put("imgType","1");
params.put("name","音色1");
params.put("imgPath","uploads/202503/4eaee34079d74517ad6b9ea1e90e8308.jpg");
params.put("imgType","4");
String body = TestUtils.doPost(url, JSON.toJSONString(params), charset);
System.out.println(body);
//param ={"imgPath":"uploads/202503/d1f8398f9bb844079f774ef496347e12.png","name":"开源项目","imgType":"1"}

@ -28,7 +28,7 @@ public class FilesControllerTest {
}
@Test
public void testGetAllAiTools() {
public void testGetFileList() {
String url = TestUtils.WebconnectURL+"/api/robot/files/file_2FbUmaefz5b4";
String charset = TestUtils.UTF8;
try {
@ -37,6 +37,8 @@ public class FilesControllerTest {
} catch (Exception e) {
e.printStackTrace();
}
//method =GET
//{"code":0,"msg":"success","data":{"id":"file_2FbUmaefz5b4","userId":"fd8ae359-64dc-4a57-ab41-5438e2fda975","fileType":"text/plain","name":"jykc2.sql","size":1864714,"url":"files/482454/d2b0edb3-15bf-4fe6-ba2e-97b5f1fe0288.sql","metadata":{"date":"482454","path":"files/482454/d2b0edb3-15bf-4fe6-ba2e-97b5f1fe0288.sql","filename":"d2b0edb3-15bf-4fe6-ba2e-97b5f1fe0288.sql","dirname":"files/482454"},"createdAt":"2025-01-14 14:29:02","updatedAt":"2025-01-14 14:29:23","fileHash":"1d66edbc0b5da3289ecea1ae00ac384b2378603193574c5f0bc9a9a988de2a16","chunkTaskId":"ab3f259e-775a-4c57-8338-b273684f70d1","embeddingTaskId":"12020d55-d453-46d7-a8c8-b0bba5f5c747","accessedAt":"2025-01-14 14:29:02","fileCategory":"1","fileStatus":"1"}}
}
// @Test
// public void testsave() {
@ -81,6 +83,9 @@ public class FilesControllerTest {
} catch (Exception e) {
e.printStackTrace();
}
//method =POST
//param ={"name":"jykc2.sql","id":"file_2FbUmaefz5b4"}
//{"code":0,"msg":"success","data":true}
}
@Test
@ -96,6 +101,9 @@ public class FilesControllerTest {
} catch (Exception e) {
e.printStackTrace();
}
//method =POST
//param ={"fileStatus":"1","id":"file_2FbUmaefz5b4"}
//{"code":0,"msg":"success","data":true}
}
@Test
public void testmodifyFileCategory() {
@ -110,5 +118,8 @@ public class FilesControllerTest {
} catch (Exception e) {
e.printStackTrace();
}
//method =POST
//param ={"fileCategory":"1","id":"file_2FbUmaefz5b4"}
//{"code":0,"msg":"success","data":true}
}
}

@ -37,6 +37,7 @@ public class KnowledgeBasesControllerTest {
} catch (Exception e) {
e.printStackTrace();
}
//{"code":0,"msg":"success","data":{"id":"kb_E5EcemGyKnh9","name":"知识库1","description":"知识库测试2","avatar":null,"type":null,"userId":"fd8ae359-64dc-4a57-ab41-5438e2fda975","isPublic":false,"settings":null,"createdAt":"2025-01-14 14:27:55","updatedAt":"2025-01-16 16:01:51","accessedAt":"2025-01-14 14:27:55"}}
}
// @Test
// public void testsave() {
@ -83,5 +84,7 @@ public class KnowledgeBasesControllerTest {
} catch (Exception e) {
e.printStackTrace();
}
//param ={"description":"知识库测试23","id":"kb_E5EcemGyKnh9"}
//{"code":0,"msg":"success","data":true}
}
}

@ -0,0 +1,86 @@
package test.com.pjilisense.flxai.controller;
import com.alibaba.fastjson.JSON;
import io.swagger.annotations.ApiModelProperty;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import test.com.pjilisense.flxai.TestUtils;
import java.util.HashMap;
import java.util.Map;
public class MyDigitalHumanControllerTest {
@BeforeEach
public void setUpTestDataWithinTransaction() {
// set up test data within the transaction
}
@AfterEach
public void tearDownWithinTransaction() {
// execute "tear down" logic within the transaction
}
@Test
public void testGetAllAiTools() {
String url = TestUtils.WebconnectURL+"/api/robot/mydigitalhuman/queryAll?language=ch";
String charset = TestUtils.UTF8;
try {
Map<String, Object> params = new HashMap<String, Object>();
params.put("language","zh");
String body = TestUtils.doPost(url, JSON.toJSONString(params), charset);
System.out.println(body);
} catch (Exception e) {
e.printStackTrace();
}
//param ={"language":"zh"}
//{"code":0,"msg":"success","data":[{"id":"1896763411581108225","name":"音色Modify","voicePath":"uploads/202503/5c799cffa5864619b817063f77fe734e.MP3","imgPath":"1896761492913520641","language":"zh"}]}
}
@Test
public void testSave() {
String url = TestUtils.WebconnectURL+"/api/robot/mydigitalhuman";
String charset = TestUtils.UTF8;
try {
Map<String, Object> params = new HashMap<String, Object>();
params.put("userid","fd8ae359-64dc-4a57-ab41-5438e2fda975");
params.put("imageid","1896761492913520641");
params.put("sceneimgid","1896761492913520641");
params.put("bkimgid","1896761492913520641");
params.put("voiceid","1896763411581108225");
params.put("videoimg","1896763411581108225");
params.put("videodir","genvideo/202503/5c799cffa5864619b817063f77fe734d.mp4");
params.put("voicedir","genvoice/202503/A1");
String body = TestUtils.doPost(url, JSON.toJSONString(params), charset);
System.out.println(body);
//param ={"voiceid":"1896763411581108225","videodir":"genvideo/202503/5c799cffa5864619b817063f77fe734d.mp4","imageid":"1896761492913520641","bkimgid":"1896761492913520641","voicedir":"genvoice/202503/A1","videoimg":"1896763411581108225","userid":"fd8ae359-64dc-4a57-ab41-5438e2fda975","sceneimgid":"1896761492913520641"}
//{"code":0,"msg":"success","data":null}
} catch (Exception e) {
e.printStackTrace();
}
}
@Test
public void testupdate() {
String url = TestUtils.WebconnectURL+"/api/robot/mydigitalhuman";
String charset = TestUtils.UTF8;
try {
Map<String, Object> params = new HashMap<String, Object>();
params.put("id","1896816581502042113");
params.put("userid","fd8ae359-64dc-4a57-ab41-5438e2fda975");
params.put("imageid","1896761492913520641");
params.put("sceneimgid","1896761492913520641");
params.put("bkimgid","1896761492913520641");
params.put("voiceid","1896763411581108225");
params.put("videoimg","1896763411581108225");
params.put("videodir","genvideo/202503/5c799cffa5864619b817063f77fe734d.mp4");
params.put("voicedir","genvoice/202503/A2");
String body = TestUtils.doPut(url, JSON.toJSONString(params), charset);
System.out.println(body);
//param ={"voiceid":"1896763411581108225","videodir":"genvideo/202503/5c799cffa5864619b817063f77fe734d.mp4","imageid":"1896761492913520641","bkimgid":"1896761492913520641","voicedir":"genvoice/202503/A2","videoimg":"1896763411581108225","id":"1896816581502042113","userid":"fd8ae359-64dc-4a57-ab41-5438e2fda975","sceneimgid":"1896761492913520641"}
//{"code":0,"msg":"success","data":null}
} catch (Exception e) {
e.printStackTrace();
}
}
}

@ -0,0 +1,82 @@
package test.com.pjilisense.flxai.controller;
import com.alibaba.fastjson.JSON;
import io.swagger.annotations.ApiModelProperty;
import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClients;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import test.com.pjilisense.flxai.TestUtils;
import java.io.*;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.HashMap;
import java.util.Map;
public class VoiceRefFilesControllerTest {
@BeforeEach
public void setUpTestDataWithinTransaction() {
// set up test data within the transaction
}
@AfterEach
public void tearDownWithinTransaction() {
// execute "tear down" logic within the transaction
}
@Test
public void testGetAllAiTools() {
String url = TestUtils.WebconnectURL+"/api/robot/voicereffiles/queryAll?language=ch";
String charset = TestUtils.UTF8;
try {
Map<String, Object> params = new HashMap<String, Object>();
params.put("language","zh");
String body = TestUtils.doPost(url, JSON.toJSONString(params), charset);
System.out.println(body);
} catch (Exception e) {
e.printStackTrace();
}
//param ={"language":"zh"}
//{"code":0,"msg":"success","data":[{"id":"1896763411581108225","name":"音色Modify","voicePath":"uploads/202503/5c799cffa5864619b817063f77fe734e.MP3","imgPath":"1896761492913520641","language":"zh"}]}
}
@Test
public void testSave() {
String url = TestUtils.WebconnectURL+"/api/robot/voicereffiles";
String charset = TestUtils.UTF8;
try {
Map<String, Object> params = new HashMap<String, Object>();
params.put("name","音色1");
params.put("voicePath","uploads/202503/5c799cffa5864619b817063f77fe734d.MP3");
params.put("imgPath","1896761492913520641");
params.put("language","zh");
params.put("voiceServerDir","A1");
String body = TestUtils.doPost(url, JSON.toJSONString(params), charset);
System.out.println(body);
//param ={"voicePath":"uploads/202503/5c799cffa5864619b817063f77fe734d.MP3","imgPath":"1896761492913520641","name":"音色1","language":"zh","voiceServerDir":"A1"}
//{"code":0,"msg":"success","data":null}
} catch (Exception e) {
e.printStackTrace();
}
}
@Test
public void testupdate() {
String url = TestUtils.WebconnectURL+"/api/robot/voicereffiles";
String charset = TestUtils.UTF8;
try {
Map<String, Object> params = new HashMap<String, Object>();
params.put("id","1896763411581108225");
params.put("name","音色Modify");
params.put("voicePath","uploads/202503/5c799cffa5864619b817063f77fe734e.MP3");
params.put("imgPath","1896761492913520641");
params.put("language","zh");
String body = TestUtils.doPut(url, JSON.toJSONString(params), charset);
System.out.println(body);
//param ={"voicePath":"uploads/202503/5c799cffa5864619b817063f77fe734e.MP3","imgPath":"1896761492913520641","name":"音色Modify","language":"zh","id":"1896763411581108225","voiceServerDir":"A1"}
//{"code":0,"msg":"success","data":null}
} catch (Exception e) {
e.printStackTrace();
}
}
}
Loading…
Cancel
Save