From 0f40c5ccfb5a06897f7ed3d52374b80a2e09ad92 Mon Sep 17 00:00:00 2001 From: Anton Nekipelov <226657+anton-107@users.noreply.github.com> Date: Thu, 4 Jun 2026 16:50:21 +0200 Subject: [PATCH] acceptance tests for experimental ssh command --- acceptance/ssh/connect-dedicated-cluster/known_hosts | 2 ++ .../ssh/connect-dedicated-cluster/out.test.toml | 4 ++++ acceptance/ssh/connect-dedicated-cluster/output.txt | 12 ++++++++++++ acceptance/ssh/connect-dedicated-cluster/script | 1 + acceptance/ssh/connect-dedicated-cluster/test.toml | 6 ++++++ acceptance/ssh/connect-serverless-cpu/out.test.toml | 4 ++++ acceptance/ssh/connect-serverless-cpu/output.txt | 12 ++++++++++++ acceptance/ssh/connect-serverless-cpu/script | 1 + acceptance/ssh/connect-serverless-cpu/test.toml | 8 ++++++++ 9 files changed, 50 insertions(+) create mode 100644 acceptance/ssh/connect-dedicated-cluster/known_hosts create mode 100644 acceptance/ssh/connect-dedicated-cluster/out.test.toml create mode 100644 acceptance/ssh/connect-dedicated-cluster/output.txt create mode 100644 acceptance/ssh/connect-dedicated-cluster/script create mode 100644 acceptance/ssh/connect-dedicated-cluster/test.toml create mode 100644 acceptance/ssh/connect-serverless-cpu/out.test.toml create mode 100644 acceptance/ssh/connect-serverless-cpu/output.txt create mode 100644 acceptance/ssh/connect-serverless-cpu/script create mode 100644 acceptance/ssh/connect-serverless-cpu/test.toml diff --git a/acceptance/ssh/connect-dedicated-cluster/known_hosts b/acceptance/ssh/connect-dedicated-cluster/known_hosts new file mode 100644 index 00000000000..b254044c77a --- /dev/null +++ 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 new file mode 100644 index 00000000000..d8567f78ef1 --- /dev/null +++ b/acceptance/ssh/connect-dedicated-cluster/out.test.toml @@ -0,0 +1,4 @@ +Local = false +Cloud = true +RequiresCluster = true +EnvMatrix.DATABRICKS_BUNDLE_ENGINE = ["direct"] diff --git a/acceptance/ssh/connect-dedicated-cluster/output.txt b/acceptance/ssh/connect-dedicated-cluster/output.txt new file mode 100644 index 00000000000..ad12ea741d9 --- /dev/null +++ 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 new file mode 100644 index 00000000000..76e688c7250 --- /dev/null +++ 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'" \ No newline at end of file diff --git a/acceptance/ssh/connect-dedicated-cluster/test.toml b/acceptance/ssh/connect-dedicated-cluster/test.toml new file mode 100644 index 00000000000..e0420d1ef0f --- /dev/null +++ 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 new file mode 100644 index 00000000000..0a085fdfda4 --- /dev/null +++ b/acceptance/ssh/connect-serverless-cpu/out.test.toml @@ -0,0 +1,4 @@ +Local = false +Cloud = true +RequiresUnityCatalog = true +EnvMatrix.DATABRICKS_BUNDLE_ENGINE = ["direct"] diff --git a/acceptance/ssh/connect-serverless-cpu/output.txt b/acceptance/ssh/connect-serverless-cpu/output.txt new file mode 100644 index 00000000000..aea0fcf1268 --- /dev/null +++ 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 new file mode 100644 index 00000000000..d6d21427984 --- /dev/null +++ b/acceptance/ssh/connect-serverless-cpu/script @@ -0,0 +1 @@ +$CLI ssh connect --releases-dir=$CLI_RELEASES_DIR \ No newline at end of file diff --git a/acceptance/ssh/connect-serverless-cpu/test.toml b/acceptance/ssh/connect-serverless-cpu/test.toml new file mode 100644 index 00000000000..00215c39c10 --- /dev/null +++ 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"]