From 34981c30a78e8bbd7791131059a9210f9928b62c Mon Sep 17 00:00:00 2001
From: 刘光辉 <347230014@qq.com>
Date: 星期四, 17 九月 2026 09:24:09 +0800
Subject: [PATCH] Merge remote-tracking branch 'origin/master' into master
---
jnpf-file/jnpf-file-upload/src/test/java/jnpf/upload/controller/FileUploadControllerTest.java | 49 +++++++++++++++++++++++++++++++++++++++++++++++++
1 files changed, 49 insertions(+), 0 deletions(-)
diff --git a/jnpf-file/jnpf-file-upload/src/test/java/jnpf/upload/controller/FileUploadControllerTest.java b/jnpf-file/jnpf-file-upload/src/test/java/jnpf/upload/controller/FileUploadControllerTest.java
new file mode 100644
index 0000000..5fc23eb
--- /dev/null
+++ b/jnpf-file/jnpf-file-upload/src/test/java/jnpf/upload/controller/FileUploadControllerTest.java
@@ -0,0 +1,49 @@
+package jnpf.upload.controller;
+
+import org.dromara.x.file.storage.core.FileInfo;
+import org.dromara.x.file.storage.core.FileStorageService;
+import org.dromara.x.file.storage.core.upload.UploadPretreatment;
+import org.junit.Test;
+import org.springframework.web.multipart.MultipartFile;
+
+import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.when;
+
+public class FileUploadControllerTest {
+
+ @Test
+ public void replaceFileKeepsStorageFilename() {
+ FileStorageService storage = mock(FileStorageService.class);
+ UploadPretreatment upload = mock(UploadPretreatment.class);
+ MultipartFile file = mock(MultipartFile.class);
+ FileInfo stored = new FileInfo();
+ stored.setFilename("template.docx");
+ when(storage.of(file)).thenReturn(upload);
+ when(upload.setPath("WebAnnexFile/")).thenReturn(upload);
+ when(upload.setOriginalFilename("template.docx")).thenReturn(upload);
+ when(upload.setSaveFilename("template.docx")).thenReturn(upload);
+ when(upload.upload()).thenReturn(stored);
+
+ assertTrue(new FileUploadController(storage).replaceFile(file, "WebAnnexFile/", "template.docx"));
+ verify(upload).setSaveFilename("template.docx");
+ }
+
+ @Test
+ public void replaceFileRejectsUnexpectedStorageFilename() {
+ FileStorageService storage = mock(FileStorageService.class);
+ UploadPretreatment upload = mock(UploadPretreatment.class);
+ MultipartFile file = mock(MultipartFile.class);
+ FileInfo stored = new FileInfo();
+ stored.setFilename("generated.docx");
+ when(storage.of(file)).thenReturn(upload);
+ when(upload.setPath("WebAnnexFile/")).thenReturn(upload);
+ when(upload.setOriginalFilename("template.docx")).thenReturn(upload);
+ when(upload.setSaveFilename("template.docx")).thenReturn(upload);
+ when(upload.upload()).thenReturn(stored);
+
+ assertFalse(new FileUploadController(storage).replaceFile(file, "WebAnnexFile/", "template.docx"));
+ }
+}
--
Gitblit v1.8.0