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

HHH-18480 Add test for issue

This commit is contained in:
Andrea Boriero 2024-09-12 11:30:28 +02:00 committed by Andrea Boriero
parent 4b33d0d067
commit 6cc292e9d3
1 changed files with 172 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

172
hibernate-core/src/test/java/org/hibernate/orm/test/mapping/basic/BlobAttributeQueryUpdateTest.java Normal file
View File

@ -0,0 +1,172 @@
package org.hibernate.orm.test.mapping.basic;
import java.nio.charset.StandardCharsets;
import java.sql.Blob;
import java.sql.SQLException;
import org.hibernate.LobHelper;
import org.hibernate.engine.jdbc.BlobProxy;
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.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import jakarta.persistence.Entity;
import jakarta.persistence.Id;
import jakarta.persistence.TypedQuery;
import static org.assertj.core.api.AssertionsForInterfaceTypes.assertThat;
@DomainModel(
annotatedClasses = {
BlobAttributeQueryUpdateTest.TestEntity.class
}
)
@SessionFactory
@JiraKey("HHH-18480")
public class BlobAttributeQueryUpdateTest {
private static final byte[] INITIAL_BYTES = "initial".getBytes( StandardCharsets.UTF_8 );
private static final byte[] UPDATED_BYTES_1 = "update1".getBytes( StandardCharsets.UTF_8 );
private static final byte[] UPDATED_BYTES_2 = "update2".getBytes( StandardCharsets.UTF_8 );
@BeforeEach
public void setup(SessionFactoryScope scope) {
scope.inTransaction(
session -> {
LobHelper lobHelper = session.getLobHelper();
TestEntity testEntity = new TestEntity(
1,
"test",
lobHelper.createBlob( INITIAL_BYTES )
);
session.persist( testEntity );
}
);
}
@AfterEach
public void tearDown(SessionFactoryScope scope) {
scope.inTransaction(
session ->
session.createMutationQuery( "delete from TestEntity" ).executeUpdate()
);
}
@Test
public void testUpdateUsingBlobProxy(SessionFactoryScope scope) {
scope.inTransaction( session -> {
TestEntity testEntity = session.find( TestEntity.class, 1 );
Blob blobValue1 = BlobProxy.generateProxy( UPDATED_BYTES_1 );
testEntity.setBlobValue( blobValue1 );
} );
scope.inTransaction( session -> {
TestEntity testEntity = session.find( TestEntity.class, 1 );
checkBlobValue( testEntity, UPDATED_BYTES_1 );
} );
scope.inTransaction(
session -> {
TypedQuery<?> query = session.createQuery(
"UPDATE TestEntity b SET b.blobValue = :blobValue WHERE b.id = :id",
null
);
query.setParameter( "id", 1 );
Blob value = BlobProxy.generateProxy( UPDATED_BYTES_2 );
query.setParameter( "blobValue", value );
query.executeUpdate();
}
);
scope.inTransaction(
session -> {
TestEntity testEntity = session.find( TestEntity.class, 1 );
checkBlobValue( testEntity, UPDATED_BYTES_2 );
}
);
}
@Test
public void testUpdateUsingLobHelper(SessionFactoryScope scope) {
scope.inTransaction( session -> {
LobHelper lobHelper = session.getLobHelper();
TestEntity testEntity = session.find( TestEntity.class, 1 );
Blob blobValue1 = lobHelper.createBlob( UPDATED_BYTES_1 );
testEntity.setBlobValue( blobValue1 );
} );
scope.inTransaction( session -> {
TestEntity testEntity = session.find( TestEntity.class, 1 );
checkBlobValue( testEntity, UPDATED_BYTES_1 );
} );
scope.inTransaction(
session -> {
LobHelper lobHelper = session.getLobHelper();
TypedQuery<?> query = session.createQuery(
"UPDATE TestEntity b SET b.blobValue = :blobValue WHERE b.id = :id",
null
);
query.setParameter( "id", 1 );
Blob value = lobHelper.createBlob( UPDATED_BYTES_2 );
query.setParameter( "blobValue", value );
query.executeUpdate();
}
);
scope.inTransaction(
session -> {
TestEntity testEntity = session.find( TestEntity.class, 1 );
checkBlobValue( testEntity, UPDATED_BYTES_2 );
}
);
}
private static void checkBlobValue(TestEntity testEntity, byte[] expectedValue) {
try {
assertThat( new String( testEntity.getBlobValue()
.getBytes( 1, (int) testEntity.getBlobValue().length() ) ) )
.isEqualTo( new String( expectedValue ) );
}
catch (SQLException e) {
throw new RuntimeException( e );
}
}
@Entity(name = "TestEntity")
public static class TestEntity {
@Id
private Integer id;
private String name;
private Blob blobValue;
public TestEntity() {
}
public TestEntity(Integer id, String name, Blob blobValue) {
this.id = id;
this.name = name;
this.blobValue = blobValue;
}
public Integer getId() {
return id;
}
public Blob getBlobValue() {
return blobValue;
}
public void setBlobValue(Blob blobValue) {
this.blobValue = blobValue;
}
}
}