TemplateParser.java

package junit.extensions.eclipse.quick.mac.growl.internal;

import org.eclipse.jdt.junit.model.ITestRunSession;

import static junit.extensions.eclipse.quick.mac.growl.internal.TemplateKey.*;

class TemplateParser {

	private String template;
	private TestCounter counter = new TestCounter();
	
	public TemplateParser() {
	}

	String pickupTestClassAndMethod(String testName){
		if(testName == null) return null;
		if(testName.indexOf('.') != 0){
			String[] split = testName.split("\\."); //$NON-NLS-1$
			if(split.length > 2){
				return split[split.length - 2] + "." + split[split.length - 1]; //$NON-NLS-1$
			}
		}
		return testName;
	}

	public String parseTemplate(ITestRunSession session) {
		String result = null;
		result = replaceResult(session,template);
		result = replaceCount(session,result);
		result = replaceName(session, result);
		return result;
	}


	private String replaceCount(ITestRunSession session, String result) {
		counter.count(session);
		result = result.replaceAll(key(TOTAL_COUNT), String.valueOf(counter.getTotalTests()));
		result = result.replaceAll(key(OK_COUNT), String.valueOf(counter.getOKTests()));
		result = result.replaceAll(key(FAIL_COUNT), String.valueOf(counter.getFailureTests()));
//		result = result.replaceAll(key(IGNORE_COUNT), String.valueOf(counter.getIgnoreTests()));
		result = result.replaceAll(key(ERROR_COUNT), String.valueOf(counter.getErrorTests()));
		return result;
	}

	private String replaceResult(ITestRunSession session,String target) {
		String result = target.replaceAll(key(RESULT_COUNT), session.getTestResult(true).toString());
		return result;
	}


	private String replaceName(ITestRunSession session, String target) {
		String testName = session.getTestRunName();
		target = target.replaceAll(key(NAME_COUNT), pickupTestClassAndMethod(testName));
		return target;
	}
	
	private String key(TemplateKey key){
		return key.regexKey();
	}
	
	public void setTemplate(String template) {
		this.template = template;
	}
	
	void setCounter(TestCounter counter) {
		this.counter = counter;
	}
	
}