/*
 * Copyright 2002-2008 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
*/
package anyframe.core.hibernate.criteria;

import java.util.List;
import java.util.Map;

import org.hibernate.Criteria;
import org.hibernate.criterion.MatchMode;
import org.hibernate.criterion.ProjectionList;
import org.hibernate.criterion.Projections;
import org.hibernate.criterion.Restrictions;
import org.hibernate.transform.AliasToBeanResultTransformer;

import anyframe.core.hibernate.AbstractConfigurationalTransactionalTest;
import anyframe.core.hibernate.SetUpInitData;
import anyframe.sample.model.bidirection.Movie;

/**
 * TestCase Name : HibernateCriteriaWithDefinedResult<br>
 * <br>
 * [Description] : Criteria를 이용한 조회 결과를 별도 정의한 객체 형태로 전달받을 수 있다.<br>
 * [Main Flow]
 * <ul>
 * <li>#-1 Positive Case : Relation 관계에 놓여 있는 두개의 테이블을 대상으로 Criteria 이용한 조회 결과를
 * Movie 객체 형태로 전달받는다.</li>
 * <li>#-2 Positive Case : Relation 관계에 놓여 있는 두개의 테이블을 대상으로 Criteria 이용한 조회 결과를
 * Map 형태로 전달받는다.</li>
 * </ul>
 * 
 * @author SoYon Lim
 */
public class HibernateCriteriaWithDefinedResult extends
		AbstractConfigurationalTransactionalTest {
	protected String getHibernateConfigLocation() {
		return "hibernateconfig/hibernate.cfg.xml";
	}

	/**
	 * [Flow #-1] Positive Case : Relation 관계에 놓여 있는 두개의 테이블을 대상으로 Criteria 이용한
	 * 조회 결과를 Movie 객체 형태로 전달받는다.
	 * 
	 * @throws Exception
	 *             throws exception which is from hibernate
	 */
	public void testFindMoviesByCategory() throws Exception {
		// 1. insert init data
		SetUpInitData.initializeData(session);

		// 2. execute criteria
		Criteria movieCriteria = session.createCriteria(Movie.class);
		ProjectionList projectionList = Projections.projectionList();
		projectionList.add(Projections.id().as("movieId"));
		projectionList.add(Projections.property("title").as("title"));
		projectionList.add(Projections.property("director").as("director"));
		projectionList.add(Projections.property("category.categoryName").as(
				"categoryName"));
		projectionList.add(Projections.property("country.countryName").as(
				"countryName"));
		movieCriteria.setProjection(projectionList);
		movieCriteria.setResultTransformer(new AliasToBeanResultTransformer(
				Movie.class));

		Criteria categoryCriteria = movieCriteria.createCriteria("categories",
				"category");
		Criteria countryCriteria = movieCriteria.createCriteria("country",
				"country");
		categoryCriteria.add(Restrictions.eq("categoryName", "Romantic"));
		countryCriteria.add(Restrictions.like("countryName", "",
				MatchMode.ANYWHERE));

		List movieList = movieCriteria.list();

		// 3. assert result - movie
		assertEquals("fail to match the size of movie list.", 2, movieList
				.size());

		Movie movie1 = (Movie) movieList.get(0);
		assertEquals("fail to match a movie title.", "My Sassy Girl", movie1
				.getTitle());
		assertEquals("fail to match a movie director.", "Jaeyong Gwak", movie1
				.getDirector());
		assertEquals("fail to match a movie category.", "Romantic", movie1
				.getCategoryName());
		assertEquals("fail to match a country of the movie.", "Korea", movie1
				.getCountryName());

		Movie movie2 = (Movie) movieList.get(1);
		assertEquals("fail to match a movie title.", "My Little Bride", movie2
				.getTitle());
		assertEquals("fail to match a movie director.", "Hojun Kim", movie2
				.getDirector());
		assertEquals("fail to match a movie category.", "Romantic", movie2
				.getCategoryName());
		assertEquals("fail to match a country of the movie.", "Korea", movie2
				.getCountryName());
	}

	/**
	 * [Flow #-2] Positive Case : Relation 관계에 놓여 있는 두개의 테이블을 대상으로 Criteria를 이용한
	 * 조회 결과를 Map 형태로 전달받는다.
	 * 
	 * @throws Exception
	 *             throws exception which is from hibernate
	 */
	public void testFindMovieMapByCategory() throws Exception {
		// 1. insert init data
		SetUpInitData.initializeData(session);

		// 2. execute criteria
		Criteria movieCriteria = session.createCriteria(Movie.class);
		ProjectionList projectionList = Projections.projectionList();
		projectionList.add(Projections.id().as("movieId"));
		projectionList.add(Projections.property("title").as("title"));
		projectionList.add(Projections.property("director").as("director"));
		projectionList.add(Projections.property("category.categoryName").as(
				"categoryName"));
		projectionList.add(Projections.property("country.countryName").as(
				"countryName"));
		movieCriteria.setProjection(projectionList);
		movieCriteria.setResultTransformer(Criteria.ALIAS_TO_ENTITY_MAP);

		Criteria categoryCriteria = movieCriteria.createCriteria("categories",
				"category");
		Criteria countryCriteria = movieCriteria.createCriteria("country",
				"country");
		categoryCriteria.add(Restrictions.eq("categoryName", "Romantic"));
		countryCriteria.add(Restrictions.like("countryName", "",
				MatchMode.ANYWHERE));

		List movieList = movieCriteria.list();

		// 3. assert result - movie
		assertEquals("fail to match the size of movie list.", 2, movieList
				.size());

		Map movie1 = (Map) movieList.get(0);
		assertEquals("fail to match a movie title.", "My Sassy Girl", movie1
				.get("title"));
		assertEquals("fail to match a movie director.", "Jaeyong Gwak", movie1
				.get("director"));
		assertEquals("fail to match a movie category.", "Romantic", movie1
				.get("categoryName"));
		assertEquals("fail to match a country of the movie.", "Korea", movie1
				.get("countryName"));

		Map movie2 = (Map) movieList.get(1);
		assertEquals("fail to match a movie title.", "My Little Bride", movie2
				.get("title"));
		assertEquals("fail to match a movie director.", "Hojun Kim", movie2
				.get("director"));
		assertEquals("fail to match a movie category.", "Romantic", movie2
				.get("categoryName"));
		assertEquals("fail to match a country of the movie.", "Korea", movie2
				.get("countryName"));
	}
}