示例#1
0
  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);
  }
}