com.google.j2cl.junit.apt.TemplateWriter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of junit-processor Show documentation
The newest version!
/*
 * Copyright 2016 Google Inc.
 *
 * Licensed 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 com.google.j2cl.junit.apt;

import static java.util.stream.Collectors.joining;

import com.google.common.collect.ImmutableMap;
import com.google.common.collect.ImmutableSet;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.List;
import javax.annotation.processing.Filer;
import javax.lang.model.element.Element;
import javax.tools.StandardLocation;

class TemplateWriter {
  private static final ImmutableSet VALID_PLATFORMS =
      ImmutableSet.of("CLOSURE", "WASM", "J2KT-JVM", "J2KT-NATIVE");

  private final List testSummary = new ArrayList<>();
  private final VelocityRenderer velocityRenderer = new VelocityRenderer(getClass());
  private final ErrorReporter errorReporter;
  private final Filer filer;
  private final String testPlatform;

  public TemplateWriter(ErrorReporter errorReporter, Filer filer, String testPlatform) {
    this.errorReporter = errorReporter;
    this.filer = filer;
    this.testPlatform = testPlatform;
    if (!VALID_PLATFORMS.contains(testPlatform)) {
      errorReporter.report(ErrorMessage.UNSUPPORTED_PLATFORM, testPlatform);
    }
  }

  public void writeSummary() {
    if (testPlatform.startsWith("J2KT")) {
      return;
    }
    if (testSummary.isEmpty()) {
      errorReporter.report(ErrorMessage.NO_TEST);
      return;
    }

    try {
      writeResource(
          "test_summary.json",
          testSummary.stream().collect(joining("\",\"", "{\"tests\": [\"", "\"]}")));
    } catch (Exception e) {
      errorReporter.report(ErrorMessage.CANNOT_WRITE_RESOURCE, exceptionToString(e));
    }
  }

  public void writeTestClass(TestClass testClass) {
    String testSuiteFileName = testClass.qualifiedName().replace('.', '/');
    if (testPlatform.startsWith("J2KT")) {
      try {
        String mergedKotlinTemplate = mergeTemplate(testClass, "KotlinJUnitAdapter.vm");
        writeResource(testSuiteFileName + ".kt", mergedKotlinTemplate);
      } catch (Exception e) {
        errorReporter.report(ErrorMessage.CANNOT_WRITE_RESOURCE, exceptionToString(e));
      }
      return;
    }

    testSummary.add(testSuiteFileName + ".js");
    String jsSuite = "JsSuite.vm";
    String jsUnitAdapter = "JsUnitAdapter.vm";
    if (testPlatform.equals("WASM")) {
      jsSuite = "WasmJsSuite.vm";
      jsUnitAdapter = "WasmJsUnitAdapter.vm";
    }
    try {
      String mergedJsTemplate = mergeTemplate(testClass, jsSuite);
      writeResource(testSuiteFileName + ".testsuite", mergedJsTemplate);
      String mergedJavaTemplate = mergeTemplate(testClass, jsUnitAdapter);
      writeClass(testClass.jsUnitAdapterQualifiedClassName(), mergedJavaTemplate);
    } catch (Exception e) {
      errorReporter.report(ErrorMessage.CANNOT_WRITE_RESOURCE, exceptionToString(e));
    }
  }

  public void handleTestSuiteFile(TestClass testClass, List actualTestClasses) {
    // Test suites rely upon JUnit infrastructure which is only supported for the JVM.
    if (!testPlatform.equals("J2KT-JVM")) {
      return;
    }

    String testSuiteFileName = testClass.qualifiedName().replace('.', '/');
    ImmutableMap velocityContext =
        ImmutableMap.of(
            "testClass", testClass,
            "actualTestClasses", actualTestClasses);
    try {
      String renderedTemplate =
          velocityRenderer.renderTemplate("J2ktJvmJUnitTestSuiteAdapter.vm", velocityContext);
      writeResource(testSuiteFileName + "_Adapter.kt", renderedTemplate);
    } catch (IOException ex) {
      errorReporter.report(ErrorMessage.CANNOT_WRITE_RESOURCE, exceptionToString(ex));
    }
  }

  private String mergeTemplate(TestClass testClass, String template) throws IOException {
    return velocityRenderer.renderTemplate(template, ImmutableMap.of("testClass", testClass));
  }

  private void writeResource(String qualifiedName, String content) throws IOException {
    try (Writer writer =
        filer
            .createResource(StandardLocation.SOURCE_OUTPUT, "", qualifiedName, new Element[0])
            .openWriter()) {
      writer.write(content);
    }
  }

  private void writeClass(String qualifiedName, String content) throws IOException {
    try (Writer writer = filer.createSourceFile(qualifiedName, new Element[0]).openWriter()) {
      writer.write(content);
    }
  }

  private static String exceptionToString(Exception e) {
    StringWriter stringWriter = new StringWriter();
    PrintWriter printWriter = new PrintWriter(stringWriter);
    printWriter.println("Exception: " + e.getMessage());
    e.printStackTrace(printWriter);
    return stringWriter.toString();
  }
}