HHH-14725 Fixed test

Author: dreab8

hibernate-envers/src/test/java/org/hibernate/orm/test/envers/integration/blob/BasicBlobTest.java

@@ -4,26 +4,25 @@
  */
 package org.hibernate.orm.test.envers.integration.blob;
 
-import static org.hamcrest.MatcherAssert.assertThat;
-import static org.hibernate.testing.transaction.TransactionUtil.doInJPA;
-import static org.junit.Assert.fail;
-
-import java.io.BufferedInputStream;
-import java.io.InputStream;
-import java.nio.file.Files;
-import java.nio.file.Path;
-import java.sql.Blob;
-
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.Id;
 import org.hamcrest.Matchers;
 import org.hibernate.engine.jdbc.proxy.BlobProxy;
 import org.hibernate.envers.Audited;
 import org.hibernate.orm.test.envers.BaseEnversJPAFunctionalTestCase;
 import org.hibernate.orm.test.envers.Priority;
 import org.junit.Test;
 
-import jakarta.persistence.Entity;
-import jakarta.persistence.GeneratedValue;
-import jakarta.persistence.Id;
+import java.io.BufferedInputStream;
+import java.io.InputStream;
+import java.nio.file.Files;
+import java.nio.file.Path;
+import java.sql.Blob;
+
+import static org.hamcrest.MatcherAssert.assertThat;
+import static org.hibernate.testing.transaction.TransactionUtil.doInJPA;
+import static org.junit.Assert.fail;
 
 /**
  * @author Chris Cranford
@@ -32,19 +31,19 @@ public class BasicBlobTest extends BaseEnversJPAFunctionalTestCase {
 
 	@Override
 	protected Class<?>[] getAnnotatedClasses() {
-		return new Class<?>[] { Asset.class };
+		return new Class<?>[] {Asset.class};
 	}
 
 	@Test
 	@Priority(10)
 	public void initData() {
-		final Path path = Path.of( getClass().getResource( "./blob.txt" ).getPath() );
+		final Path path = Path.of( Thread.currentThread().getContextClassLoader()
+				.getResource( "org/hibernate/orm/test/envers/integration/blob/blob.txt" ).getPath() );
 		doInJPA( this::entityManagerFactory, entityManager -> {
-			try {
-				final Asset asset = new Asset();
-				asset.setFileName( "blob.txt" );
+			final Asset asset = new Asset();
+			asset.setFileName( "blob.txt" );
 
-				final InputStream stream = new BufferedInputStream( Files.newInputStream( path ) );
+			try (final InputStream stream = new BufferedInputStream( Files.newInputStream( path ) )) {
 				assertThat( stream.markSupported(), Matchers.is( true ) );
 
 				// We use the method readAllBytes instead of passing the raw stream to the proxy
@@ -70,6 +69,37 @@ public void initData() {
 				fail( "Failed to persist the entity" );
 			}
 		} );
+
+		try (final InputStream stream = new BufferedInputStream( Files.newInputStream( path ) )) {
+			doInJPA( this::entityManagerFactory, entityManager -> {
+				final Asset asset = new Asset();
+				asset.setFileName( "blob.txt" );
+
+				assertThat( stream.markSupported(), Matchers.is( true ) );
+
+				// We use the method readAllBytes instead of passing the raw stream to the proxy
+				// since this is the only guaranteed way that will work across all dialects in a
+				// deterministic way.  Postgres and Sybase will automatically close the stream
+				// after the blob has been written by the driver, which prevents Envers from
+				// then writing the contents of the stream to the audit table.
+				//
+				// If the driver and dialect are known not to close the input stream after the
+				// contents have been written by the driver, then it's safe to pass the stream
+				// here instead and the stream will be automatically marked and reset so that
+				// Envers can serialize the data after Hibernate has done so.  Dialects like
+				// H2, MySQL, Oracle, SQL Server work this way.
+				//
+				//
+				Blob blob = BlobProxy.generateProxy( stream, 200L );
+
+				asset.setData( blob );
+				entityManager.persist( asset );
+			} );
+		}
+		catch (Exception e) {
+			e.printStackTrace();
+			fail( "Failed to persist the entity" );
+		}
 	}
 
 	@Audited