https://jira.baeldung.com/browse/BAEL-7958

Author: Michaelin007

persistence-modules/core-java-persistence-3/retrieve-warehouse.png

@@ -18,7 +18,6 @@ public class JdbcConnection {
     public static Connection connect() throws SQLException {
         Connection connection = DriverManager.getConnection("jdbc:h2:./test", "sa", "");
         return connection;
-
     }
 
     public static byte[] convertFileToByteArray(String filePath) throws IOException {

persistence-modules/core-java-persistence-3/src/main/java/storefileblob/JdbcConnection.java

@@ -1,50 +1,54 @@
 package storefileblob;
 
 import org.junit.Test;
-import org.junit.jupiter.api.MethodOrderer;
-import org.junit.jupiter.api.Order;
-import org.junit.jupiter.api.TestMethodOrder;
-
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.CsvSource;
 import java.io.IOException;
 import java.sql.SQLException;
-
+import static org.junit.jupiter.api.Assertions.assertArrayEquals;
 import static org.junit.jupiter.api.Assertions.assertTrue;
 
-@TestMethodOrder(MethodOrderer.OrderAnnotation.class)
 public class JdbcConnectionUnitTest {
 
+    private static final String TEST_FILE_PATH = "warehouse.png";
+    private static final String RETRIEVED_FILE_PATH = "retrieve-warehouse.png";
+    private static JdbcConnection jdbcConnection;
     String selectBlob = """
         SELECT picture FROM warehouses WHERE id=?
         """;
 
+    @BeforeAll
+    public static void setUp() throws SQLException {
+        jdbcConnection = new JdbcConnection();
+        jdbcConnection.createSchema();
+    }
+
     @Test
-    @Order(1)
     public void givenDatabaseSchema_whenJdbcConnectionIsEstablished_thenCreateSchema() throws SQLException {
-        JdbcConnection jdbcConnection = new JdbcConnection();
         assertTrue(jdbcConnection.createSchema());
     }
 
-    @Test
-    @Order(3)
-    public void givenBlobFile_whenInsertingTheBlobFileIntoTheWarehouseTable_thenSuccessful() throws SQLException, IOException {
-        JdbcConnection jdbcConnection = new JdbcConnection();
-        jdbcConnection.createSchema();
-        assertTrue(jdbcConnection.insertFile(4, "Liu", 3000, "warehouse.png"));
+    @ParameterizedTest
+    @CsvSource({ "1, 'Liu', 3000", "2, 'Walmart', 5000" })
+    public void givenBlobFile_whenInsertingTheBlobFileAsByteArray_thenSuccessful(int id, String name, int capacity) throws SQLException, IOException {
+        boolean result = jdbcConnection.insertFile(id, name, capacity, TEST_FILE_PATH);
+        assertTrue(result);
     }
 
-    @Test
-    @Order(2)
-    public void givenBlobFile_whenInsertingTheFileIntoTheDatabaseAsStream_thenSuccessful() throws SQLException, IOException {
-        JdbcConnection jdbcConnection = new JdbcConnection();
-        jdbcConnection.createSchema();
-        assertTrue(jdbcConnection.insertFileAsStream(6, "Liu", 3000, "warehouse.png"));
+    @ParameterizedTest
+    @CsvSource({ "7, 'Dylan', 8000", "8, 'Jumia', 1000" })
+    public void givenBlobFile_whenInsertingTheFileIntoTheDatabaseInChunks_thenSuccessful(int id, String name, int capacity) throws SQLException, IOException {
+        boolean result = jdbcConnection.insertFileAsStream(id, name, capacity, TEST_FILE_PATH);
+        assertTrue(result);
     }
 
     @Test
-    @Order(4)
     public void givenAnEntityWithBlob_whenReadingBlobAndConvertingItToFile_thenSuccessful() throws SQLException, IOException {
-        JdbcConnection jdbcConnection = new JdbcConnection();
-        assertTrue(JdbcConnection.writeBlobToFile(selectBlob, 1, 6, "retrieve-warehouse.png"));
+        boolean result = JdbcConnection.writeBlobToFile(selectBlob, 1, 2, RETRIEVED_FILE_PATH);
+        byte[] expectedBytes = JdbcConnection.convertFileToByteArray(TEST_FILE_PATH);
+        byte[] retrievedBytes = JdbcConnection.convertFileToByteArray(RETRIEVED_FILE_PATH);
+        assertArrayEquals(expectedBytes, retrievedBytes);
     }
 
 }