AbstractJSTests.java

/*******************************************************************************
 * Copyright (c) 2008, 2011 VMware Inc.
 * All rights reserved. This program and the accompanying materials
 * are 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
 *
 * Contributors:
 *   VMware Inc. - initial contribution
 *******************************************************************************/
package org.eclipse.virgo.management.console;

import java.io.FileReader;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;

import javax.script.ScriptException;

import junit.framework.Assert;

import org.eclipse.virgo.management.console.stubs.objects.Dollar;
import org.eclipse.virgo.management.console.stubs.objects.Util;
import org.eclipse.virgo.management.console.stubs.objects.Window;
import org.eclipse.virgo.management.console.stubs.types.Element;
import org.eclipse.virgo.management.console.stubs.types.Server;
import org.junit.After;
import org.junit.Before;
import org.mozilla.javascript.Context;
import org.mozilla.javascript.Function;
import org.mozilla.javascript.FunctionObject;
import org.mozilla.javascript.Scriptable;
import org.mozilla.javascript.ScriptableObject;

/**
 * 
 *
 */
public abstract class AbstractJSTests {
	
	protected Context context; 
	
	protected ScriptableObject scope;

	protected static String alertMsg;
	
	protected Util commonUtil = null;
	
	@Before
	public void setUp() throws ScriptException, IOException, IllegalAccessException, InstantiationException, InvocationTargetException, SecurityException, NoSuchMethodException{
		context = Context.enter();
		scope = context.initStandardObjects();

		//Create the browser environment
		ScriptableObject.defineClass(scope, Element.class);
		ScriptableObject.defineClass(scope, Server.class);
		ScriptableObject.putProperty(scope, "window", Context.javaToJS(new Window(), scope));
		
		Dollar.init(context, scope);
		FunctionObject dollarFunction = new FunctionObject("$", Dollar.class.getDeclaredMethod("dollar", ScriptableObject.class), scope);
		ScriptableObject.putProperty(scope, "$", dollarFunction);
		ScriptableObject.putProperty(dollarFunction, "ajax", new FunctionObject("ajax", Dollar.class.getDeclaredMethod("ajax", Scriptable.class), dollarFunction));
		ScriptableObject.putProperty(dollarFunction, "browser", Context.javaToJS(new Object(), scope));
		ScriptableObject.putProperty(dollarFunction, "each", new FunctionObject("each", Dollar.class.getDeclaredMethod("each", Scriptable.class, Function.class), dollarFunction));
		
		FunctionObject alertFunction = new FunctionObject("alert", AbstractJSTests.class.getDeclaredMethod("alert", String.class), scope);
		ScriptableObject.putProperty(scope, alertFunction.getFunctionName(), alertFunction);
	}

	@After
	public void closeDown(){
		Context.exit();
	}
	
	// callback from JS function doesn't work with non static field
	public static void alert(String msg){
		System.out.println(msg);
		alertMsg = msg;
	}
	
	protected final void readFile(String fileName) throws IOException{
		context.evaluateReader(scope, new FileReader(fileName), fileName, 0, null);
	}
	
	protected final void readString(String js) throws IOException{
		context.evaluateString(scope, js, "snippet", 0, null);
	}
	
	protected final void addCommonObjects(){
		commonUtil = new Util(context, scope);
		Object wrapped = Context.javaToJS(commonUtil, scope);
		ScriptableObject.putProperty(scope, "util", wrapped);
	}
	
	protected final void invokePageInit() throws ScriptException, NoSuchMethodException{
		Object fObj = scope.get("pageinit", scope);
		if (fObj instanceof Function) {
		    Function f = (Function)fObj;
		    f.call(context, scope, scope, Context.emptyArgs);
		} else {
			Assert.fail("pageinit function not found");
		}		
	}

}