Revert "set client during initialization"

This reverts commit 8cc50c2a19.

bundle/bundle.go

@@ -143,14 +143,20 @@ func (b *Bundle) InitializeWorkspaceClient() (*databricks.WorkspaceClient, error
 }
 
 func (b *Bundle) WorkspaceClient() *databricks.WorkspaceClient {
-	if b.client == nil {
-		panic("workspace client not initialized yet. This is a bug in the Databricks CLI.")
-	}
+	b.clientOnce.Do(func() {
+		var err error
+		b.client, err = b.InitializeWorkspaceClient()
+		if err != nil {
+			panic(err)
+		}
+	})
 	return b.client
 }
 
 // SetWorkpaceClient sets the workspace client for this bundle.
+// This is used to inject a mock client for testing.
 func (b *Bundle) SetWorkpaceClient(w *databricks.WorkspaceClient) {
+	b.clientOnce.Do(func() {})
 	b.client = w
 }
 

bundle/deploy/terraform/init_test.go

@@ -48,10 +48,7 @@ func TestInitEnvironmentVariables(t *testing.T) {
 	// TODO(pietern): create test fixture that initializes a mocked client.
 	t.Setenv("DATABRICKS_HOST", "https://x")
 	t.Setenv("DATABRICKS_TOKEN", "foobar")
-
-	client, err := b.InitializeWorkspaceClient()
-	require.NoError(t, err)
-	b.SetWorkpaceClient(client)
+	b.WorkspaceClient()
 
 	diags := bundle.Apply(context.Background(), b, Initialize())
 	require.NoError(t, diags.Error())

bundle/deploy/terraform/load_test.go

@@ -30,10 +30,7 @@ func TestLoadWithNoState(t *testing.T) {
 
 	t.Setenv("DATABRICKS_HOST", "https://x")
 	t.Setenv("DATABRICKS_TOKEN", "foobar")
-
-	client, err := b.InitializeWorkspaceClient()
-	require.NoError(t, err)
-	b.SetWorkpaceClient(client)
+	b.WorkspaceClient()
 
 	diags := bundle.Apply(context.Background(), b, bundle.Seq(
 		Initialize(),

cmd/root/bundle.go

@@ -93,7 +93,6 @@ func configureBundle(cmd *cobra.Command, b *bundle.Bundle) (*bundle.Bundle, diag
 	}
 	ctx = context.WithValue(ctx, &configUsed, client.Config)
 	cmd.SetContext(ctx)
-	b.SetWorkpaceClient(client)
 
 	return b, diags
 }

libs/template/renderer_test.go

@@ -91,10 +91,7 @@ func assertBuiltinTemplateValid(t *testing.T, template string, settings map[stri
 	})
 
 	b.Tagging = tags.ForCloud(w.Config)
-
-	client, err := b.InitializeWorkspaceClient()
-	require.NoError(t, err)
-	b.SetWorkpaceClient(client)
+	b.WorkspaceClient()
 
 	diags = bundle.Apply(ctx, b, bundle.Seq(
 		phases.Initialize(),