Skip to content

Commit db89a2e

Browse files
authored
Make test_jobs_with_no_inventory_database resilient to concurrent object deletes (#586)
Closes #541
1 parent fcddc68 commit db89a2e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/integration/test_installation.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from datetime import timedelta
33

44
import pytest
5-
from databricks.sdk.errors import OperationFailed
5+
from databricks.sdk.errors import NotFound, OperationFailed
66
from databricks.sdk.retries import retried
77
from databricks.sdk.service.catalog import SchemaInfo
88
from databricks.sdk.service.iam import PermissionLevel
@@ -77,6 +77,7 @@ def test_logs_are_available(ws, sql_backend, env_or_skip, make_random):
7777
assert len(workflow_run_logs) == 1
7878

7979

80+
@retried(on=[NotFound], timeout=timedelta(minutes=15))
8081
def test_jobs_with_no_inventory_database(
8182
ws,
8283
sql_backend,
@@ -247,6 +248,6 @@ def validate_tacl():
247248
logger.debug(f"cleaning up install folder: {install._install_folder}")
248249
ws.workspace.delete(install._install_folder, recursive=True)
249250

250-
for step, job_id in install._deployed_steps.items():
251+
for step, job_id in install._state.jobs.items():
251252
logger.debug(f"cleaning up {step} job_id={job_id}")
252253
ws.jobs.delete(job_id)

0 commit comments

Comments
 (0)