//$Id: EJBQLTest.java,v 1.19 2005/04/13 06:24:29 oneovthafew Exp $package org.hibernate.test.hql;import antlr.RecognitionException;import antlr.TokenStreamException;import antlr.collections.AST;import junit.framework.Test;import junit.framework.TestSuite;import org.hibernate.engine.SessionFactoryImplementor;import org.hibernate.hql.QueryTranslator;import org.hibernate.hql.QueryTranslatorFactory;import org.hibernate.hql.ast.ASTQueryTranslatorFactory;import org.hibernate.hql.ast.HqlParser;import org.hibernate.test.TestCase;import java.io.ByteArrayOutputStream;import java.io.PrintStream;import java.util.Collections;/** * @author <a href="mailto:alex@jboss.org">Alexey Loubyansky</a> */public class EJBQLTest		extends TestCase {	public EJBQLTest(String x) {		super( x );	}	/**	 * SELECT OBJECT(identifier)	 */	public void testSelectObjectClause() throws Exception {		//parse("select object(m) from Model m");		assertEjbqlEqualsHql( "select object(m) from Model m", "from Model m" );	}	/**	 * IN(collection_valued_path) identifier	 */	public void testCollectionMemberDeclaration() throws Exception {		String hql = "select o from Animal a inner join a.offspring o";		String ejbql = "select object(o) from Animal a, in(a.offspring) o";		//parse(hql);		//parse(ejbql);		assertEjbqlEqualsHql( ejbql, hql );	}	/**	 * collection_valued_path IS [NOT] EMPTY	 */	public void testIsEmpty() throws Exception {		//String hql = "from Animal a where not exists (from a.offspring)";		String hql = "from Animal a where not exists elements(a.offspring)";		String ejbql = "select object(a) from Animal a where a.offspring is empty";		//parse(hql);		//parse(ejbql);		assertEjbqlEqualsHql(ejbql, hql);				hql = "from Animal a where exists (from a.mother.father.offspring)";		ejbql = "select object(a) from Animal a where a.mother.father.offspring is not empty";		assertEjbqlEqualsHql( ejbql, hql );	}	/**	 * [NOT] MEMBER OF	 */	public void testMemberOf() throws Exception {		String hql = "from Animal a where a.mother in (from a.offspring)";		//String hql = "from Animal a where a.mother in elements(a.offspring)";		String ejbql = "select object(a) from Animal a where a.mother member of a.offspring";		//parse(hql);		//parse(ejbql);		assertEjbqlEqualsHql( ejbql, hql );		hql = "from Animal a where a.mother not in (from a.offspring)";		//hql = "from Animal a where a.mother not in elements(a.offspring)";		ejbql = "select object(a) from Animal a where a.mother not member of a.offspring";		//parse(hql);		//parse(ejbql);		assertEjbqlEqualsHql( ejbql, hql );	}	/**	 * Various functions.	 * Tests just parsing for now which means it doesn't guarantee that the generated SQL is as expected or even valid.	 */	public void testEJBQLFunctions() throws Exception {		String hql = "select object(a) from Animal a where a.description = concat('1', concat('2','3'), '4'||'5')||0";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "from Animal a where substring(a.description, 1, 3) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select substring(a.description, 1, 3) from Animal a";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "from Animal a where lower(a.description) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select lower(a.description) from Animal a";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "from Animal a where upper(a.description) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select upper(a.description) from Animal a";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "from Animal a where length(a.description) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select length(a.description) from Animal a";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "from Animal a where locate(a.description, 'abc', 2) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select locate(a.description, :p1, 2) from Animal a";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where trim(trailing '_' from a.description) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select trim(trailing '_' from a.description) from Animal a";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where trim(leading '_' from a.description) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where trim(both a.description) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where trim(a.description) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where abs(a.bodyWeight) = sqrt(a.bodyWeight)";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where mod(a.bodyWeight, a.mother.bodyWeight) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where BIT_LENGTH(a.bodyWeight) = :p1";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select BIT_LENGTH(a.bodyWeight) from Animal a";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where CURRENT_DATE = :p1 or CURRENT_TIME = :p2 or CURRENT_TIMESTAMP = :p3";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		// todo the following is not supported		//hql = "select CURRENT_DATE, CURRENT_TIME, CURRENT_TIMESTAMP from Animal a";		//parse(hql, true);		//System.out.println("sql: " + toSql(hql));		hql = "select object(a) from Animal a where a.bodyWeight like '%a%'";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where a.bodyWeight not like '%a%'";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );		hql = "select object(a) from Animal a where a.bodyWeight like '%a%' escape '%'";		parse( hql, false );		System.out.println( "sql: " + toSql( hql ) );	}	public void testTrueFalse() throws Exception {		assertEjbqlEqualsHql( "from Human h where h.pregnant is true", "from Human h where h.pregnant = true" );		assertEjbqlEqualsHql( "from Human h where h.pregnant is false", "from Human h where h.pregnant = false" );		assertEjbqlEqualsHql( "from Human h where not(h.pregnant is true)", "from Human h where not( h.pregnant=true )" );	}	// Private	private void assertEjbqlEqualsHql(String ejbql, String hql) {		SessionFactoryImplementor factory = getSessionFactoryImplementor();		QueryTranslatorFactory ast = new ASTQueryTranslatorFactory();		QueryTranslator queryTranslator = ast.createQueryTranslator( hql, Collections.EMPTY_MAP, factory );		queryTranslator.compile( Collections.EMPTY_MAP, true );		String hqlSql = queryTranslator.getSQLString();		queryTranslator = ast.createQueryTranslator( ejbql, Collections.EMPTY_MAP, factory );		queryTranslator.compile( Collections.EMPTY_MAP, true );		String ejbqlSql = queryTranslator.getSQLString();		assertEquals( hqlSql, ejbqlSql );	}	private void assertEjbqlEqualsSql(String sql, String hql) {		SessionFactoryImplementor factory = getSessionFactoryImplementor();		QueryTranslatorFactory ast = new ASTQueryTranslatorFactory();		QueryTranslator queryTranslator = ast.createQueryTranslator( hql, Collections.EMPTY_MAP, factory );		queryTranslator.compile( Collections.EMPTY_MAP, true );		assertEquals( sql, queryTranslator.getSQLString() );	}	private void parse(String input, boolean logging) throws RecognitionException, TokenStreamException {		if ( logging ) {			System.out.println( "input: ->" + input + "<-" );		}		HqlParser parser = HqlParser.getInstance( input );		parser.setFilter( false );		parser.statement();		AST ast = parser.getAST();		if ( logging ) {			System.out.println( "AST  :  " + ast.toStringTree() + "" );			ByteArrayOutputStream baos = new ByteArrayOutputStream();			parser.showAst( ast, new PrintStream( baos ) );			System.out.println( baos.toString() );		}		assertEquals( "At least one error occurred during parsing!", 0, parser.getParseErrorHandler().getErrorCount() );	}	private String toSql(String hql) {		SessionFactoryImplementor factory = getSessionFactoryImplementor();		QueryTranslatorFactory ast = new ASTQueryTranslatorFactory();		QueryTranslator queryTranslator = ast.createQueryTranslator( hql, Collections.EMPTY_MAP, factory );		queryTranslator.compile( Collections.EMPTY_MAP, true );		return queryTranslator.getSQLString();	}	private SessionFactoryImplementor getSessionFactoryImplementor() {		SessionFactoryImplementor factory = ( SessionFactoryImplementor ) getSessions();		if ( factory == null ) {			throw new NullPointerException( "Unable to create factory!" );		}		return factory;	}	protected String[] getMappings() {		return new String[]{			"hql/Animal.hbm.xml",			"batchfetch/ProductLine.hbm.xml",			"cid/Customer.hbm.xml",			"cid/Order.hbm.xml",			"cid/LineItem.hbm.xml",			"cid/Product.hbm.xml",			"legacy/Glarch.hbm.xml",			"legacy/Fee.hbm.xml",			"legacy/Qux.hbm.xml",			"legacy/Fum.hbm.xml",			"legacy/Holder.hbm.xml",			"legacy/One.hbm.xml",			"legacy/FooBar.hbm.xml",			"legacy/Many.hbm.xml",			"legacy/Baz.hbm.xml",			"legacy/Simple.hbm.xml",			"legacy/Middle.hbm.xml",			"legacy/Category.hbm.xml",			"legacy/Multi.hbm.xml",			"legacy/Commento.hbm.xml",			"legacy/Marelo.hbm.xml",			"compositeelement/Parent.hbm.xml",			"legacy/Container.hbm.xml",		};	}	public static Test suite() {		return new TestSuite( EJBQLTest.class );	}}