test(bigquery): import googlenotfound error to handle overlapping tab…

…le name test
ibis-project · Aug 4, 2024 · 907227e · 907227e
1 parent 2bb92d9
commit 907227e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/ibis/backends/tests/errors.py b/ibis/backends/tests/errors.py
@@ -55,8 +55,9 @@
 
 try:
     from google.api_core.exceptions import BadRequest as GoogleBadRequest
+    from google.api_core.exceptions import NotFound as GoogleNotFound
 except ImportError:
-    GoogleBadRequest = None
+    GoogleBadRequest = GoogleNotFound = None
 
 try:
     from polars.exceptions import ColumnNotFoundError as PolarsColumnNotFoundError

diff --git a/ibis/backends/tests/test_client.py b/ibis/backends/tests/test_client.py
@@ -29,6 +29,7 @@
 from ibis.backends.tests.errors import (
     ClickHouseDatabaseError,
     ExaQueryError,
+    GoogleNotFound,
     ImpalaHiveServer2Error,
     MySQLProgrammingError,
     OracleDatabaseError,
@@ -1659,6 +1660,7 @@ def test_no_accidental_cross_database_table_load(con_create_database):
                 MySQLProgrammingError,
                 ExaQueryError,
                 SnowflakeProgrammingError,
+                GoogleNotFound,
             ),
         ),
         # datafusion really needs to get their exception story in order