steve/databricks-cli
steve
/
databricks-cli
mirror of https://github.com/databricks/cli.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix CLI nightlies on our UC workspaces (#1225) 

This PR fixes some test helper functions so that they work properly on
our test environments for AWS and Azure UC workspaces.
This commit is contained in:
shreyas-goenka 2024-02-21 18:36:03 +05:30 committed by GitHub
parent 26833418c3
commit 4ac1c1655b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
internal/helpers.go
View File

@ -523,7 +523,8 @@ func TemporaryRepo(t *testing.T, w *databricks.WorkspaceClient) string {
func GetNodeTypeId(env string) string { func GetNodeTypeId(env string) string {
if env == "gcp" { if env == "gcp" {
return "n1-standard-4" return "n1-standard-4"
} else if env == "aws" { } else if env == "aws" || env == "ucws" {
// aws-prod-ucws has CLOUD_ENV set to "ucws"
return "i3.xlarge" return "i3.xlarge"
} }
return "Standard_DS4_v2" return "Standard_DS4_v2"

3
internal/testutil/cloud.go
View File

@ -41,6 +41,9 @@ func GetCloud(t *testing.T) Cloud {
return Azure return Azure
case "gcp": case "gcp":
return GCP return GCP
// CLOUD_ENV is set to "ucws" in the "aws-prod-ucws" test environment
case "ucws":
return AWS
default: default:
t.Fatalf("Unknown cloud environment: %s", env) t.Fatalf("Unknown cloud environment: %s", env)
} }