compiler.tests/feature-tests/as/ASFeatureTestsBase.java - royale-compiler - Git at Google

 /*
  *
  *  Licensed to the Apache Software Foundation (ASF) under one or more
  *  contributor license agreements.  See the NOTICE file distributed with
  *  this work for additional information regarding copyright ownership.
  *  The ASF licenses this file to You 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 as;

 import static org.hamcrest.core.Is.is;
 import static org.junit.Assert.assertThat;
 import static org.junit.Assert.fail;

 import java.io.BufferedWriter;
 import java.io.File;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
 import java.util.concurrent.*;

 import org.apache.flex.compiler.clients.MXMLC;
 import org.apache.flex.compiler.problems.ICompilerProblem;
 import org.apache.flex.utils.EnvProperties;
 import org.apache.flex.utils.FilenameNormalization;
 import org.apache.flex.utils.StringUtils;


 /**
  * Base class for AS feature tests which compile AS code with MXMLC and run it in the standalone Flash Player.
  * Copied and modified from MXMLFeatureTestsBase.java
  */
 public class ASFeatureTestsBase
 {
 	private static EnvProperties env = EnvProperties.initiate();

 	private static final String PLAYERGLOBAL_SWC = FilenameNormalization.normalize(env.FPSDK + "\\" + env.FPVER + "\\playerglobal.swc");

 	private static final String FRAMEWORK_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\libs\\framework.swc");
 	private static final String FRAMEWORK_RB_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\locale\\en_US\\framework_rb.swc");

 	private static final String RPC_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\libs\\rpc.swc");
 	private static final String RPC_RB_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\locale\\en_US\\rpc_rb.swc");

 	private static final String SPARK_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\libs\\spark.swc");
 	private static final String SPARK_RB_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\locale\\en_US\\spark_rb.swc");

 	private static final String NAMESPACE_2009 = "http://ns.adobe.com/mxml/2009";

 	private static final String MANIFEST_2009 = FilenameNormalization.normalize(env.SDK + "\\frameworks\\mxml-2009-manifest.xml");

     // The Ant script for compiler.tests copies a standalone player to the temp directory.
     private static final String FLASHPLAYER = FilenameNormalization.normalize(env.FDBG);

 	protected void compileAndRun(String source, boolean withFramework, boolean withRPC, boolean withSpark, String[] otherOptions)
 	{
 		System.out.println("Generating test:");

 		// Write the MXML into a temp file.
 		String tempDir = FilenameNormalization.normalize("temp");
 		File tempASFile = null;
 		try
 		{
 			tempASFile = File.createTempFile(getClass().getSimpleName(), ".as", new File(tempDir));
 			tempASFile.deleteOnExit();

 			BufferedWriter out = new BufferedWriter(new FileWriter(tempASFile));
 			String className = tempASFile.getName();
 			// chop off .as
 			className = className.substring(0, className.length() - 3);

 			source = source.replaceAll("%0", className);
 		    out.write(source);
 		    out.close();
 		}
 		catch (IOException e1)
 		{
 			e1.printStackTrace();
 			fail("Error generating test code");
 		}

 		// Build the list of SWCs to compile against on the library path.
 		List<String> swcs = new ArrayList<String>();
 		if (withFramework)
 		{
 			swcs.add(FRAMEWORK_SWC);
 			swcs.add(FRAMEWORK_RB_SWC);
 		}
 		if (withRPC)
 		{
 			swcs.add(RPC_SWC);
 			swcs.add(RPC_RB_SWC);
 		}
 		if (withSpark)
 		{
 			swcs.add(SPARK_SWC);
 			swcs.add(SPARK_RB_SWC);
 		}
 		String libraryPath = "-library-path=" + StringUtils.join(swcs.toArray(new String[swcs.size()]), ",");

 		List<String> args = new ArrayList<String>();
 		args.add("-external-library-path=" + PLAYERGLOBAL_SWC);
 		args.add(libraryPath);
 		args.add("-namespace=" + NAMESPACE_2009 + "," + MANIFEST_2009);
 		if (otherOptions != null)
 		{
 			Collections.addAll(args, otherOptions);
 		}
 		args.add(tempASFile.getAbsolutePath());

 		// Use MXMLC to compile the MXML file against playerglobal.swc and possibly other SWCs.
 		MXMLC mxmlc = new MXMLC();
 		StringBuffer cmdLine = new StringBuffer();
 		for(String arg : args) {
 			cmdLine.append(arg).append(" ");
 		}
 		System.out.println("Compiling test:\n" + cmdLine.toString());
 		int exitCode = mxmlc.mainNoExit(args.toArray(new String[args.size()]));

 		// Check that there were no compilation problems.
 		List<ICompilerProblem> problems = mxmlc.getProblems().getProblems();
 		StringBuilder sb = new StringBuilder("Unxpected compilation problems:\n");
 		for (ICompilerProblem problem : problems)
 		{
 			sb.append(problem.toString());
 			sb.append('\n');
 		}
 		assertThat(sb.toString(), exitCode, is(0));

 		// Check the existence of the flashplayer executable
 		File playerExecutable = new File(FLASHPLAYER);
 		if(!playerExecutable.isFile() || !playerExecutable.exists()) {
 			fail("The flashplayer executable " + FLASHPLAYER + " doesn't exist.");
 		}

 		// Run the SWF in the standalone player amd wait until the SWF calls System.exit().
 		String swf = FilenameNormalization.normalize(tempASFile.getAbsolutePath());
 		swf = swf.replace(".as", ".swf");
 		String[] runArgs = new String[] { FLASHPLAYER, swf };
 		try
 		{
 			System.out.println("Executing test:\n" + Arrays.toString(runArgs));
 			exitCode = executeCommandWithTimeout(runArgs, 20);
 		}
 		catch (Exception e)
 		{
 			e.printStackTrace();
 		}

 	    // Check that the runtime exit code was 0, meaning that no asserts failed.
 		assertThat(exitCode, is(0));
 	}

 	protected void compileAndRun(String source)
 	{
 		compileAndRun(source, false, false, false, null);
 	}

 	public static int executeCommandWithTimeout(String[] args, long timeoutInSeconds) throws Exception {
 		ExecutorService service = Executors.newSingleThreadExecutor();
 		Process process = Runtime.getRuntime().exec(args);
 		try {
 			Callable<Integer> call = new CallableProcess(process);
 			Future<Integer> future = service.submit(call);
 			return future.get(timeoutInSeconds, TimeUnit.SECONDS);
 		} catch (ExecutionException e) {
 			throw new Exception("Process failed to execute", e);
 		} catch (TimeoutException e) {
 			process.destroy();
 			throw new Exception("Process timed out", e);
 		} finally {
 			service.shutdown();
 		}
 	}

 	private static class CallableProcess implements Callable<Integer> {
 		private Process p;

 		public CallableProcess(Process process) {
 			p = process;
 		}

 		public Integer call() throws Exception {
 			return p.waitFor();
 		}
 	}

 	/**
 	 * Combines various code snippets to make a complete one-file MXML Sprite-based application.
 	 */
     protected String getAS(String[] imports, String[] declarations, String[] testCode, String[] extraCode)
     {
         String[] template = new String[]
         {
             "package {",
             "import flash.display.Sprite;",
             "import flash.events.Event;",
             "import flash.system.System;",
             "%1",
             "public class %0 extends flash.display.Sprite",
             "{",
             "    public function %0()",
             "    {",
             "        loaderInfo.addEventListener(flash.events.Event.INIT, initHandler);",
             "    }",
             "    %2",
             "    private function initHandler(e:flash.events.Event):void",
             "    {",
             "        %3",
             "        System.exit(0);",
             "    }",
             "    private function assertEqual(message:String, actualValue:*, expectedValue:*):void",
             "    {",
             "        if (actualValue !== expectedValue)",
             "        {",
             "            trace(message, actualValue, expectedValue);",
             "            System.exit(1);",
             "        }",
             "    }",
             "}",
             "}",
             "%4"
         };
         String source = StringUtils.join(template, "\n");
         source = source.replace("%1", StringUtils.join(imports, "\n"));
         source = source.replace("%2", StringUtils.join(declarations, "\n        "));
         source = source.replace("%3", StringUtils.join(testCode, "\n"));
         source = source.replace("%4", StringUtils.join(extraCode, "\n"));
         return source;
     }


 }
	/*
	*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You 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 as;

	import static org.hamcrest.core.Is.is;
	import static org.junit.Assert.assertThat;
	import static org.junit.Assert.fail;

	import java.io.BufferedWriter;
	import java.io.File;
	import java.io.FileWriter;
	import java.io.IOException;
	import java.util.ArrayList;
	import java.util.Arrays;
	import java.util.Collections;
	import java.util.List;
	import java.util.concurrent.*;

	import org.apache.flex.compiler.clients.MXMLC;
	import org.apache.flex.compiler.problems.ICompilerProblem;
	import org.apache.flex.utils.EnvProperties;
	import org.apache.flex.utils.FilenameNormalization;
	import org.apache.flex.utils.StringUtils;


	/**
	* Base class for AS feature tests which compile AS code with MXMLC and run it in the standalone Flash Player.
	* Copied and modified from MXMLFeatureTestsBase.java
	*/
	public class ASFeatureTestsBase
	{
	private static EnvProperties env = EnvProperties.initiate();

	private static final String PLAYERGLOBAL_SWC = FilenameNormalization.normalize(env.FPSDK + "\\" + env.FPVER + "\\playerglobal.swc");

	private static final String FRAMEWORK_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\libs\\framework.swc");
	private static final String FRAMEWORK_RB_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\locale\\en_US\\framework_rb.swc");

	private static final String RPC_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\libs\\rpc.swc");
	private static final String RPC_RB_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\locale\\en_US\\rpc_rb.swc");

	private static final String SPARK_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\libs\\spark.swc");
	private static final String SPARK_RB_SWC = FilenameNormalization.normalize(env.SDK + "\\frameworks\\locale\\en_US\\spark_rb.swc");

	private static final String NAMESPACE_2009 = "http://ns.adobe.com/mxml/2009";

	private static final String MANIFEST_2009 = FilenameNormalization.normalize(env.SDK + "\\frameworks\\mxml-2009-manifest.xml");

	// The Ant script for compiler.tests copies a standalone player to the temp directory.
	private static final String FLASHPLAYER = FilenameNormalization.normalize(env.FDBG);

	protected void compileAndRun(String source, boolean withFramework, boolean withRPC, boolean withSpark, String[] otherOptions)
	{
	System.out.println("Generating test:");

	// Write the MXML into a temp file.
	String tempDir = FilenameNormalization.normalize("temp");
	File tempASFile = null;
	try
	{
	tempASFile = File.createTempFile(getClass().getSimpleName(), ".as", new File(tempDir));
	tempASFile.deleteOnExit();

	BufferedWriter out = new BufferedWriter(new FileWriter(tempASFile));
	String className = tempASFile.getName();
	// chop off .as
	className = className.substring(0, className.length() - 3);

	source = source.replaceAll("%0", className);
	out.write(source);
	out.close();
	}
	catch (IOException e1)
	{
	e1.printStackTrace();
	fail("Error generating test code");
	}

	// Build the list of SWCs to compile against on the library path.
	List<String> swcs = new ArrayList<String>();
	if (withFramework)
	{
	swcs.add(FRAMEWORK_SWC);
	swcs.add(FRAMEWORK_RB_SWC);
	}
	if (withRPC)
	{
	swcs.add(RPC_SWC);
	swcs.add(RPC_RB_SWC);
	}
	if (withSpark)
	{
	swcs.add(SPARK_SWC);
	swcs.add(SPARK_RB_SWC);
	}
	String libraryPath = "-library-path=" + StringUtils.join(swcs.toArray(new String[swcs.size()]), ",");

	List<String> args = new ArrayList<String>();
	args.add("-external-library-path=" + PLAYERGLOBAL_SWC);
	args.add(libraryPath);
	args.add("-namespace=" + NAMESPACE_2009 + "," + MANIFEST_2009);
	if (otherOptions != null)
	{
	Collections.addAll(args, otherOptions);
	}
	args.add(tempASFile.getAbsolutePath());

	// Use MXMLC to compile the MXML file against playerglobal.swc and possibly other SWCs.
	MXMLC mxmlc = new MXMLC();
	StringBuffer cmdLine = new StringBuffer();
	for(String arg : args) {
	cmdLine.append(arg).append(" ");
	}
	System.out.println("Compiling test:\n" + cmdLine.toString());
	int exitCode = mxmlc.mainNoExit(args.toArray(new String[args.size()]));

	// Check that there were no compilation problems.
	List<ICompilerProblem> problems = mxmlc.getProblems().getProblems();
	StringBuilder sb = new StringBuilder("Unxpected compilation problems:\n");
	for (ICompilerProblem problem : problems)
	{
	sb.append(problem.toString());
	sb.append('\n');
	}
	assertThat(sb.toString(), exitCode, is(0));

	// Check the existence of the flashplayer executable
	File playerExecutable = new File(FLASHPLAYER);
	if(!playerExecutable.isFile() \|\| !playerExecutable.exists()) {
	fail("The flashplayer executable " + FLASHPLAYER + " doesn't exist.");
	}

	// Run the SWF in the standalone player amd wait until the SWF calls System.exit().
	String swf = FilenameNormalization.normalize(tempASFile.getAbsolutePath());
	swf = swf.replace(".as", ".swf");
	String[] runArgs = new String[] { FLASHPLAYER, swf };
	try
	{
	System.out.println("Executing test:\n" + Arrays.toString(runArgs));
	exitCode = executeCommandWithTimeout(runArgs, 20);
	}
	catch (Exception e)
	{
	e.printStackTrace();
	}

	// Check that the runtime exit code was 0, meaning that no asserts failed.
	assertThat(exitCode, is(0));
	}

	protected void compileAndRun(String source)
	{
	compileAndRun(source, false, false, false, null);
	}

	public static int executeCommandWithTimeout(String[] args, long timeoutInSeconds) throws Exception {
	ExecutorService service = Executors.newSingleThreadExecutor();
	Process process = Runtime.getRuntime().exec(args);
	try {
	Callable<Integer> call = new CallableProcess(process);
	Future<Integer> future = service.submit(call);
	return future.get(timeoutInSeconds, TimeUnit.SECONDS);
	} catch (ExecutionException e) {
	throw new Exception("Process failed to execute", e);
	} catch (TimeoutException e) {
	process.destroy();
	throw new Exception("Process timed out", e);
	} finally {
	service.shutdown();
	}
	}

	private static class CallableProcess implements Callable<Integer> {
	private Process p;

	public CallableProcess(Process process) {
	p = process;
	}

	public Integer call() throws Exception {
	return p.waitFor();
	}
	}

	/**
	* Combines various code snippets to make a complete one-file MXML Sprite-based application.
	*/
	protected String getAS(String[] imports, String[] declarations, String[] testCode, String[] extraCode)
	{
	String[] template = new String[]
	{
	"package {",
	"import flash.display.Sprite;",
	"import flash.events.Event;",
	"import flash.system.System;",
	"%1",
	"public class %0 extends flash.display.Sprite",
	"{",
	" public function %0()",
	" {",
	" loaderInfo.addEventListener(flash.events.Event.INIT, initHandler);",
	" }",
	" %2",
	" private function initHandler(e:flash.events.Event):void",
	" {",
	" %3",
	" System.exit(0);",
	" }",
	" private function assertEqual(message:String, actualValue:, expectedValue:):void",
	" {",
	" if (actualValue !== expectedValue)",
	" {",
	" trace(message, actualValue, expectedValue);",
	" System.exit(1);",
	" }",
	" }",
	"}",
	"}",
	"%4"
	};
	String source = StringUtils.join(template, "\n");
	source = source.replace("%1", StringUtils.join(imports, "\n"));
	source = source.replace("%2", StringUtils.join(declarations, "\n "));
	source = source.replace("%3", StringUtils.join(testCode, "\n"));
	source = source.replace("%4", StringUtils.join(extraCode, "\n"));
	return source;
	}


	}