Use getCode instead of getReason

Author: Bingqin Zhou

kcbq-connector/src/main/java/com/wepay/kafka/connect/bigquery/write/row/AdaptiveBigQueryWriter.java

@@ -81,7 +81,7 @@ private boolean isTableMissingSchema(BigQueryException exception) {
   private boolean isTableNotExistedException(BigQueryException exception) {
     // If a table does not exist, it will raise a BigQueryException that the input is notFound
     // Referring to Google Cloud Error Codes Doc: https://cloud.google.com/bigquery/docs/error-messages?hl=en
-    return exception.getReason() != null && exception.getReason().equalsIgnoreCase("notFound");
+    return exception.getCode() == 404;
   }
 
   /**

kcbq-connector/src/test/java/com/wepay/kafka/connect/bigquery/write/row/BigQueryWriterTest.java

@@ -117,7 +117,7 @@ public void testAutoCreateTables() {
     when(insertAllResponse.getInsertErrors()).thenReturn(emptyMap);
 
     BigQueryException missTableException = mock(BigQueryException.class);
-    when(missTableException.getReason()).thenReturn("notFound");
+    when(missTableException.getCode()).thenReturn(404);
 
     when(bigQuery.insertAll(anyObject())).thenThrow(missTableException).thenReturn(insertAllResponse);
 
@@ -152,7 +152,7 @@ public void testNonAutoCreateTables() {
     when(insertAllResponse.getInsertErrors()).thenReturn(emptyMap);
 
     BigQueryException missTableException = mock(BigQueryException.class);
-    when(missTableException.getReason()).thenReturn("notFound");
+    when(missTableException.getCode()).thenReturn(404);
 
     when(bigQuery.insertAll(anyObject())).thenThrow(missTableException).thenReturn(insertAllResponse);