feedback

Author: logachev

java-bigquery/google-cloud-bigquery-jdbc/src/main/java/com/google/cloud/bigquery/jdbc/BigQueryStatement.java

@@ -961,13 +961,16 @@ void processQueryResponse(String query, TableResult results) throws SQLException
   }
 
   private boolean isPermissionDeniedException(Throwable t) {
-    if (t == null) {
-      return false;
-    }
-    if (t instanceof StatusRuntimeException) {
-      return ((StatusRuntimeException) t).getStatus().getCode() == Status.Code.PERMISSION_DENIED;
+    while (t != null) {
+      if (t instanceof StatusRuntimeException) {
+        return ((StatusRuntimeException) t).getStatus().getCode() == Status.Code.PERMISSION_DENIED;
+      }
+      if (t instanceof ApiException) {
+        return ((ApiException) t).getStatusCode().getCode() == StatusCode.Code.PERMISSION_DENIED;
+      }
+      t = t.getCause();
     }
-    return isPermissionDeniedException(t.getCause());
+    return false;
   }
 
   // The read Ratio should be met

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

@@ -20,6 +20,7 @@
 import static com.google.common.truth.Truth.assertThat;
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.fail;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.doReturn;
@@ -68,7 +69,6 @@
 import org.apache.arrow.vector.FieldVector;
 import org.apache.arrow.vector.IntVector;
 import org.apache.arrow.vector.VectorSchemaRoot;
-import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Disabled;
 import org.junit.jupiter.api.Test;
@@ -551,7 +551,7 @@ public void testProcessQueryResponseNoFallbackOnNonPermissionFailure() throws SQ
     // Assert that the exception is propagated
     try {
       statementSpy.processQueryResponse("SELECT 1", tableResultMock);
-      Assertions.fail("Expected SQLException to be thrown");
+      fail("Expected SQLException to be thrown");
     } catch (SQLException e) {
       assertEquals(exceptionToThrow, e);
     }