diff --git a/src/test/java/cn/celess/blog/service/fileserviceimpl/QiniuFileServiceImplTest.java b/src/test/java/cn/celess/blog/service/fileserviceimpl/QiniuFileServiceImplTest.java index 7997c5d..2d2c1b4 100644 --- a/src/test/java/cn/celess/blog/service/fileserviceimpl/QiniuFileServiceImplTest.java +++ b/src/test/java/cn/celess/blog/service/fileserviceimpl/QiniuFileServiceImplTest.java @@ -1,6 +1,7 @@ package cn.celess.blog.service.fileserviceimpl; import cn.celess.blog.BaseTest; +import cn.celess.blog.entity.model.FileInfo; import cn.celess.blog.entity.model.FileResponse; import lombok.SneakyThrows; import lombok.extern.slf4j.Slf4j; @@ -11,19 +12,49 @@ import org.springframework.test.context.ActiveProfiles; import java.io.File; import java.io.FileInputStream; import java.io.FileOutputStream; +import java.util.List; import static org.junit.Assert.*; @Slf4j -@ActiveProfiles("prod") public class QiniuFileServiceImplTest extends BaseTest { @Autowired QiniuFileServiceImpl qiniuFileService; + private String fileName; @SneakyThrows @Test public void uploadFile() { + fileName = null; + File file = createFile(); + FileResponse fileResponse = qiniuFileService.uploadFile(new FileInputStream(file), file.getName()); + assertEquals(file.getName(), fileResponse.key); + assertEquals("qiniu", fileResponse.bucket); + assertNotNull(fileResponse.hash); + fileName = fileResponse.key; + + qiniuFileService.deleteFile(fileName); + + file.deleteOnExit(); + } + + @Test + public void getFileList() { + List fileList = qiniuFileService.getFileList(); + fileList.forEach(fileInfo -> { + assertNotNull(fileInfo.key); + assertNotNull(fileInfo.hash); + }); + } + + @Test + public void deleteFile() { + uploadFile(); + } + + @SneakyThrows + private File createFile() { String fileName = "test." + randomStr(3); File file = new File(fileName); if (!file.exists() && file.createNewFile()) { @@ -36,15 +67,6 @@ public class QiniuFileServiceImplTest extends BaseTest { outputStream.flush(); outputStream.close(); } - FileResponse fileResponse = qiniuFileService.uploadFile(new FileInputStream(file), fileName); - assertEquals(fileName, fileResponse.key); - } - - @Test - public void getFileList() { - } - - @Test - public void deleteFile() { + return file; } } \ No newline at end of file