Skip to content

Commit bb6ba4f

Browse files
committed
bugfix cluster id
1 parent af5674a commit bb6ba4f

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

tests/integration/fixtures/test_connect.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,12 +5,19 @@
55

66
@fixture
77
def serverless_env():
8-
os.environ['DATABRICKS_SERVERLESS_COMPUTE_ID'] = 'auto'
8+
os.environ['DATABRICKS_SERVERLESS_COMPUTE_ID'] = "auto"
99
yield
1010
os.environ.pop('DATABRICKS_SERVERLESS_COMPUTE_ID')
1111

1212

13-
def test_databricks_connect(ws, spark):
13+
@fixture
14+
def debug_env_bugfix(monkeypatch, debug_env):
15+
# This is a workaround to set shared cluster
16+
# TODO: Update secret vault for acceptance testing and remove the bugfix
17+
monkeypatch.setitem(debug_env, "DATABRICKS_CLUSTER_ID", "1114-152544-29g1w07e")
18+
19+
20+
def test_databricks_connect(debug_env_bugfix, ws, spark):
1421
rows = spark.sql("SELECT 1").collect()
1522
assert rows[0][0] == 1
1623

0 commit comments

Comments
 (0)