From e3fb2b273f939314d9cdbce539f373d6fc77d0ad Mon Sep 17 00:00:00 2001
From: Mridula <66699525+mpeddada1@users.noreply.github.com>
Date: Tue, 22 Feb 2022 10:07:26 -0500
Subject: [PATCH] fix(java): make system property accessible for native image
 compilation (#1694)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* fix(java): make system properties accessible for native image compilation

* 🦉 Updates from OwlBot

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
---
 google-cloud-spanner/pom.xml                  | 38 +++++++++++++++++--
 .../cloud/spanner/IntegrationTestEnv.java     |  7 ++--
 2 files changed, 38 insertions(+), 7 deletions(-)

diff --git a/google-cloud-spanner/pom.xml b/google-cloud-spanner/pom.xml
index 46b11e401c8..2aeaa0ddfc7 100644
--- a/google-cloud-spanner/pom.xml
+++ b/google-cloud-spanner/pom.xml
@@ -16,6 +16,10 @@
   <properties>
     <site.installationModule>google-cloud-spanner</site.installationModule>
     <opencensus.version>0.31.0</opencensus.version>
+    <spanner.testenv.config.class>com.google.cloud.spanner.GceTestEnvConfig</spanner.testenv.config.class>
+    <spanner.testenv.instance>projects/gcloud-devel/instances/spanner-testing-east1</spanner.testenv.instance>
+    <spanner.gce.config.project_id>gcloud-devel</spanner.gce.config.project_id>
+    <spanner.testenv.kms_key.name>projects/gcloud-devel/locations/us-east1/keyRings/cmek-test-key-ring/cryptoKeys/cmek-test-key</spanner.testenv.kms_key.name>
   </properties>
 
 
@@ -70,10 +74,10 @@
         <artifactId>maven-failsafe-plugin</artifactId>
         <configuration>
           <systemPropertyVariables>
-            <spanner.testenv.config.class>com.google.cloud.spanner.GceTestEnvConfig</spanner.testenv.config.class>
-            <spanner.testenv.instance>projects/gcloud-devel/instances/spanner-testing-east1</spanner.testenv.instance>
-            <spanner.gce.config.project_id>gcloud-devel</spanner.gce.config.project_id>
-            <spanner.testenv.kms_key.name>projects/gcloud-devel/locations/us-east1/keyRings/cmek-test-key-ring/cryptoKeys/cmek-test-key</spanner.testenv.kms_key.name>
+            <spanner.testenv.config.class>${spanner.testenv.config.class}</spanner.testenv.config.class>
+            <spanner.testenv.instance>${spanner.testenv.instance}</spanner.testenv.instance>
+            <spanner.gce.config.project_id>${spanner.gce.config.project_id}</spanner.gce.config.project_id>
+            <spanner.testenv.kms_key.name>${spanner.testenv.kms_key.name}</spanner.testenv.kms_key.name>
           </systemPropertyVariables>
           <forkedProcessTimeoutInSeconds>3000</forkedProcessTimeoutInSeconds>
         </configuration>
@@ -100,6 +104,32 @@
           </execution>
         </executions>
       </plugin>
+
+  <!--Configurations for native image tests-->
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-surefire-plugin</artifactId>
+        <configuration>
+          <systemPropertyVariables>
+            <spanner.testenv.config.class>${spanner.testenv.config.class}</spanner.testenv.config.class>
+            <spanner.testenv.instance>${spanner.testenv.instance}</spanner.testenv.instance>
+            <spanner.gce.config.project_id>${spanner.gce.config.project_id}</spanner.gce.config.project_id>
+            <spanner.testenv.kms_key.name>${spanner.testenv.kms_key.name}</spanner.testenv.kms_key.name>
+          </systemPropertyVariables>
+          <forkedProcessTimeoutInSeconds>3000</forkedProcessTimeoutInSeconds>
+        </configuration>
+      </plugin>
+      <plugin>
+        <groupId>org.graalvm.buildtools</groupId>
+        <artifactId>native-maven-plugin</artifactId>
+        <configuration>
+          <buildArgs combine.children="append">
+            <buildArg>-Dspanner.testenv.config.class=${spanner.testenv.config.class}</buildArg>
+          </buildArgs>
+        </configuration>
+      </plugin>
+      <!--Configurations for native image tests-->
+
       <plugin>
         <?m2e ignore?>
         <groupId>org.codehaus.mojo</groupId>
diff --git a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/IntegrationTestEnv.java b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/IntegrationTestEnv.java
index 103b6f626ba..3f4db8b9135 100644
--- a/google-cloud-spanner/src/test/java/com/google/cloud/spanner/IntegrationTestEnv.java
+++ b/google-cloud-spanner/src/test/java/com/google/cloud/spanner/IntegrationTestEnv.java
@@ -42,6 +42,8 @@ public class IntegrationTestEnv extends ExternalResource {
   /** Names a property that provides the class name of the {@link TestEnvConfig} to use. */
   public static final String TEST_ENV_CONFIG_CLASS_NAME = "spanner.testenv.config.class";
 
+  public static final String CONFIG_CLASS = System.getProperty(TEST_ENV_CONFIG_CLASS_NAME, null);
+
   /**
    * Names a property that, if set, identifies an existing Cloud Spanner instance to use for tests.
    */
@@ -62,12 +64,11 @@ public RemoteSpannerHelper getTestHelper() {
   @SuppressWarnings("unchecked")
   protected void initializeConfig()
       throws ClassNotFoundException, InstantiationException, IllegalAccessException {
-    String configClassName = System.getProperty(TEST_ENV_CONFIG_CLASS_NAME, null);
-    if (configClassName == null) {
+    if (CONFIG_CLASS == null) {
       throw new NullPointerException("Property " + TEST_ENV_CONFIG_CLASS_NAME + " needs to be set");
     }
     Class<? extends TestEnvConfig> configClass;
-    configClass = (Class<? extends TestEnvConfig>) Class.forName(configClassName);
+    configClass = (Class<? extends TestEnvConfig>) Class.forName(CONFIG_CLASS);
     config = configClass.newInstance();
   }