Package org.hibernate.test.hql

Source Code of org.hibernate.test.hql.FunctionNameAsColumnTest

/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* Copyright (c) 2010, Red Hat Middleware LLC or third-party contributors as
* indicated by the @author tags or express copyright attribution
* statements applied by the authors.  All third-party contributions are
* distributed under license by Red Hat Middleware LLC.
*
* This copyrighted material is made available to anyone wishing to use, modify,
* copy, or redistribute it subject to the terms and conditions of the GNU
* Lesser General Public License, as published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
* or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
* for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this distribution; if not, write to:
* Free Software Foundation, Inc.
* 51 Franklin Street, Fifth Floor
* Boston, MA  02110-1301  USA
*
*/
package org.hibernate.test.hql;

import java.util.List;

import junit.framework.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import org.hibernate.FetchMode;
import org.hibernate.Hibernate;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.cfg.Configuration;
import org.hibernate.cfg.Environment;
import org.hibernate.criterion.Restrictions;
import org.hibernate.dialect.function.SQLFunction;
import org.hibernate.engine.SessionFactoryImplementor;
import org.hibernate.testing.junit.functional.FunctionalTestCase;
import org.hibernate.testing.junit.functional.FunctionalTestClassTestSuite;

/**
* Tests HQL and Criteria queries using DB columns having the same name as registerd functions.
*
* @author Gail Badner
*/
public class FunctionNameAsColumnTest  extends FunctionalTestCase {

  private static final Logger log = LoggerFactory.getLogger( FunctionNameAsColumnTest.class );

  public FunctionNameAsColumnTest(String name) {
    super( name );
  }

  public String[] getMappings() {
    return new String[] {
        "hql/FunctionNamesAsColumns.hbm.xml"
    };
  }

  public void configure(Configuration cfg) {
    super.configure( cfg );
    cfg.setProperty( Environment.USE_QUERY_CACHE, "false" );
  }

  public static Test suite() {
    return new FunctionalTestClassTestSuite( FunctionNameAsColumnTest.class );
  }

  public void testGetOneColumnSameNameAsArgFunctionHQL() {
    Session s = openSession();
    Transaction t = s.beginTransaction();
    EntityWithArgFunctionAsColumn e = new EntityWithArgFunctionAsColumn();
    e.setLower( 3 );
    e.setUpper( " abc " );
    s.persist( e );
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    e = ( EntityWithArgFunctionAsColumn ) s.createQuery( "from EntityWithArgFunctionAsColumn" ).uniqueResult();
    assertEquals( 3, e.getLower() );
    assertEquals( " abc ", e.getUpper() );
    t.commit();
    s.close();

    cleanup();
  }

  public void testGetOneColumnSameNameAsArgFunctionCriteria() {
    Session s = openSession();
    Transaction t = s.beginTransaction();
    EntityWithArgFunctionAsColumn e = new EntityWithArgFunctionAsColumn();
    e.setLower( 3 );
    e.setUpper( " abc " );
    s.persist( e );
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    e = ( EntityWithArgFunctionAsColumn ) s.createCriteria( EntityWithArgFunctionAsColumn.class ).uniqueResult();
    assertEquals( 3, e.getLower() );
    assertEquals( " abc ", e.getUpper() );
    t.commit();
    s.close();

    cleanup();
  }

  public void testGetMultiColumnSameNameAsArgFunctionHQL() {
    Session s = openSession();
    Transaction t = s.beginTransaction();
    EntityWithArgFunctionAsColumn e1 = new EntityWithArgFunctionAsColumn();
    e1.setLower( 3 );
    e1.setUpper( " abc " );
    EntityWithArgFunctionAsColumn e2 = new EntityWithArgFunctionAsColumn();
    e2.setLower( 999 );
    e2.setUpper( " xyz " );
    EntityWithFunctionAsColumnHolder holder1 = new EntityWithFunctionAsColumnHolder();
    holder1.getEntityWithArgFunctionAsColumns().add( e1 );
    EntityWithFunctionAsColumnHolder holder2 = new EntityWithFunctionAsColumnHolder();
    holder2.getEntityWithArgFunctionAsColumns().add( e2 );
    holder1.setNextHolder( holder2 );
    s.save( holder1 );
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    holder1 = ( EntityWithFunctionAsColumnHolder ) s.createQuery(
        "from EntityWithFunctionAsColumnHolder h left join fetch h.entityWithArgFunctionAsColumns " +
            "left join fetch h.nextHolder left join fetch h.nextHolder.entityWithArgFunctionAsColumns " +
            "where h.nextHolder is not null" )
        .uniqueResult();
    assertTrue( Hibernate.isInitialized( holder1.getEntityWithArgFunctionAsColumns() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder().getEntityWithArgFunctionAsColumns() ) );
    assertEquals( 1, holder1.getEntityWithArgFunctionAsColumns().size() );
    e1 = ( EntityWithArgFunctionAsColumn ) holder1.getEntityWithArgFunctionAsColumns().iterator().next();
    assertEquals( 3, e1.getLower() );
    assertEquals( " abc ", e1.getUpper() );
    assertEquals( 1, holder1.getNextHolder().getEntityWithArgFunctionAsColumns().size() );
    e2 = ( EntityWithArgFunctionAsColumn ) ( holder1.getNextHolder() ).getEntityWithArgFunctionAsColumns().iterator().next();
    assertEquals( 999, e2.getLower() );
    assertEquals( " xyz ", e2.getUpper() );
    t.commit();
    s.close();

    cleanup();
  }

  public void testGetMultiColumnSameNameAsArgFunctionCriteria() {
    Session s = openSession();
    Transaction t = s.beginTransaction();
    EntityWithArgFunctionAsColumn e1 = new EntityWithArgFunctionAsColumn();
    e1.setLower( 3 );
    e1.setUpper( " abc " );
    EntityWithArgFunctionAsColumn e2 = new EntityWithArgFunctionAsColumn();
    e2.setLower( 999 );
    e2.setUpper( " xyz " );
    EntityWithFunctionAsColumnHolder holder1 = new EntityWithFunctionAsColumnHolder();
    holder1.getEntityWithArgFunctionAsColumns().add( e1 );
    EntityWithFunctionAsColumnHolder holder2 = new EntityWithFunctionAsColumnHolder();
    holder2.getEntityWithArgFunctionAsColumns().add( e2 );
    holder1.setNextHolder( holder2 );
    s.save( holder1 );
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    holder1 = ( EntityWithFunctionAsColumnHolder ) s.createCriteria( EntityWithFunctionAsColumnHolder.class )
        .add( Restrictions.isNotNull( "nextHolder" ))
        .setFetchMode( "entityWithArgFunctionAsColumns", FetchMode.JOIN )
        .setFetchMode( "nextHolder", FetchMode.JOIN )
        .setFetchMode( "nextHolder.entityWithArgFunctionAsColumns", FetchMode.JOIN )
        .uniqueResult();
    assertTrue( Hibernate.isInitialized( holder1.getEntityWithArgFunctionAsColumns() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder().getEntityWithArgFunctionAsColumns() ) );
    assertEquals( 1, holder1.getEntityWithArgFunctionAsColumns().size() );
    e1 = ( EntityWithArgFunctionAsColumn ) holder1.getEntityWithArgFunctionAsColumns().iterator().next();
    assertEquals( 3, e1.getLower() );
    assertEquals( " abc ", e1.getUpper() );
    assertEquals( 1, holder1.getNextHolder().getEntityWithArgFunctionAsColumns().size() );
    e2 = ( EntityWithArgFunctionAsColumn ) ( holder1.getNextHolder() ).getEntityWithArgFunctionAsColumns().iterator().next();
    assertEquals( 999, e2.getLower() );
    assertEquals( " xyz ", e2.getUpper() );
    t.commit();
    s.close();

    cleanup();
  }

  public void testGetMultiColumnSameNameAsNoArgFunctionHQL() throws Exception {
    SQLFunction function =
        ( ( SessionFactoryImplementor ) getSessions() ).getSqlFunctionRegistry().findSQLFunction( "current_date" );
    if ( function == null || function.hasParenthesesIfNoArguments() ) {
      reportSkip( "current_date reuires ()", "tests noarg function that does not require ()" );
      return;
    }

    Session s = openSession();
    Transaction t = s.beginTransaction();
    EntityWithNoArgFunctionAsColumn e1 = new EntityWithNoArgFunctionAsColumn();
    e1.setCurrentDate( "blah blah blah" );
    EntityWithNoArgFunctionAsColumn e2 = new EntityWithNoArgFunctionAsColumn();
    e2.setCurrentDate( "yadda yadda yadda" );
    EntityWithFunctionAsColumnHolder holder1 = new EntityWithFunctionAsColumnHolder();
    holder1.getEntityWithNoArgFunctionAsColumns().add( e1 );
    EntityWithFunctionAsColumnHolder holder2 = new EntityWithFunctionAsColumnHolder();
    holder2.getEntityWithNoArgFunctionAsColumns().add( e2 );
    holder1.setNextHolder( holder2 );
    s.save( holder1 );
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    holder1 = ( EntityWithFunctionAsColumnHolder ) s.createQuery(
        "from EntityWithFunctionAsColumnHolder h left join fetch h.entityWithNoArgFunctionAsColumns " +
            "left join fetch h.nextHolder left join fetch h.nextHolder.entityWithNoArgFunctionAsColumns " +
            "where h.nextHolder is not null" )
        .uniqueResult();
    assertTrue( Hibernate.isInitialized( holder1.getEntityWithNoArgFunctionAsColumns() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder().getEntityWithNoArgFunctionAsColumns() ) );
    assertEquals( 1, holder1.getEntityWithNoArgFunctionAsColumns().size() );
    t.commit();
    s.close();

    e1 = ( EntityWithNoArgFunctionAsColumn ) holder1.getEntityWithNoArgFunctionAsColumns().iterator().next();
    assertEquals( "blah blah blah", e1.getCurrentDate() );
    assertEquals( 1, holder1.getNextHolder().getEntityWithNoArgFunctionAsColumns().size() );
    e2 = ( EntityWithNoArgFunctionAsColumn ) ( holder1.getNextHolder() ).getEntityWithNoArgFunctionAsColumns().iterator().next();
    assertEquals( "yadda yadda yadda", e2.getCurrentDate() );

    cleanup();
  }

  public void testGetMultiColumnSameNameAsNoArgFunctionCriteria() {
    SQLFunction function =
        ( ( SessionFactoryImplementor ) getSessions() ).getSqlFunctionRegistry().findSQLFunction( "current_date" );
    if ( function == null || function.hasParenthesesIfNoArguments() ) {
      reportSkip( "current_date reuires ()", "tests noarg function that does not require ()" );
      return;
    }

    Session s = openSession();
    Transaction t = s.beginTransaction();
    EntityWithNoArgFunctionAsColumn e1 = new EntityWithNoArgFunctionAsColumn();
    e1.setCurrentDate( "blah blah blah" );
    EntityWithNoArgFunctionAsColumn e2 = new EntityWithNoArgFunctionAsColumn();
    e2.setCurrentDate( "yadda yadda yadda" );
    EntityWithFunctionAsColumnHolder holder1 = new EntityWithFunctionAsColumnHolder();
    holder1.getEntityWithNoArgFunctionAsColumns().add( e1 );
    EntityWithFunctionAsColumnHolder holder2 = new EntityWithFunctionAsColumnHolder();
    holder2.getEntityWithNoArgFunctionAsColumns().add( e2 );
    holder1.setNextHolder( holder2 );
    s.save( holder1 );
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    holder1 = ( EntityWithFunctionAsColumnHolder ) s.createCriteria( EntityWithFunctionAsColumnHolder.class )
        .add( Restrictions.isNotNull( "nextHolder" ))
        .setFetchMode( "entityWithNoArgFunctionAsColumns", FetchMode.JOIN )
        .setFetchMode( "nextHolder", FetchMode.JOIN )
        .setFetchMode( "nextHolder.entityWithNoArgFunctionAsColumns", FetchMode.JOIN )
        .uniqueResult();
    assertTrue( Hibernate.isInitialized( holder1.getEntityWithNoArgFunctionAsColumns() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder() ) );
    assertTrue( Hibernate.isInitialized( holder1.getNextHolder().getEntityWithNoArgFunctionAsColumns() ) );
    assertEquals( 1, holder1.getEntityWithNoArgFunctionAsColumns().size() );
    e1 = ( EntityWithNoArgFunctionAsColumn ) holder1.getEntityWithNoArgFunctionAsColumns().iterator().next();
    assertEquals( "blah blah blah", e1.getCurrentDate() );
    assertEquals( 1, holder1.getNextHolder().getEntityWithNoArgFunctionAsColumns().size() );
    e2 = ( EntityWithNoArgFunctionAsColumn ) ( holder1.getNextHolder() ).getEntityWithNoArgFunctionAsColumns().iterator().next();
    assertEquals( "yadda yadda yadda", e2.getCurrentDate() );
    t.commit();
    s.close();

    cleanup();
  }

  public void testNoArgFcnAndColumnSameNameAsNoArgFunctionHQL() {
    SQLFunction function =
        ( ( SessionFactoryImplementor ) getSessions() ).getSqlFunctionRegistry().findSQLFunction( "current_date" );
    if ( function == null || function.hasParenthesesIfNoArguments() ) {
      reportSkip( "current_date reuires ()", "tests noarg function that does not require ()" );
      return;
    }

    Session s = openSession();
    Transaction t = s.beginTransaction();
    EntityWithNoArgFunctionAsColumn e1 = new EntityWithNoArgFunctionAsColumn();
    e1.setCurrentDate( "blah blah blah" );
    EntityWithNoArgFunctionAsColumn e2 = new EntityWithNoArgFunctionAsColumn();
    e2.setCurrentDate( "yadda yadda yadda" );
    EntityWithFunctionAsColumnHolder holder1 = new EntityWithFunctionAsColumnHolder();
    holder1.getEntityWithNoArgFunctionAsColumns().add( e1 );
    EntityWithFunctionAsColumnHolder holder2 = new EntityWithFunctionAsColumnHolder();
    holder2.getEntityWithNoArgFunctionAsColumns().add( e2 );
    holder1.setNextHolder( holder2 );
    s.save( holder1 );
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    List results = s.createQuery(
            "select str(current_date), currentDate from EntityWithNoArgFunctionAsColumn"
        ).list();
    assertEquals( 2, results.size() );
    assertEquals( ( ( Object[] ) results.get( 0 ) )[ 0 ], ( ( Object[] ) results.get( 1 ) )[ 0 ] );
    assertTrue( ! ( ( Object[] ) results.get( 0 ) )[ 0 ].equals( ( ( Object[] ) results.get( 0 ) )[ 1 ] ) );
    assertTrue( ! ( ( Object[] ) results.get( 1 ) )[ 0 ].equals( ( ( Object[] ) results.get( 1 ) )[ 1 ] ) );
    assertTrue( ( ( Object[] ) results.get( 0 ) )[ 1 ].equals( e1.getCurrentDate() ) ||
             ( ( Object[] ) results.get( 0 ) )[ 1 ].equals( e2.getCurrentDate() ) );
    assertTrue( ( ( Object[] ) results.get( 1 ) )[ 1 ].equals( e1.getCurrentDate() ) ||
             ( ( Object[] ) results.get( 1 ) )[ 1 ].equals( e2.getCurrentDate() ) );
    assertFalse( ( ( Object[] ) results.get( 0 ) )[ 1 ].equals( ( ( Object[] ) results.get( 1 ) )[ 1 ] ) );
    t.commit();
    s.close();

    cleanup();
  }

  private void cleanup() {
    Session s = openSession();
    Transaction t = s.beginTransaction();
    s.createQuery( "delete from EntityWithArgFunctionAsColumn" ).executeUpdate();
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    s.createQuery( "delete from EntityWithNoArgFunctionAsColumn" ).executeUpdate();
    t.commit();
    s.close();

    s = openSession();
    t = s.beginTransaction();
    s.createQuery( "delete from EntityWithFunctionAsColumnHolder where nextHolder is not null" ).executeUpdate();
    s.createQuery( "delete from EntityWithFunctionAsColumnHolder" ).executeUpdate();
    t.commit();
    s.close();
  }
}
TOP

Related Classes of org.hibernate.test.hql.FunctionNameAsColumnTest

TOP
Copyright © 2018 www.massapi.com. All rights reserved.
All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.