StepExpectedResultDaoImpl.java

/*******************************************************************************
 * ATE, Automation Test Engine
 *
 * Copyright 2014, Montreal PROT, or individual 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 Montreal PROT.
 *
 * 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 org.bigtester.ate.model.data.dao;

import java.util.List;

import javax.persistence.TypedQuery;

import org.bigtester.ate.constant.ExceptionErrorCode;
import org.bigtester.ate.constant.ExceptionMessage;
import org.bigtester.ate.model.data.dbtable.StepErElementExistence;
import org.bigtester.ate.model.data.dbtable.StepErPageProperty;
import org.bigtester.ate.model.data.dbtable.StepExpectedResult;
import org.bigtester.ate.model.data.exception.RepeatTestDataException;
import org.bigtester.ate.model.data.exception.TestDataException;

// TODO: Auto-generated Javadoc
/**
 * This class StepExpectedResultDaoImpl defines ....
 * 
 * @author Peidong Hu
 *
 */
public class StepExpectedResultDaoImpl extends BaseDaoImpl {

	/**
	 * Gets the step expected results list.
	 *
	 * @param sERSetID
	 *            the s er set id
	 * @return the e rs
	 * @throws TestDataException
	 *             the test data exception
	 */
	public List<StepExpectedResult> getERs(String sERSetID)
			throws TestDataException {

		List<StepExpectedResult> sERs = (List<StepExpectedResult>) getDbEM()
				.createQuery(
						"select p from StepExpectedResult p where FirstTimeExecution= 'Yes' and p.stepERSetID = :stepERSetID",
						StepExpectedResult.class)
				.setParameter("stepERSetID", sERSetID)// NOPMD
				.getResultList();
		if (sERs.isEmpty()) {
			throw new TestDataException(ExceptionMessage.MSG_TESTDATA_NOTFOUND,
					ExceptionErrorCode.TESTDATA_NOTFOUND);
		} else {
			return sERs;
		}
	}

	/**
	 * Gets the er element existences.
	 *
	 * @param sERSetID
	 *            the s er set id
	 * @param repeatStepName
	 *            the repeat step name
	 * @param iteration
	 *            the iteration
	 * @return the er element existences
	 * @throws TestDataException
	 *             the test data exception
	 */
	public List<StepErElementExistence> getErElementExistences(String sERSetID,
			String repeatStepName, int iteration) throws RepeatTestDataException {
		String sql = "select p from StepErElementExistence p where repeatStepExternalLoopPath is null and FirstTimeExecution= 'No' and p.stepERSetID = :stepERSetID and p.repeatStepName=:repeatStepName and p.iteration=:iteration";
		TypedQuery<StepErElementExistence> query = getDbEM().createQuery(sql,
				StepErElementExistence.class);
		query.setParameter("stepERSetID", sERSetID);
		query.setParameter("repeatStepName", repeatStepName);
		query.setParameter("iteration", iteration);
		List<StepErElementExistence> sERs = (List<StepErElementExistence>) query
				.getResultList();
		if (sERs.isEmpty()) {
			throw new RepeatTestDataException(ExceptionMessage.MSG_TESTDATA_NOTFOUND,
					ExceptionErrorCode.TESTDATA_NOTFOUND, repeatStepName, "", iteration);
		} else {
			return sERs;
		}
	}

	/**
	 * Gets the er element existences.
	 *
	 * @param sERSetID the s er set id
	 * @return the er element existences
	 * @throws TestDataException the test data exception
	 */
	public List<StepErElementExistence> getErElementExistences(String sERSetID) throws TestDataException {
		String sql = "select p from StepErElementExistence p where FirstTimeExecution= 'Yes' and p.stepERSetID = :stepERSetID ";
		TypedQuery<StepErElementExistence> query = getDbEM().createQuery(sql,
				StepErElementExistence.class);
		query.setParameter("stepERSetID", sERSetID);
		List<StepErElementExistence> sERs = (List<StepErElementExistence>) query
				.getResultList();
		if (sERs.isEmpty()) {
			throw new TestDataException(ExceptionMessage.MSG_TESTDATA_NOTFOUND,
					ExceptionErrorCode.TESTDATA_NOTFOUND);
		} else {
			return sERs;
		}
	}
	/**
	 * Gets the er element existences.
	 *
	 * @param sERSetID the s er set id
	 * @param repeatStepName the repeat step name
	 * @param repeatStepExternalLoopPath the repeat step external loop path
	 * @param iteration the iteration
	 * @return the er element existences
	 * @throws TestDataException the test data exception
	 */
	public List<StepErElementExistence> getErElementExistences(String sERSetID,
			String repeatStepName, String repeatStepExternalLoopPath, int iteration)
			throws RepeatTestDataException {
		List<StepErElementExistence> retVal;
		if ("".equals(repeatStepExternalLoopPath))
			retVal = getErElementExistences(sERSetID, repeatStepName, iteration);
		else {
			String sql = "select p from StepErElementExistence p where repeatStepExternalLoopPath=:repeatStepExternalLoopPath and FirstTimeExecution= 'No' and p.stepERSetID = :stepERSetID and p.repeatStepName=:repeatStepName and p.iteration=:iteration";
			TypedQuery<StepErElementExistence> query = getDbEM().createQuery(
					sql, StepErElementExistence.class);
			query.setParameter("repeatStepExternalLoopPath", repeatStepExternalLoopPath);
			query.setParameter("stepERSetID", sERSetID);
			query.setParameter("repeatStepName", repeatStepName);
			query.setParameter("iteration", iteration);
			retVal = (List<StepErElementExistence>) query
					.getResultList();
			if (retVal.isEmpty()) {
				throw new RepeatTestDataException(
						ExceptionMessage.MSG_TESTDATA_NOTFOUND,
						ExceptionErrorCode.TESTDATA_NOTFOUND, repeatStepName, repeatStepExternalLoopPath, iteration );
			}
		}

		return retVal;

	}

//	/**
//	 * Gets the er element existences.
//	 *
//	 * @param sERSetID
//	 *            the s er set id
//	 * @return the er element existences
//	 * @throws TestDataException
//	 *             the test data exception
//	 */
//	public List<StepErElementExistence> getErElementExistences(String sERSetID)
//			throws TestDataException {
//
//		List<StepErElementExistence> sERs = (List<StepErElementExistence>) getDbEM()
//				.createQuery(
//						"select p from StepErElementExistence p where p.stepERSetID = :stepERSetID",
//						StepErElementExistence.class)
//				.setParameter("stepERSetID", sERSetID).getResultList();
//		if (sERs.isEmpty()) {
//			throw new TestDataException(ExceptionMessage.MSG_TESTDATA_NOTFOUND,
//					ExceptionErrorCode.TESTDATA_NOTFOUND);
//		} else {
//			return sERs;
//		}
//	}

	/**
	 * Gets the step expected results list.
	 *
	 * @param sERSetID
	 *            the s er set id
	 * @return the e rs
	 * @throws TestDataException
	 *             the test data exception
	 */
	public List<StepErPageProperty> getErPageProperties(String sERSetID)
			throws TestDataException {

		List<StepErPageProperty> sERs = (List<StepErPageProperty>) getDbEM()
				.createQuery(
						"select p from StepErPageProperty p where p.stepERSetID = :stepERSetID",
						StepErPageProperty.class)
				.setParameter("stepERSetID", sERSetID).getResultList();
		if (sERs.isEmpty()) {
			throw new TestDataException(ExceptionMessage.MSG_TESTDATA_NOTFOUND,
					ExceptionErrorCode.TESTDATA_NOTFOUND);
		} else {
			return sERs;
		}
	}

}