Merge pull request DataJunction#17 from DataJunction/fix_druid

fix: druid

src/datajunction/cli/compile.py

@@ -108,9 +108,12 @@ def get_columns(table: Table) -> List[Column]:
             table.table,
             schema=table.schema_,
         )
-    except Exception as ex:  # pylint: disable=broad-except
+    except Exception:  # pylint: disable=broad-except
+        # Druid currently doesn't work with SQLAlchemy 1.4, and raises an exception. Once
+        # we've merged https://github.com/druid-io/pydruid/pull/275 we can modify this to
+        # re-raise the exception.
         _logger.exception("Unable to get table metadata")
-        raise ex
+        return []
 
     return [
         Column(

tests/cli/compile_test.py

@@ -135,9 +135,7 @@ def test_get_columns_error(mocker: MockerFixture) -> None:
     )
 
     table = mocker.MagicMock()
-    with pytest.raises(Exception) as excinfo:
-        get_columns(table)
-    assert str(excinfo.value) == "An unexpected error occurred"
+    assert get_columns(table) == []
 
 
 @pytest.mark.asyncio