From 24681c81c09022f584a57006f2534b5f74723414 Mon Sep 17 00:00:00 2001
From: 云 <2163098428@qq.com>
Date: 星期二, 30 六月 2026 09:27:31 +0800
Subject: [PATCH] 初始化项目
---
yudao-module-infra/src/test/java/cn/iocoder/yudao/module/infra/service/file/FileServiceImplTest.java | 104 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 104 insertions(+), 0 deletions(-)
diff --git a/yudao-module-infra/src/test/java/cn/iocoder/yudao/module/infra/service/file/FileServiceImplTest.java b/yudao-module-infra/src/test/java/cn/iocoder/yudao/module/infra/service/file/FileServiceImplTest.java
index 8cd8fe7..4b14aec 100644
--- a/yudao-module-infra/src/test/java/cn/iocoder/yudao/module/infra/service/file/FileServiceImplTest.java
+++ b/yudao-module-infra/src/test/java/cn/iocoder/yudao/module/infra/service/file/FileServiceImplTest.java
@@ -5,6 +5,7 @@
import cn.iocoder.yudao.framework.common.util.object.ObjectUtils;
import cn.iocoder.yudao.framework.test.core.ut.BaseDbUnitTest;
import cn.iocoder.yudao.framework.test.core.util.AssertUtils;
+import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.FileCreateReqVO;
import cn.iocoder.yudao.module.infra.controller.admin.file.vo.file.FilePageReqVO;
import cn.iocoder.yudao.module.infra.dal.dataobject.file.FileDO;
import cn.iocoder.yudao.module.infra.dal.mysql.file.FileMapper;
@@ -22,6 +23,7 @@
import static cn.iocoder.yudao.framework.test.core.util.AssertUtils.assertServiceException;
import static cn.iocoder.yudao.framework.test.core.util.RandomUtils.*;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.FILE_NOT_EXISTS;
+import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.FILE_PATH_INVALID;
import static org.junit.jupiter.api.Assertions.*;
import static org.mockito.ArgumentMatchers.same;
import static org.mockito.Mockito.*;
@@ -173,6 +175,16 @@
}
@Test
+ public void testDeleteFile_pathInvalid() {
+ // mock 鏁版嵁
+ FileDO dbFile = randomPojo(FileDO.class, o -> o.setConfigId(10L).setPath("../tudou.jpg"));
+ fileMapper.insert(dbFile);
+
+ // 璋冪敤锛屽苟鏂█寮傚父
+ assertServiceException(() -> fileService.deleteFile(dbFile.getId()), FILE_PATH_INVALID);
+ }
+
+ @Test
public void testGetFileContent() throws Exception {
// 鍑嗗鍙傛暟
Long configId = 10L;
@@ -187,6 +199,76 @@
byte[] result = fileService.getFileContent(configId, path);
// 鏂█
assertSame(result, content);
+ }
+
+ @Test
+ public void testGetFileContent_pathInvalid() {
+ // 鍑嗗鍙傛暟
+ Long configId = 10L;
+ String path = "../tudou.jpg";
+
+ // 璋冪敤锛屽苟鏂█寮傚父
+ assertServiceException(() -> fileService.getFileContent(configId, path), FILE_PATH_INVALID);
+ }
+
+ @Test
+ public void testGetFileByConfigIdAndPath() {
+ // mock 鏁版嵁
+ FileDO dbFile = randomPojo(FileDO.class, o -> o.setConfigId(10L).setPath("avatar/涓枃 100%+鏂囦欢.jpg"));
+ fileMapper.insert(dbFile);
+ FileDO latestFile = ObjectUtils.cloneIgnoreId(dbFile, o -> o.setName("鏈�鏂版枃浠跺悕.jpg"));
+ fileMapper.insert(latestFile);
+ fileMapper.insert(ObjectUtils.cloneIgnoreId(dbFile, o -> o.setPath("avatar/other.jpg")));
+ fileMapper.insert(ObjectUtils.cloneIgnoreId(dbFile, o -> o.setConfigId(20L)));
+
+ // 璋冪敤
+ FileDO result = fileService.getFileByConfigIdAndPath(10L, "avatar/涓枃 100%+鏂囦欢.jpg");
+
+ // 鏂█
+ AssertUtils.assertPojoEquals(latestFile, result);
+ }
+
+ @Test
+ public void testCreateFileByPresignedPath_success() {
+ // 鍑嗗鍙傛暟
+ FileCreateReqVO reqVO = randomPojo(FileCreateReqVO.class, o -> {
+ o.setPath("avatar/test.jpg");
+ o.setName("test.jpg");
+ o.setUrl("https://www.iocoder.cn/test.jpg?token=123");
+ });
+
+ // 璋冪敤
+ Long fileId = fileService.createFile(reqVO);
+
+ // 鏂█
+ FileDO file = fileMapper.selectById(fileId);
+ assertEquals("avatar/test.jpg", file.getPath());
+ assertEquals("test.jpg", file.getName());
+ assertEquals("https://www.iocoder.cn/test.jpg", file.getUrl());
+ }
+
+ @Test
+ public void testCreateFileByPresignedPath_nameInvalid() {
+ // 鍑嗗鍙傛暟
+ FileCreateReqVO reqVO = randomPojo(FileCreateReqVO.class, o -> {
+ o.setPath("avatar/test.jpg");
+ o.setName("../test.jpg");
+ });
+
+ // 璋冪敤锛屽苟鏂█寮傚父
+ assertServiceException(() -> fileService.createFile(reqVO), FILE_PATH_INVALID);
+ }
+
+ @Test
+ public void testCreateFileByPresignedPath_pathInvalid() {
+ // 鍑嗗鍙傛暟
+ FileCreateReqVO reqVO = randomPojo(FileCreateReqVO.class, o -> {
+ o.setPath("../test.jpg");
+ o.setName("test.jpg");
+ });
+
+ // 璋冪敤锛屽苟鏂█寮傚父
+ assertServiceException(() -> fileService.createFile(reqVO), FILE_PATH_INVALID);
}
@Test
@@ -343,6 +425,28 @@
}
@Test
+ public void testGenerateUploadPath_FileNameInvalid() {
+ // 鍑嗗鍙傛暟
+ String name = "../test.jpg";
+ String directory = "avatar";
+ FileServiceImpl.PATH_PREFIX_DATE_ENABLE = false;
+ FileServiceImpl.PATH_SUFFIX_TIMESTAMP_ENABLE = false;
+
+ // 璋冪敤锛屽苟鏂█寮傚父
+ assertServiceException(() -> fileService.generateUploadPath(name, directory), FILE_PATH_INVALID);
+ }
+
+ @Test
+ public void testGenerateUploadPath_DirectoryInvalid() {
+ // 鍑嗗鍙傛暟
+ String name = "test.jpg";
+ String directory = "../avatar";
+
+ // 璋冪敤锛屽苟鏂█寮傚父
+ assertServiceException(() -> fileService.generateUploadPath(name, directory), FILE_PATH_INVALID);
+ }
+
+ @Test
public void testGenerateUploadPath_DirectoryEmpty() {
// 鍑嗗鍙傛暟
String name = "test.jpg";
--
Gitblit v1.9.3