From 6cc292e9d33bbb38977268ea3231e404cce8ee70 Mon Sep 17 00:00:00 2001
From: Andrea Boriero <andrea@hibernate.org>
Date: Thu, 12 Sep 2024 11:30:28 +0200
Subject: [PATCH] HHH-18480 Add test for issue

---
 .../basic/BlobAttributeQueryUpdateTest.java   | 172 ++++++++++++++++++
 1 file changed, 172 insertions(+)
 create mode 100644 hibernate-core/src/test/java/org/hibernate/orm/test/mapping/basic/BlobAttributeQueryUpdateTest.java

diff --git a/hibernate-core/src/test/java/org/hibernate/orm/test/mapping/basic/BlobAttributeQueryUpdateTest.java b/hibernate-core/src/test/java/org/hibernate/orm/test/mapping/basic/BlobAttributeQueryUpdateTest.java
new file mode 100644
index 0000000000..05afcf6cf1
--- /dev/null
+++ b/hibernate-core/src/test/java/org/hibernate/orm/test/mapping/basic/BlobAttributeQueryUpdateTest.java
@@ -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;
+		}
+	}
+}