EntityIsGenerated.java

 /*******************************************************************************
  * Copyright (c) 2016 Red Hat, Inc.
  * Distributed under license by Red Hat, Inc. All rights reserved.
  * This program is made available under the terms of the
  * Eclipse Public License v1.0 which accompanies this distribution,
  * and is available at http://www.eclipse.org/legal/epl-v10.html
  *
  * Contributor:
  *     Red Hat, Inc. - initial API and implementation
  ******************************************************************************/
package org.jboss.tools.hibernate.reddeer.condition;

import org.jboss.reddeer.common.condition.WaitCondition;
import org.jboss.reddeer.eclipse.jdt.ui.packageexplorer.PackageExplorer;

public class EntityIsGenerated implements WaitCondition{
	
	private String project;
	private PackageExplorer pe;
	private String[] entityPath;
	
	public EntityIsGenerated(String project, String... entityPath) {
		this.entityPath = entityPath;
		this.project = project;
		pe = new PackageExplorer();
		pe.open();
	}

	@Override
	public boolean test() {
		if(pe.getProject(project).containsItem(entityPath)){
			return true;
		} else {
			pe.getProject(project).refresh();
			return false;
		}
	}

	@Override
	public String description() {
		return "entity "+entityPath[entityPath.length-1]+" is generated";
	}

	@Override
	public String errorMessage() {
		return "entity "+entityPath[entityPath.length-1]+" was not generated";
	}
	
	

}