fmt

Author: mwojtyczka

acceptance/notify/slack.go

@@ -5,10 +5,10 @@ import (
 	"strings"
 	"time"
 
+	"github.com/databricks/databricks-sdk-go/common/environment"
 	"github.com/databricks/databricks-sdk-go/openapi/code"
 	"github.com/databrickslabs/sandbox/acceptance/ecosystem"
 	"github.com/databrickslabs/sandbox/go-libs/slack"
-	"github.com/databricks/databricks-sdk-go/common/environment"
 )
 
 type Notification struct {

acceptance/notify/slack_test.go

@@ -3,11 +3,11 @@ package notify_test
 import (
 	"testing"
 
+	"github.com/databricks/databricks-sdk-go/common/environment"
 	"github.com/databrickslabs/sandbox/acceptance/ecosystem"
 	"github.com/databrickslabs/sandbox/acceptance/notify"
 	"github.com/databrickslabs/sandbox/go-libs/fixtures"
 	"github.com/databrickslabs/sandbox/go-libs/slack"
-	"github.com/databricks/databricks-sdk-go/common/environment"
 )
 
 func TestMessageDryRun(t *testing.T) {

acceptance/testenv/githubOidc.go

@@ -10,12 +10,12 @@ import (
 
 	"github.com/Azure/azure-sdk-for-go/sdk/azcore"
 	"github.com/Azure/azure-sdk-for-go/sdk/azcore/policy"
+	"github.com/databricks/databricks-sdk-go/common/environment"
 	"github.com/databricks/databricks-sdk-go/config"
+	"github.com/databricks/databricks-sdk-go/config/credentials"
 	"github.com/sethvargo/go-githubactions"
 	"golang.org/x/oauth2"
 	"golang.org/x/oauth2/clientcredentials"
-	"github.com/databricks/databricks-sdk-go/common/environment"
-	"github.com/databricks/databricks-sdk-go/config/credentials"
 )
 
 func NewWithGitHubOIDC(a *githubactions.Action, vaultURI string) *vaultEnv {
@@ -64,11 +64,11 @@ func (c *ghOidcCreds) Name() string {
 }
 
 type credentialsProviderFunc struct {
-    setHeadersFunc func(r *http.Request) error
+	setHeadersFunc func(r *http.Request) error
 }
 
 func (cpf *credentialsProviderFunc) SetHeaders(r *http.Request) error {
-    return cpf.setHeadersFunc(r)
+	return cpf.setHeadersFunc(r)
 }
 
 // Configure implements credentials provider for Databricks SDK
@@ -81,8 +81,8 @@ func (c *ghOidcCreds) Configure(ctx context.Context, cfg *config.Config) (creden
 		setHeadersFunc: func(r *http.Request) error {
 			token, err := ts.Token()
 			if err != nil {
-			return fmt.Errorf("token: %w", err)
-		}
+				return fmt.Errorf("token: %w", err)
+			}
 			token.SetAuthHeader(r)
 			return nil
 		},
@@ -107,16 +107,16 @@ func (c *ghOidcCreds) GetToken(ctx context.Context, options policy.TokenRequestO
 }
 
 func (c *ghOidcCreds) SetHeaders(r *http.Request) error {
-    env := environment.GetEnvironmentForHostname(r.Host)
-    azureManagementURL := env.AzureResourceManagerEndpoint()
-    token, err := c.GetToken(r.Context(), policy.TokenRequestOptions{
-        Scopes: []string{azureManagementURL + "/.default"},
-    })
-    if err != nil {
-        return fmt.Errorf("token: %w", err)
-    }
-    r.Header.Set("Authorization", "Bearer "+token.Token)
-    return nil
+	env := environment.GetEnvironmentForHostname(r.Host)
+	azureManagementURL := env.AzureResourceManagerEndpoint()
+	token, err := c.GetToken(r.Context(), policy.TokenRequestOptions{
+		Scopes: []string{azureManagementURL + "/.default"},
+	})
+	if err != nil {
+		return fmt.Errorf("token: %w", err)
+	}
+	r.Header.Set("Authorization", "Bearer "+token.Token)
+	return nil
 }
 
 type msiToken struct {

acceptance/testenv/loaded.go

@@ -11,11 +11,11 @@ import (
 	"time"
 
 	"github.com/databricks/databricks-sdk-go/apierr"
+	"github.com/databricks/databricks-sdk-go/common/environment"
 	"github.com/databricks/databricks-sdk-go/config"
 	"github.com/databricks/databricks-sdk-go/logger"
 	"github.com/databrickslabs/sandbox/acceptance/redaction"
 	"github.com/databrickslabs/sandbox/go-libs/env"
-	"github.com/databricks/databricks-sdk-go/common/environment"
 )
 
 type loadedEnv struct {