Skip to content

Commit f7058bc

Browse files
authored
Fixed failing unit tests (#1150)
## Changes 2 unit tests failed after pytest upgrade due to incorrect error matching
1 parent b4b3b52 commit f7058bc

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

tests/unit/hive_metastore/test_mapping.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,10 @@
33

44
import pytest
55
from databricks.labs.blueprint.installation import Installation, MockInstallation
6-
from databricks.labs.blueprint.parallel import ManyError
76
from databricks.labs.lsql.backends import MockBackend, SqlBackend
87
from databricks.sdk import WorkspaceClient
98
from databricks.sdk.errors import NotFound
9+
from databricks.sdk.errors.platform import ResourceConflict
1010
from databricks.sdk.service.catalog import TableInfo
1111

1212
from databricks.labs.ucx.account import WorkspaceInfo
@@ -410,7 +410,7 @@ def test_mismatch_from_table_raises_exception():
410410
name="table1",
411411
),
412412
]
413-
with pytest.raises(ManyError, match="ResourceConflict"):
413+
with pytest.raises(ResourceConflict):
414414
table_mapping.get_tables_to_migrate(tables_crawler)
415415

416416
assert ["DESCRIBE SCHEMA EXTENDED schema1"] == backend.queries

tests/unit/test_install.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,7 @@
2626
PermissionDenied,
2727
Unknown,
2828
)
29+
from databricks.sdk.errors.platform import BadRequest
2930
from databricks.sdk.service import iam, jobs, sql
3031
from databricks.sdk.service.compute import (
3132
ClusterDetails,
@@ -222,7 +223,7 @@ def test_create_database(ws, caplog, mock_installation, any_prompt):
222223
PRODUCT_INFO,
223224
)
224225

225-
with pytest.raises(ManyError) as failure:
226+
with pytest.raises(BadRequest) as failure:
226227
workspace_installation.run()
227228

228229
assert "Kindly uninstall and reinstall UCX" in str(failure.value)

0 commit comments

Comments
 (0)