Hibernate
/
hibernate-orm
mirror of https://github.com/hibernate/hibernate-orm
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

HHH-14725 Fixed test

This commit is contained in:
Andrea Boriero 2024-12-11 12:04:12 +01:00 committed by Andrea Boriero
parent c7734e9b21
commit c475770d09
2 changed files with 65 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -4,9 +4,18 @@
*/ */
package org.hibernate.orm.test.envers.integration.blob; package org.hibernate.orm.test.envers.integration.blob;
import static org.hamcrest.MatcherAssert.assertThat; import jakarta.persistence.Entity;
import static org.hibernate.testing.transaction.TransactionUtil.doInJPA; import jakarta.persistence.GeneratedValue;
import static org.junit.Assert.fail; import jakarta.persistence.Id;
import org.hamcrest.Matchers;
import org.hibernate.dialect.PostgreSQLDialect;
import org.hibernate.dialect.SQLServerDialect;
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.hibernate.testing.SkipForDialect;
import org.junit.Test;
import java.io.BufferedInputStream; import java.io.BufferedInputStream;
import java.io.InputStream; import java.io.InputStream;
@ -14,16 +23,9 @@ import java.nio.file.Files;
import java.nio.file.Path; import java.nio.file.Path;
import java.sql.Blob; import java.sql.Blob;
import org.hamcrest.Matchers; import static org.hamcrest.MatcherAssert.assertThat;
import org.hibernate.engine.jdbc.proxy.BlobProxy; import static org.hibernate.testing.transaction.TransactionUtil.doInJPA;
import org.hibernate.envers.Audited; import static org.junit.Assert.fail;
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;
/** /**
* @author Chris Cranford * @author Chris Cranford
@ -32,19 +34,19 @@ public class BasicBlobTest extends BaseEnversJPAFunctionalTestCase {
@Override @Override
protected Class<?>[] getAnnotatedClasses() { protected Class<?>[] getAnnotatedClasses() {
return new Class<?>[] { Asset.class }; return new Class<?>[] {Asset.class};
} }
@Test @Test
@Priority(10) @Priority(10)
public void initData() { public void testGenerateProxyNoStream() {
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 -> { doInJPA( this::entityManagerFactory, entityManager -> {
try { final Asset asset = new Asset();
final Asset asset = new Asset(); asset.setFileName( "blob.txt" );
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 ) ); assertThat( stream.markSupported(), Matchers.is( true ) );
// We use the method readAllBytes instead of passing the raw stream to the proxy // We use the method readAllBytes instead of passing the raw stream to the proxy
@ -70,6 +72,49 @@ public class BasicBlobTest extends BaseEnversJPAFunctionalTestCase {
fail( "Failed to persist the entity" ); fail( "Failed to persist the entity" );
} }
} ); } );
}
@Test
@Priority(10)
@SkipForDialect(value = PostgreSQLDialect.class,
comment = "The driver closes the stream, so it cannot be reused by envers")
@SkipForDialect(value = SQLServerDialect.class,
comment = "The driver closes the stream, so it cannot be reused by envers")
public void testGenerateProxyStream() {
final Path path = Path.of( Thread.currentThread().getContextClassLoader()
.getResource( "org/hibernate/orm/test/envers/integration/blob/blob.txt" ).getPath() );
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, 9192L );
asset.setData( blob );
entityManager.persist( asset );
} );
}
catch (Exception e) {
e.printStackTrace();
fail( "Failed to persist the entity" );
}
} }
@Audited @Audited

0
hibernate-envers/src/test/java/org/hibernate/orm/test/envers/integration/blob/blob.txt → hibernate-envers/src/test/resources/org/hibernate/orm/test/envers/integration/blob/blob.txt
View File