You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

86 lines
2.9 KiB
Java

package me.zhengjie;
import com.alibaba.fastjson.JSON;
import me.zhengjie.modules.upload.domain.UploadFile;
import me.zhengjie.modules.upload.repository.UploadFileRepository;
import me.zhengjie.modules.upload.service.UploadFileService;
import me.zhengjie.modules.upload.service.dto.UploadFileDto;
import me.zhengjie.modules.upload.service.dto.UploadFileQueryCriteria;
import me.zhengjie.modules.upload.service.mapstruct.UploadFileMapper;
import me.zhengjie.modules.upload.task.SaveToFileTask;
import me.zhengjie.utils.PageUtil;
import me.zhengjie.utils.QueryHelp;
import me.zhengjie.utils.SecurityUtils;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.test.context.junit4.SpringRunner;
import java.sql.Timestamp;
@RunWith(SpringRunner.class)
@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class SpringApplicationTest {
@Autowired
private SaveToFileTask saveToFileTask;
@Autowired
private UploadFileRepository uploadFileRepository;
@Autowired
private UploadFileMapper uploadFileMapper;
@Autowired
private UploadFileService uploadFileService;
/**
* 获取当前的用户信息
*/
@Test
public void testGetName(){
System.out.println(SecurityUtils.getCurrentUsername());
}
/**
* 功能测试 - 向远程发送更新记录
*/
@Test
public void testRemoteRec(){
String path = "C:\\Users\\Administrator\\Desktop\\c76l.txt";
UploadFileDto uploadFileDto = new UploadFileDto();
uploadFileDto.setUploadTime(new Timestamp(new java.util.Date().getTime()));
uploadFileDto.setOperation("测试用户1");
uploadFileDto.setFileCount(1000000L);
uploadFileDto.setFileTransSuccessCount(999999L);
uploadFileDto.setUploadFileTaskName("测试任务1");
// System.out.println(saveToFileTask.sendUpdatePostReq(path, uploadFileDto));
}
@Test
public void testSFTPTrans(){
// saveToFileTask.transFileToOtherServer("C:\\Users\\Administrator\\Desktop\\c76l.txt");
}
@Test
public void testFindAll(){
Pageable pageable = PageRequest.of(1, 10);
UploadFileQueryCriteria uploadFileQueryCriteria = new UploadFileQueryCriteria();
uploadFileQueryCriteria.setOperation("xiaoxing");
Page<UploadFile> page = uploadFileRepository.findAll((root, criteriaQuery, criteriaBuilder) -> QueryHelp.getPredicate(root, uploadFileQueryCriteria, criteriaBuilder), pageable);
String jsonString = JSON.toJSONString(PageUtil.toPage(page.map(uploadFileMapper::toDto)));
System.out.println(jsonString);
}
}