HHH-19464 Create random temporary ZIP file instead of using one from ressources

Author: cigaly

hibernate-core/src/test/java/org/hibernate/orm/test/lob/JarFileEntryBlobTest.java

@@ -4,36 +4,43 @@
  */
 package org.hibernate.orm.test.lob;
 
-import org.hibernate.bugs.TestEntity;
+import jakarta.persistence.Entity;
+import jakarta.persistence.Id;
+import jakarta.persistence.Lob;
 import org.hibernate.engine.jdbc.env.internal.NonContextualLobCreator;
 import org.hibernate.testing.orm.junit.DomainModel;
 import org.hibernate.testing.orm.junit.JiraKey;
 import org.hibernate.testing.orm.junit.SessionFactory;
 import org.hibernate.testing.orm.junit.SessionFactoryScope;
+import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
 
 import java.io.File;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
 import java.sql.Blob;
 import java.sql.SQLException;
+import java.util.Random;
 import java.util.jar.JarEntry;
 import java.util.jar.JarFile;
+import java.util.zip.ZipEntry;
+import java.util.zip.ZipOutputStream;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
-
-@DomainModel(annotatedClasses = TestEntity.class)
+@DomainModel(annotatedClasses = JarFileEntryBlobTest.TestEntity.class)
 @SessionFactory
-@JiraKey( "HHH-19464" )
+@JiraKey("HHH-19464")
 class JarFileEntryBlobTest {
 
+	public static final String ZIP_ENTRY_NAME = "test.bin";
+
 	@Test
-	void hibernate_blob_streaming(SessionFactoryScope scope) throws Exception {
-		final var zipFilePath = getClass().getClassLoader().getResource( "org/hibernate/orm/test/lob/JarFileEntryBlobTest.zip" );
-		File file = new File( zipFilePath.toURI() );
+	void hibernate_blob_streaming(SessionFactoryScope scope, @TempDir File tempDir) throws Exception {
+		final var file = createZipFile( tempDir );
 
 		try (JarFile jarFile = new JarFile( file )) {
-			JarEntry entry = jarFile.getJarEntry( "pizza.png" );
+			JarEntry entry = jarFile.getJarEntry( ZIP_ENTRY_NAME );
 			long size = entry.getSize();
 			scope.inTransaction( entityManager -> {
 						try {
@@ -56,12 +63,61 @@ void hibernate_blob_streaming(SessionFactoryScope scope) throws Exception {
 			scope.inStatelessSession( session -> {
 				final var entity = session.get( TestEntity.class, 1L );
 				try {
-					assertEquals( size, entity.getData().length() );
+					Assertions.assertEquals( size, entity.getData().length() );
 				}
 				catch (SQLException e) {
 					throw new RuntimeException( e );
 				}
 			} );
 		}
 	}
+
+	private static File createZipFile(File tempDir) throws IOException {
+		final var testFile = new File( tempDir, ZIP_ENTRY_NAME );
+		final var bytes = new byte[320000];
+		new Random().nextBytes( bytes );
+		try (var output = new FileOutputStream( testFile )) {
+			try (var zipStream = new ZipOutputStream( output )) {
+				final var zipEntry = new ZipEntry( ZIP_ENTRY_NAME );
+				zipStream.putNextEntry( zipEntry );
+				zipStream.write( bytes );
+			}
+		}
+		return testFile;
+	}
+
+	@Entity
+	public static class TestEntity {
+
+		@Id
+		Long id;
+
+		@Lob
+		Blob data;
+
+		public Long getId() {
+			return id;
+		}
+
+		public void setId(Long id) {
+			this.id = id;
+		}
+
+		public Blob getData() {
+			return data;
+		}
+
+		public InputStream getInputStream() {
+			try {
+				return data.getBinaryStream();
+			}
+			catch (SQLException e) {
+				throw new IllegalArgumentException( "Could not obtain requested input stream", e );
+			}
+		}
+
+		public void setData(Blob data) {
+			this.data = data;
+		}
+	}
 }