BAEL-5942 Code changes (#13147)

* BAEL-5942 Code changes * BAEL-5942 Code changes * BAEL-5942 Code updates from Review * BAEL-5942 Code updates from Review * BAEL-5942 Code updates from Review * BAEL-5942 Code updates from Review * BAEL-5942 Code updates from Review * BAEL-5942 Code updates from Review * BAEL-5942 Test updates from Review * BAEL-5942 Test updates from Review * BAEL-5942 Test updates from Review
2023-01-22 12:48:56 +01:00
parent 42d27a6bfe
commit 506125b060
7 changed files with 189 additions and 0 deletions
--- a/spring-web-modules/spring-mvc-file/src/test/java/com/baeldung/file/CustomMultipartFileUnitTest.java
+++ b/spring-web-modules/spring-mvc-file/src/test/java/com/baeldung/file/CustomMultipartFileUnitTest.java
@@ -0,0 +1,49 @@
+package com.baeldung.file;
+
+import java.io.IOException;
+
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.springframework.mock.web.MockMultipartFile;
+
+class CustomMultipartFileUnitTest {
+
+    @Test
+    void whenProvidingByteArray_thenMultipartFileCreated() throws IOException {
+        byte[] inputArray = "Test String".getBytes();
+        CustomMultipartFile customMultipartFile = new CustomMultipartFile(inputArray);
+        Assertions.assertFalse(customMultipartFile.isEmpty());
+        Assertions.assertArrayEquals(inputArray, customMultipartFile.getBytes());
+        Assertions.assertEquals(inputArray.length, customMultipartFile.getSize());
+    }
+
+    @Test
+    void whenProvidingEmptyByteArray_thenMockMultipartFileIsEmpty() throws IOException {
+        byte[] inputArray = "".getBytes();
+        MockMultipartFile mockMultipartFile = new MockMultipartFile("tempFileName", inputArray);
+        Assertions.assertTrue(mockMultipartFile.isEmpty());
+    }
+
+    @Test
+    void whenProvidingNullByteArray_thenMockMultipartFileIsEmpty() throws IOException {
+        byte[] inputArray = null;
+        MockMultipartFile mockMultipartFile = new MockMultipartFile("tempFileName", inputArray);
+        Assertions.assertTrue(mockMultipartFile.isEmpty());
+    }
+
+    @Test
+    void whenProvidingByteArray_thenMultipartFileInputSizeMatches() throws IOException {
+        byte[] inputArray = "Testing String".getBytes();
+        CustomMultipartFile customMultipartFile = new CustomMultipartFile(inputArray);
+        Assertions.assertEquals(inputArray.length, customMultipartFile.getSize());
+    }
+
+    @Test
+    void whenProvidingByteArray_thenMockMultipartFileCreated() throws IOException {
+        byte[] inputArray = "Test String".getBytes();
+        MockMultipartFile mockMultipartFile = new MockMultipartFile("tempFileName", inputArray);
+        Assertions.assertFalse(mockMultipartFile.isEmpty());
+        Assertions.assertArrayEquals(inputArray, mockMultipartFile.getBytes());
+        Assertions.assertEquals(inputArray.length, mockMultipartFile.getSize());
+    }
+}