No public description

PiperOrigin-RevId: 811415014

Author: colaboratory-team

google/colab/sql/bigquery.py

@@ -105,7 +105,7 @@ def validate(sql: str) -> ValidationResult:
     bytes_processed = 0
     try:
       bytes_processed = int(dry_run_series.get('totalBytesProcessed', 0))
-    except ValueError:
+    except (TypeError, ValueError):
       pass
     compiled_sql = dry_run_series.get('dispatchedSql', '')
     referenced_tables = dry_run_series.get('referencedTables', None) or []

google/colab/sql/bigquery_test.py

@@ -377,6 +377,24 @@ def test_validate_sql_with_unknown_exception(self):
 
     self.assertEqual(expected, result)
 
+  def test_validate_sql_with_none_bytes_processed(self):
+    dry_run_result = pandas.Series(
+        {'totalBytesProcessed': None, 'dispatchedSql': _QUERY}
+    )
+    self.mock_read_gbq_colab.return_value = dry_run_result
+    expected = bigquery.ValidationSuccess(
+        bytes_processed=0,
+        compiled_sql=_QUERY,
+        tables=[],
+        schema=[],
+    )
+
+    self.assertEqual(expected, bigquery.validate(_QUERY))
+
+    self.mock_read_gbq_colab.assert_called_once_with(
+        _QUERY, dry_run=True, pyformat_args={}
+    )
+
 
 class RunTest(unittest.TestCase):