private int run(String[] argv) {
if (!config.processArguments(argv)) {
return -1;
}
boolean success = false;
MessageContext mc = null;
ParseTreeNode compiledOutput = null;
try {
PluginMeta meta = new PluginMeta(makeEnvironment(config));
meta.setDebugMode(config.debugMode());
meta.setValijaMode(config.cajaMode());
PluginCompiler compiler = new PluginCompiler(BuildInfo.getInstance(), meta, mq);
mc = compiler.getMessageContext();
compiler.setCssSchema(config.getCssSchema(mq));
compiler.setHtmlSchema(config.getHtmlSchema(mq));
success = parseInputs(config.getInputUris(), compiler) && compiler.run();
if (success) {
compiledOutput = compiler.getJavascript();
}
} finally {
if (mc == null) {
mc = new MessageContext();
}
MessageLevel maxMessageLevel = dumpMessages(mq, mc, System.err);
success &= MessageLevel.ERROR.compareTo(maxMessageLevel) > 0;
}
if (success) {
writeFile(config.getOutputJsFile(), compiledOutput);
} else {
// Make sure there is no previous output file from a failed run.
config.getOutputJsFile().delete();
// If it wasn't there in the first place, or is not writable, that's OK,
// so ignore the return value.
}
return success ? 0 : -1;
}
/** * Tests the running the cajoler as a webservice * * @author [email protected] (Jasvir Nagra) */ public class CajolingServiceTest extends ServiceTestCase { private final String cajaBuildVersionParam = "&build-version=" + BuildInfo.getInstance().getBuildVersion(); @Override protected JSONObject requestGet(String queryString) throws Exception { return (JSONObject) json((String) super.requestGet(queryString)); } @Override protected JSONObject requestPost( String queryString, byte[] content, String contentType, String contentEncoding) throws Exception { return (JSONObject) json((String) super.requestPost(queryString, content, contentType, contentEncoding)); } protected void assertNoError(JSONObject result) { JSONArray messages = (JSONArray) result.get("messages"); for (int i = 0; i < messages.size(); i++) { JSONObject message = (JSONObject) messages.get(i); String levelName = (String) message.get("name"); assertFalse(levelName.contains("ERROR")); } } protected void assertError(JSONObject result) { JSONArray messages = (JSONArray) result.get("messages"); for (int i = 0; i < messages.size(); i++) { JSONObject message = (JSONObject) messages.get(i); String levelName = (String) message.get("name"); if (levelName.contains("ERROR")) { return; } } fail("Expected errors, but did not find any"); } protected void assertErrorMessage(JSONObject result, String substring) { JSONArray messages = (JSONArray) result.get("messages"); for (int i = 0; i < messages.size(); i++) { JSONObject message = (JSONObject) messages.get(i); String messageString = (String) message.get("message"); if (messageString.contains(substring)) { return; } } fail("Expected error message \"" + substring + "\" but did not find it"); } // Tests that, if the content at the "url=" parameter is an unsupported // type, the request is rejected. public final void testUnsupportedInputContentType() throws Exception { registerUri("http://foo/bar.vb", "zoicks()", "text/vbscript"); JSONObject result = requestGet("?url=http://foo/bar.vb&mime-type=text/javascript" + cajaBuildVersionParam); assertNull(result.get("js")); assertNull(result.get("html")); assertError(result); } // Tests that, if the content at the "url=" parameter is supported but is // not appropriate for the request being made, the request is rejected. public final void testNonMatchingInputContentType() throws Exception { registerUri("http://foo/bar.gif", "foo()", "text/javascript"); JSONObject result = requestGet("?url=http://foo/bar.gif&mime-type=image/*" + cajaBuildVersionParam); assertNull(result.get("js")); assertNull(result.get("html")); assertError(result); } // Tests requests for various mime types work. public final void testSupportedContentTypes() throws Exception { registerUri("http://foo/bar.js", "foo()", "text/javascript"); registerUri("http://foo/bar.html", "<b>Hello</b>", "text/html"); { JSONObject result = requestGet( "?url=http://foo/bar.js&input-mime-type=text/javascript" + cajaBuildVersionParam); assertNotNull(result.get("js")); assertNoError(result); } { JSONObject result = requestGet("?url=http://foo/bar.html&input-mime-type=text/html" + cajaBuildVersionParam); assertNotNull(result.get("js")); assertNotNull(result.get("html")); assertNoError(result); } } public final void testWrongBuildVersion() throws Exception { registerUri("http://foo/bar.js", "foo()", "text/javascript"); JSONObject result = requestGet( "?url=http://foo/bar.js&input-mime-type=text/javascript" + "&build-version=0000"); assertNull(result.get("js")); assertNull(result.get("html")); assertError(result); assertErrorMessage(result, "Build version error"); } public final void testWithoutBuildVersion() throws Exception { registerUri("http://foo/bar.js", "foo()", "text/javascript"); JSONObject result = requestGet("?url=http://foo/bar.js&input-mime-type=text/javascript"); assertNotNull(result.get("js")); assertNoError(result); } // Tests that POST-ing to the service works just as well as GET-ting from it. public final void testPost() throws Exception { JSONObject result = requestPost( "?url=http://foo/bar.html&input-mime-type=text/html&alt=json" + cajaBuildVersionParam, "<p>Foo bar</p>".getBytes("UTF-8"), "text/html", null); assertContainsIgnoreSpace((String) result.get("html"), "Foo bar"); assertNoError(result); } public final void testUnexpectedMimeType() throws Exception { registerUri("http://foo/bar.gif", "foo()", "text/javascript"); JSONObject result = requestGet("?url=http://foo/bar.gif&input-mime-type=image/*" + cajaBuildVersionParam); assertNull(result.get("js")); assertNull(result.get("html")); assertError(result); } public final void testEmptyContent() throws Exception { registerUri("http://foo/bar.html", "", "text/html"); byte[] byteData = {}; JSONObject result = requestPost( "?url=http://foo/bar.html&input-mime-type=text/html" + "&output-mime-type=application/json" + cajaBuildVersionParam, byteData, "text/html", null); assertTrue("Output missing 'html' key", result.containsKey("html")); assertTrue("Output missing 'javascript' key", result.containsKey("js")); assertEquals("", (String) result.get("html")); assertNoError(result); } }
