HHH-11614 : Add test case using PostgreSQL Large Object functions in …

…native queries; add test case showing how varchar data can be changed to Long Object.

hibernate-core/src/test/java/org/hibernate/test/lob/LobStringTest.java

@@ -0,0 +1,233 @@
+/*
+ * Hibernate, Relational Persistence for Idiomatic Java
+ *
+ * License: GNU Lesser General Public License (LGPL), version 2.1 or later.
+ * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
+ */
+package org.hibernate.test.lob;
+
+import java.io.UnsupportedEncodingException;
+import java.sql.Clob;
+import java.sql.SQLException;
+import java.util.List;
+import javax.persistence.Column;
+import javax.persistence.Entity;
+import javax.persistence.GeneratedValue;
+import javax.persistence.Id;
+import javax.persistence.Lob;
+import javax.persistence.Table;
+
+import org.hibernate.dialect.PostgreSQL81Dialect;
+import org.hibernate.query.Query;
+
+import org.hibernate.testing.RequiresDialect;
+import org.hibernate.testing.TestForIssue;
+import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
+import org.junit.Test;
+
+import static org.hamcrest.core.Is.is;
+import static org.hibernate.testing.transaction.TransactionUtil.doInHibernate;
+import static org.junit.Assert.assertThat;
+import static org.junit.Assert.fail;
+
+/**
+ * @author Andrea Boriero
+ */
+@TestForIssue(jiraKey = "HHH-11477")
+@RequiresDialect(PostgreSQL81Dialect.class)
+public class LobStringTest extends BaseCoreFunctionalTestCase {
+
+	private static final int LONG_STRING_SIZE = 3999;
+
+	private final String value1 = buildRecursively( LONG_STRING_SIZE, 'x' );
+	private final String value2 = buildRecursively( LONG_STRING_SIZE, 'y' );
+
+	@Override
+	protected Class<?>[] getAnnotatedClasses() {
+		return new Class[] {TestEntity.class};
+	}
+
+	@Override
+	protected void prepareTest() throws Exception {
+		TestEntity entity = new TestEntity();
+		doInHibernate( this::sessionFactory, session -> {
+
+						   entity.setFirstLobField( value1 );
+						   entity.setSecondLobField( value2 );
+						   entity.setClobField( session.getLobHelper().createClob( value2 ) );
+						   session.save( entity );
+					   } );
+
+		doInHibernate( this::sessionFactory, session -> {
+						   final TestEntity testEntity = session.find( TestEntity.class, entity.getId() );
+						   assertThat( testEntity.getFirstLobField(), is( value1 ) );
+					   } );
+	}
+
+	@Test
+	@TestForIssue(jiraKey = "HHH-11477")
+	public void testHqlQuery() {
+		doInHibernate( this::sessionFactory, session -> {
+						   final Query query = session.createQuery( "from TestEntity" );
+
+						   final List<TestEntity> results = query.list();
+
+						   assertThat( results.size(), is( 1 ) );
+
+						   final TestEntity testEntity = results.get( 0 );
+						   assertThat( testEntity.getFirstLobField(), is( value1 ) );
+						   assertThat( testEntity.getSecondLobField(), is( value2 ) );
+						   final Clob clobField = testEntity.getClobField();
+						   try {
+
+							   assertThat( clobField.getSubString( 1, (int) clobField.length() ), is( value2 ) );
+						   }
+						   catch (SQLException e) {
+							   fail( e.getMessage() );
+						   }
+					   } );
+	}
+
+	@Test
+	@TestForIssue(jiraKey = "HHH-11477")
+	public void testUsingStringLobAnnotatedPropertyInNativeQuery() {
+		doInHibernate( this::sessionFactory, session -> {
+						   final List<TestEntity> results = session.createNativeQuery(
+								   "select te.* " +
+										   "from test_entity te " +
+										   "where lower(convert_from(lo_get(cast(te.firstLobField as oid)), 'UTF8')) LIKE :value", TestEntity.class )
+								   .setParameter( "value", value1 )
+								   .getResultList();
+
+						   assertThat( results.size(), is( 1 ) );
+
+						   final TestEntity testEntity = results.get( 0 );
+						   assertThat( testEntity.getFirstLobField(), is( value1 ) );
+						   assertThat( testEntity.getSecondLobField(), is( value2 ) );
+						   final Clob clobField = testEntity.getClobField();
+						   try {
+							   assertThat( clobField.getSubString( 1, (int) clobField.length() ), is( value2 ) );
+						   }
+						   catch (SQLException e) {
+							   fail( e.getMessage() );
+						   }
+					   } );
+	}
+
+	@Test
+	@TestForIssue(jiraKey = "HHH-11477")
+	public void testSelectStringLobAnnotatedInNativeQuery() {
+		doInHibernate( this::sessionFactory, session -> {
+						   final List<String> results = session.createNativeQuery(
+								   "select convert_from(lo_get(cast(te.secondLobField as oid)), 'UTF8') " +
+										   "from test_entity te " +
+										   "where lower(convert_from(lo_get(cast(te.firstLobField as oid)), 'UTF8')) LIKE :value" )
+								   .setParameter( "value", value1 )
+								   .list();
+
+						   assertThat( results.size(), is( 1 ) );
+
+						   assertThat( results.get( 0 ), is( value2 ) );
+					   } );
+	}
+
+	@Test
+	@TestForIssue(jiraKey = "HHH-11477")
+	public void testUsingLobPropertyInNativeQuery() {
+		doInHibernate( this::sessionFactory, session -> {
+						   final List<String> results = session.createNativeQuery(
+								   "select convert_from(lo_get(cast(te.secondLobField as oid)), 'UTF8') " +
+										   "from test_entity te " +
+										   "where lower(convert_from(lo_get(cast(te.clobField as oid)), 'UTF8')) LIKE :value" )
+								   .setParameter( "value", value2 )
+								   .list();
+
+						   assertThat( results.size(), is( 1 ) );
+
+						   assertThat( results.get( 0 ), is( value2 ) );
+					   } );
+	}
+
+	@Test
+	@TestForIssue(jiraKey = "HHH-11477")
+	public void testSelectClobPropertyInNativeQuery() {
+		doInHibernate( this::sessionFactory, session -> {
+						   final List<byte[]> results = session.createNativeQuery(
+								   "select lo_get(cast(te.clobField as oid)) " +
+										   "from test_entity te " +
+										   "where lower(convert_from(lo_get(cast(te.clobField as oid)), 'UTF8')) LIKE :value" )
+								   .setParameter( "value", value2 )
+								   .list();
+
+						   assertThat( results.size(), is( 1 ) );
+
+						   try {
+							   assertThat( new String( results.get( 0 ), "UTF8"), is( value2 ) );
+						   }
+						   catch (UnsupportedEncodingException e) {
+							   fail(e.getMessage());
+						   }
+					   } );
+	}
+
+	@Entity(name = "TestEntity")
+	@Table(name = "TEST_ENTITY")
+	public static class TestEntity {
+		@Id
+		@GeneratedValue
+		private long id;
+
+		@Lob
+		@Column(length = LONG_STRING_SIZE) //needed by HSQLDialect
+		String firstLobField;
+
+		@Lob
+		@Column(length = LONG_STRING_SIZE) //needed by HSQLDialect
+		String secondLobField;
+
+		@Lob
+		@Column(length = LONG_STRING_SIZE) //needed by HSQLDialect
+		Clob clobField;
+
+		public long getId() {
+			return id;
+		}
+
+		public String getFirstLobField() {
+			return firstLobField;
+		}
+
+		public void setFirstLobField(String firstLobField) {
+			this.firstLobField = firstLobField;
+		}
+
+		public String getSecondLobField() {
+			return secondLobField;
+		}
+
+		public void setSecondLobField(String secondLobField) {
+			this.secondLobField = secondLobField;
+		}
+
+		public Clob getClobField() {
+			return clobField;
+		}
+
+		public void setClobField(Clob clobField) {
+			this.clobField = clobField;
+		}
+	}
+
+	@Override
+	protected boolean isCleanupTestDataRequired() {
+		return true;
+	}
+
+	private String buildRecursively(int size, char baseChar) {
+		StringBuilder buff = new StringBuilder();
+		for ( int i = 0; i < size; i++ ) {
+			buff.append( baseChar );
+		}
+		return buff.toString();
+	}
+}