[ggj][infra][3/5] feat: implement update golden bazel rules for dummy test

dependencies.properties

@@ -32,5 +32,6 @@ maven.org_threeten_threetenbp=org.threeten:threetenbp:1.3.3
 
 # Testing.
 maven.junit_junit=junit:junit:4.13
+maven.org_hamcrest_hamcrest_core=org.hamcrest:hamcrest-core:1.3
 maven.org_mockito_mockito_core=org.mockito:mockito-core:2.21.0
 maven.com_google_truth_truth=com.google.truth:truth:0.42

rules_bazel/java/java_diff_test.bzl

@@ -0,0 +1,85 @@
+def _junit_output_impl(ctx):
+    test_class_name = ctx.attr.test_class_name
+    inputs = ctx.files.srcs
+    output = ctx.outputs.output
+    test_runner = ctx.executable.test_runner
+
+    command = """
+    mkdir local_tmp  
+    TEST_OUTPUT_HOME="$(pwd)/local_tmp" \
+    {test_runner_path} $@ &&
+    cd local_tmp 
+    zip -r ../{output} .
+        """.format(
+            test_runner_path = test_runner.path,
+            output=output.path,
+        )
+    ctx.actions.run_shell(
+        inputs = inputs,
+        outputs = [output],
+        arguments = [test_class_name],
+        tools = [test_runner],
+        command = command,
+    )
+
+junit_output_zip = rule(
+    attrs = {
+        "test_class_name": attr.string(mandatory=True),
+        "srcs": attr.label_list(
+            allow_files = True,
+            mandatory = True,
+        ),
+        "test_runner": attr.label(
+            mandatory = True,
+            executable = True,
+            cfg = "host",
+        ),
+        "output_suffix": attr.string(mandatory = False, default = ".zip"),
+    },
+    outputs = {
+        "output": "%{name}%{output_suffix}",
+    },
+    implementation = _junit_output_impl,  
+)
+
+def _overwrite_golden_impl(ctx):
+    script_content = """\
+    #!/bin/bash
+    cd ${{BUILD_WORKSPACE_DIRECTORY}}
+    unzip -ao {unit_test_results} -d src/test/java
+    """.format(
+        unit_test_results = ctx.file.unit_test_results.path,
+    )
+    ctx.actions.write(
+        output = ctx.outputs.bin,
+        content = script_content,
+        is_executable = True,
+    )
+    return [DefaultInfo(executable = ctx.outputs.bin)]
+
+
+overwrite_golden = rule(
+    attrs = {
+        "unit_test_results": attr.label(
+            mandatory = True,
+            allow_single_file = True),
+    },
+    outputs = {
+        "bin": "%{name}.sh",
+    },
+    executable = True,
+    implementation = _overwrite_golden_impl,
+)
+
+def update_golden(name, test_class_name, test_runner, srcs):
+    jnuit_output_name = "%s_output" % name
+    junit_output_zip(
+        name = jnuit_output_name,
+        test_class_name = test_class_name,
+        test_runner = test_runner,
+        srcs = srcs,
+    )
+    overwrite_golden(
+        name = name,
+        unit_test_results = ":%s" % jnuit_output_name
+    )

src/test/java/com/google/api/generator/gapic/dummy/BUILD.bazel

@@ -1,3 +1,5 @@
+load("//:rules_bazel/java/java_diff_test.bzl", "update_golden")
+
 package(default_visibility = ["//visibility:public"])
 
 TESTS = [
@@ -21,3 +23,39 @@ filegroup(
         "@junit_junit//jar",
     ],
 ) for test_name in TESTS]
+
+# JUnit runner binary, this is used to generate test output for updating goldens files.
+java_binary(
+    name = "junit_runner_binary",
+    srcs = [
+        ":dummy_files",
+        "//src/test/java/com/google/api/generator/test/framework:framework_files",
+    ],
+    data = ["//src/test/java/com/google/api/generator/gapic/dummy/goldens:goldens_files"],
+    jvm_flags = ["-Xmx512m"],
+    main_class = "com.google.api.generator.test.framework.SingleJUnitTestRunner",
+    visibility = ["//visibility:public"],
+    deps = [
+        "//src/main/java/com/google/api/generator/engine/ast",
+        "//src/main/java/com/google/api/generator/engine/writer",
+        "//src/test/java/com/google/api/generator/test/framework",
+        "@io_github_java_diff_utils//jar",
+        "@junit_junit//jar",
+        "@org_hamcrest_hamcrest_core//jar",
+    ],
+)
+
+TEST_CLASS_NAME = "com.google.api.generator.gapic.dummy.FileDiffInfraDummyTest"
+
+# Run `bazel run src/test/java/com/google/api/generator/gapic/dummy:FileDiffInfraDummyTest.update`
+# to update goldens as expected generated code.
+update_golden(
+    name = "FileDiffInfraDummyTest.update",
+    srcs = [
+        ":dummy_files",
+        "//src/test/java/com/google/api/generator/gapic/dummy/goldens:goldens_files",
+        "//src/test/java/com/google/api/generator/test/framework:framework_files",
+    ],
+    test_class_name = TEST_CLASS_NAME,
+    test_runner = "//src/test/java/com/google/api/generator/gapic/dummy:junit_runner_binary",
+)

src/test/java/com/google/api/generator/gapic/dummy/FileDiffInfraDummyTest.java

@@ -21,6 +21,7 @@
 import com.google.api.generator.engine.ast.ScopeNode;
 import com.google.api.generator.engine.writer.JavaWriterVisitor;
 import com.google.api.generator.test.framework.Assert;
+import com.google.api.generator.test.framework.Utils;
 import java.nio.file.Path;
 import java.nio.file.Paths;
 import java.util.Arrays;
@@ -36,6 +37,9 @@ public class FileDiffInfraDummyTest {
   // TODO(xiaozhenliu): remove this test class once the file-diff infra is in place and well-tested.
   private static final String GOLDENFILES_DIRECTORY =
       "src/test/java/com/google/api/generator/gapic/dummy/goldens/";
+  private static final String GOLDENFILES_SIMPLE_CLASS = "FileDiffInfraDummyTestSimpleClass.golden";
+  private static final String GOLDENFILES_CLASS_WITH_HEADER =
+      "FileDiffInfraDummyTestClassWithHeader.golden";
 
   @Test
   public void simpleClass() {
@@ -51,8 +55,8 @@ public void simpleClass() {
             .build();
     JavaWriterVisitor visitor = new JavaWriterVisitor();
     classDef.accept(visitor);
-    Path goldenFilePath =
-        Paths.get(GOLDENFILES_DIRECTORY, "FileDiffInfraDummyTestSimpleClass.golden");
+    Utils.saveCodegenToFile(this.getClass(), GOLDENFILES_SIMPLE_CLASS, visitor.write());
+    Path goldenFilePath = Paths.get(GOLDENFILES_DIRECTORY, GOLDENFILES_SIMPLE_CLASS);
     Assert.assertCodeEquals(goldenFilePath, visitor.write());
   }
 
@@ -69,8 +73,9 @@ public void classWithHeader() {
             .build();
     JavaWriterVisitor visitor = new JavaWriterVisitor();
     classDef.accept(visitor);
-    Path goldenFilePath =
-        Paths.get(GOLDENFILES_DIRECTORY, "FileDiffInfraDummyTestClassWithHeader.golden");
+    // Save the generated code to a file for updating goldens if needed.
+    Utils.saveCodegenToFile(this.getClass(), GOLDENFILES_CLASS_WITH_HEADER, visitor.write());
+    Path goldenFilePath = Paths.get(GOLDENFILES_DIRECTORY, GOLDENFILES_CLASS_WITH_HEADER);
     Assert.assertCodeEquals(goldenFilePath, visitor.write());
   }
 

src/test/java/com/google/api/generator/test/framework/BUILD.bazel

@@ -10,8 +10,10 @@ java_library(
     srcs = [
         ":framework_files",
     ],
+    data = ["//src/test/java/com/google/api/generator/gapic/dummy/goldens:goldens_files"],
     deps = [
         "@io_github_java_diff_utils//jar",
         "@junit_junit//jar",
+        "@org_hamcrest_hamcrest_core//jar",
     ],
 )

src/test/java/com/google/api/generator/test/framework/SingleJUnitTestRunner.java

@@ -0,0 +1,53 @@
+// Copyright 2020 Google LLC
+//
+// 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.api.generator.test.framework;
+
+import org.junit.runner.JUnitCore;
+import org.junit.runner.Request;
+import org.junit.runner.Result;
+
+public class SingleJUnitTestRunner {
+  // SingleJUnitTestRunner runs a JUnit test whose name is passed through `args`
+  // when `bazel run //:testTarget.update` for updating goldens files.
+  public static void main(String... args) {
+    if (args.length < 1) {
+      throw new MissingRequiredArgException("Missing the JUnit class name argument.");
+    }
+    String className = args[0];
+    Class clazz = null;
+    try {
+      clazz = Class.forName(className);
+    } catch (ClassNotFoundException e) {
+      throw new JUnitClassNotFoundException(
+          String.format("JUnit test class %s is not found.", className));
+    }
+    Result result = new JUnitCore().run(Request.aClass(clazz));
+    if (!result.wasSuccessful()) {
+      System.out.println("Tests have failures: " + result.getFailures());
+    }
+  }
+
+  private static class JUnitClassNotFoundException extends RuntimeException {
+    public JUnitClassNotFoundException(String errorMessage) {
+      super(errorMessage);
+    }
+  }
+
+  private static class MissingRequiredArgException extends RuntimeException {
+    public MissingRequiredArgException(String errorMessage) {
+      super(errorMessage);
+    }
+  }
+}

src/test/java/com/google/api/generator/test/framework/Utils.java

@@ -0,0 +1,56 @@
+// Copyright 2020 Google LLC
+//
+// 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.api.generator.test.framework;
+
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.nio.file.Path;
+import java.nio.file.Paths;
+
+public class Utils {
+
+  public static void saveCodegenToFile(Class clazz, String fileName, String codegen) {
+    // This system environment variable `TEST_OUTPUT_HOME` is used to specify a folder
+    // which contains generated output from JUnit test.
+    // It will be set when running `bazel run testTarget.update` command.
+    String testOutputHome = System.getenv("TEST_OUTPUT_HOME");
+    // For example: com/google/api/generator/gapic/dummy/goldens/
+    String relativeGoldenDir = getTestoutGoldenDir(clazz);
+    Path testOutputDir = Paths.get(testOutputHome, relativeGoldenDir);
+    testOutputDir.toFile().mkdirs();
+    try {
+      File testOutputFile = Paths.get(testOutputHome, relativeGoldenDir, fileName).toFile();
+      FileWriter myWriter = new FileWriter(testOutputFile);
+      myWriter.write(codegen);
+      myWriter.flush();
+      myWriter.close();
+    } catch (IOException e) {
+      throw new SaveCodegenToFileException(
+          String.format(
+              "Error occured when saving codegen to file %s/%s", relativeGoldenDir, fileName));
+    }
+  }
+
+  private static String getTestoutGoldenDir(Class clazz) {
+    return clazz.getPackage().getName().replace(".", "/") + "/goldens/";
+  }
+
+  private static class SaveCodegenToFileException extends RuntimeException {
+    public SaveCodegenToFileException(String errorMessage) {
+      super(errorMessage);
+    }
+  }
+}