fix test

Author: Neenu1995

java-bigquery/google-cloud-bigquery-jdbc/pom.xml

@@ -307,6 +307,11 @@
       <artifactId>mockito-core</artifactId>
       <scope>test</scope>
     </dependency>
+    <dependency>
+      <groupId>org.mockito</groupId>
+      <artifactId>mockito-inline</artifactId>
+      <scope>test</scope>
+    </dependency>
     <dependency>
       <groupId>org.mockito</groupId>
       <artifactId>mockito-junit-jupiter</artifactId>

java-bigquery/google-cloud-bigquery-jdbc/src/test/java/com/google/cloud/bigquery/jdbc/BigQueryCallableStatementTest.java

@@ -914,7 +914,7 @@ public void testSetDateCalParamByName() throws SQLException {
     Calendar expectedCal = mock(Calendar.class);
 
     doReturn(1L).when(expectedDate).getTime();
-    doReturn(1L).when(expectedCal).getTime();
+    doReturn(new java.util.Date(1L)).when(expectedCal).getTime();
     doReturn(1L).when(expectedCal).getTimeInMillis();
     statement.setDate(PARAM_KEY, expectedDate, expectedCal);
     Date actual = statement.getDate(PARAM_KEY);
@@ -1033,7 +1033,7 @@ public void testSetTimeCalParamByName() throws SQLException {
     Calendar expectedCal = mock(Calendar.class);
 
     doReturn(1L).when(expectedTime).getTime();
-    doReturn(1L).when(expectedCal).getTime();
+    doReturn(new java.util.Date(1L)).when(expectedCal).getTime();
     doReturn(1L).when(expectedCal).getTimeInMillis();
     statement.setTime(PARAM_KEY, expectedTime, expectedCal);
     Time actual = statement.getTime(PARAM_KEY);
@@ -1062,7 +1062,7 @@ public void testSetTimestampCalParamByName() throws SQLException {
     Calendar expectedCal = mock(Calendar.class);
 
     doReturn(1L).when(expectedTimestamp).getTime();
-    doReturn(1L).when(expectedCal).getTime();
+    doReturn(new java.util.Date(1L)).when(expectedCal).getTime();
     doReturn(1L).when(expectedCal).getTimeInMillis();
     statement.setTimestamp(PARAM_KEY, expectedTimestamp, expectedCal);
     Timestamp actual = statement.getTimestamp(PARAM_KEY);

java-bigquery/google-cloud-bigquery-jdbc/src/test/java/com/google/cloud/bigquery/jdbc/BigQueryJdbcOAuthUtilityTest.java

@@ -395,15 +395,11 @@ public void testParseUserImpersonationForADC() {
 
   @Test
   public void testGetServiceAccountImpersonatedCredentialsForADC() throws Exception {
-    java.nio.file.Path tempAdcFile = createTempAdcFile();
-    String originalEnv = System.getenv("GOOGLE_APPLICATION_CREDENTIALS");
+    GoogleCredentials dummySourceCredentials = GoogleCredentials.newBuilder().build();
 
-    try {
-      if (originalEnv == null) {
-        setEnvironmentVariable(
-            "GOOGLE_APPLICATION_CREDENTIALS", tempAdcFile.toAbsolutePath().toString());
-        clearDefaultCredentialsCache();
-      }
+    try (org.mockito.MockedStatic<GoogleCredentials> mockedCreds =
+        org.mockito.Mockito.mockStatic(GoogleCredentials.class)) {
+      mockedCreds.when(GoogleCredentials::getApplicationDefault).thenReturn(dummySourceCredentials);
 
       Map<String, String> authProperties =
           BigQueryJdbcOAuthUtility.parseOAuthProperties(
@@ -418,11 +414,6 @@ public void testGetServiceAccountImpersonatedCredentialsForADC() throws Exceptio
               authProperties, java.util.Collections.EMPTY_MAP, false, null);
 
       assertThat(credentials).isInstanceOf(ImpersonatedCredentials.class);
-    } finally {
-      if (originalEnv == null) {
-        removeEnvironmentVariable("GOOGLE_APPLICATION_CREDENTIALS");
-        clearDefaultCredentialsCache();
-      }
     }
   }
 
@@ -480,56 +471,4 @@ public void testPrivateKeyFromP12Bytes_wrong_password() {
       assertTrue(false);
     }
   }
-
-  private static java.nio.file.Path createTempAdcFile() throws Exception {
-    String dummyJson =
-        "{\n"
-            + "  \"type\": \"service_account\",\n"
-            + "  \"project_id\": \"dummy-project\",\n"
-            + "  \"private_key_id\": \"dummy-key-id\",\n"
-            + "  \"private_key\": \""
-            + fake_pkcs8_key.replace("\n", "\\n")
-            + "\",\n"
-            + "  \"client_email\": \"dummy@email.com\",\n"
-            + "  \"client_id\": \"dummy-client-id\",\n"
-            + "  \"auth_uri\": \"https://accounts.google.com/o/oauth2/auth\",\n"
-            + "  \"token_uri\": \"https://oauth2.googleapis.com/token\"\n"
-            + "}";
-
-    java.nio.file.Path tempFile = java.nio.file.Files.createTempFile("adc", ".json");
-    java.nio.file.Files.write(tempFile, dummyJson.getBytes());
-    tempFile.toFile().deleteOnExit();
-    return tempFile;
-  }
-
-  private static void setEnvironmentVariable(String key, String value) throws Exception {
-    Class<?> processEnvironmentClass = Class.forName("java.lang.ProcessEnvironment");
-    java.lang.reflect.Field theEnvironmentField =
-        processEnvironmentClass.getDeclaredField("theEnvironment");
-    theEnvironmentField.setAccessible(true);
-    java.util.Map<String, String> env =
-        (java.util.Map<String, String>) theEnvironmentField.get(null);
-    env.put(key, value);
-  }
-
-  private static void removeEnvironmentVariable(String key) throws Exception {
-    Class<?> processEnvironmentClass = Class.forName("java.lang.ProcessEnvironment");
-    java.lang.reflect.Field theEnvironmentField =
-        processEnvironmentClass.getDeclaredField("theEnvironment");
-    theEnvironmentField.setAccessible(true);
-    java.util.Map<String, String> env =
-        (java.util.Map<String, String>) theEnvironmentField.get(null);
-    env.remove(key);
-  }
-
-  private static void clearDefaultCredentialsCache() throws Exception {
-    Class<?> providerClass = Class.forName("com.google.auth.oauth2.DefaultCredentialsProvider");
-    java.lang.reflect.Field defaultField = providerClass.getDeclaredField("DEFAULT");
-    defaultField.setAccessible(true);
-    Object provider = defaultField.get(null);
-
-    java.lang.reflect.Field cachedCredsField = providerClass.getDeclaredField("cachedCredentials");
-    cachedCredsField.setAccessible(true);
-    cachedCredsField.set(provider, null);
-  }
 }