From 85b891e23ae5bdfad68e4f67bdf594a732b094eb Mon Sep 17 00:00:00 2001 From: Anton Nekipelov <226657+anton-107@users.noreply.github.com> Date: Thu, 19 Feb 2026 14:01:51 +0100 Subject: [PATCH] Add an acceptance test for ssh connection to serverless GPU clusters --- acceptance/ssh/connect-serverless-gpu/out.stdout.txt | 1 + acceptance/ssh/connect-serverless-gpu/out.test.toml | 6 ++++++ acceptance/ssh/connect-serverless-gpu/output.txt | 0 acceptance/ssh/connect-serverless-gpu/script | 6 ++++++ acceptance/ssh/connect-serverless-gpu/test.toml | 6 ++++++ 5 files changed, 19 insertions(+) create mode 100644 acceptance/ssh/connect-serverless-gpu/out.stdout.txt create mode 100644 acceptance/ssh/connect-serverless-gpu/out.test.toml create mode 100644 acceptance/ssh/connect-serverless-gpu/output.txt create mode 100644 acceptance/ssh/connect-serverless-gpu/script create mode 100644 acceptance/ssh/connect-serverless-gpu/test.toml diff --git a/acceptance/ssh/connect-serverless-gpu/out.stdout.txt b/acceptance/ssh/connect-serverless-gpu/out.stdout.txt new file mode 100644 index 0000000000..41cae5e7d1 --- /dev/null +++ b/acceptance/ssh/connect-serverless-gpu/out.stdout.txt @@ -0,0 +1 @@ +Connection successful diff --git a/acceptance/ssh/connect-serverless-gpu/out.test.toml b/acceptance/ssh/connect-serverless-gpu/out.test.toml new file mode 100644 index 0000000000..e0420d1ef0 --- /dev/null +++ b/acceptance/ssh/connect-serverless-gpu/out.test.toml @@ -0,0 +1,6 @@ +Local = false +Cloud = true +RequiresCluster = true + +[EnvMatrix] + DATABRICKS_BUNDLE_ENGINE = ["direct"] diff --git a/acceptance/ssh/connect-serverless-gpu/output.txt b/acceptance/ssh/connect-serverless-gpu/output.txt new file mode 100644 index 0000000000..e69de29bb2 diff --git a/acceptance/ssh/connect-serverless-gpu/script b/acceptance/ssh/connect-serverless-gpu/script new file mode 100644 index 0000000000..8c8538f6e3 --- /dev/null +++ b/acceptance/ssh/connect-serverless-gpu/script @@ -0,0 +1,6 @@ +errcode $CLI ssh connect --name serverless-gpu-test --accelerator=GPU_1xA10 --releases-dir=$CLI_RELEASES_DIR -- "echo 'Connection successful'" >out.stdout.txt 2>LOG.stderr + +if ! grep -q "Connection successful" out.stdout.txt; then + run_id=$(cat LOG.stderr | grep -o "Job submitted successfully with run ID: [0-9]*" | grep -o "[0-9]*$") + trace $CLI jobs get-run "$run_id" > LOG.job +fi diff --git a/acceptance/ssh/connect-serverless-gpu/test.toml b/acceptance/ssh/connect-serverless-gpu/test.toml new file mode 100644 index 0000000000..e0420d1ef0 --- /dev/null +++ b/acceptance/ssh/connect-serverless-gpu/test.toml @@ -0,0 +1,6 @@ +Local = false +Cloud = true +RequiresCluster = true + +[EnvMatrix] + DATABRICKS_BUNDLE_ENGINE = ["direct"]