acceptance tests for experimental ssh command

anton-107 · anton-107 · commit 0f40c5ccfb5a · 2026-06-04T16:50:21.000+02:00
diff --git a/acceptance/ssh/connect-dedicated-cluster/known_hosts b/acceptance/ssh/connect-dedicated-cluster/known_hosts
@@ -0,0 +1,2 @@
+# not actually checked by tests
+cluster ssh-rsa key
diff --git a/acceptance/ssh/connect-dedicated-cluster/out.test.toml b/acceptance/ssh/connect-dedicated-cluster/out.test.toml
diff --git a/acceptance/ssh/connect-dedicated-cluster/output.txt b/acceptance/ssh/connect-dedicated-cluster/output.txt
@@ -0,0 +1,12 @@
+Connecting to [TEST_DEFAULT_CLUSTER_ID]...
+Checking cluster state...
+Uploading binaries...
+Starting SSH server...
+Job submitted successfully with run ID: [NUMID]
+Health check successful, starting ssh WebSocket connection...
+Connected!
+Checking cluster state...
+Warning: Permanently added '[TEST_DEFAULT_CLUSTER_ID]' (RSA) to the list of known hosts.
+Connection successful
+Received termination signal, cleaning up...
+Error: context canceled
diff --git a/acceptance/ssh/connect-dedicated-cluster/script b/acceptance/ssh/connect-dedicated-cluster/script
@@ -0,0 +1 @@
+$CLI ssh connect --cluster=$TEST_DEFAULT_CLUSTER_ID --user-known-hosts-file=known_hosts --releases-dir=$CLI_RELEASES_DIR -- "echo 'Connection successful'"
diff --git a/acceptance/ssh/connect-dedicated-cluster/test.toml b/acceptance/ssh/connect-dedicated-cluster/test.toml
@@ -0,0 +1,6 @@
+Local = false
+Cloud = true
+RequiresCluster = true
+
+[EnvMatrix]
+  DATABRICKS_BUNDLE_ENGINE = ["direct"]
diff --git a/acceptance/ssh/connect-serverless-cpu/out.test.toml b/acceptance/ssh/connect-serverless-cpu/out.test.toml
diff --git a/acceptance/ssh/connect-serverless-cpu/output.txt b/acceptance/ssh/connect-serverless-cpu/output.txt
@@ -0,0 +1,12 @@
+Connecting to databricks-cpu-6e7644d0...
+WARNING: serverless compute without an accelerator is in private preview. If you are not enrolled, this command will likely time out with an error. Contact your Databricks account team to enroll.
+Uploading binaries...
+Starting SSH server...
+Job submitted successfully with run ID: [NUMID]
+Error: failed to ensure that ssh server is running: failed to start the ssh server: server metadata error
+server is not ok, status code 503
+
+This may be because serverless compute without an accelerator is in private preview.
+Contact your Databricks account team to enroll.
+
+Exit code: 1
diff --git a/acceptance/ssh/connect-serverless-cpu/script b/acceptance/ssh/connect-serverless-cpu/script
@@ -0,0 +1 @@
+$CLI ssh connect --releases-dir=$CLI_RELEASES_DIR
diff --git a/acceptance/ssh/connect-serverless-cpu/test.toml b/acceptance/ssh/connect-serverless-cpu/test.toml
@@ -0,0 +1,8 @@
+Local = false
+Cloud = true
+
+# Serverless worklaods are only available in newer environments
+RequiresUnityCatalog = true
+
+[EnvMatrix]
+  DATABRICKS_BUNDLE_ENGINE = ["direct"]

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1,2 @@`
	`1`	`+# not actually checked by tests`
	`2`	`+cluster ssh-rsa key`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+$CLI ssh connect --cluster=$TEST_DEFAULT_CLUSTER_ID --user-known-hosts-file=known_hosts --releases-dir=$CLI_RELEASES_DIR -- "echo 'Connection successful'"`
Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	`+$CLI ssh connect --releases-dir=$CLI_RELEASES_DIR`