mirror of https://github.com/databricks/cli.git
52 lines
1.7 KiB
Go
52 lines
1.7 KiB
Go
package auth_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/internal/testcli"
|
|
"github.com/databricks/databricks-sdk-go"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAuthDescribeSuccess(t *testing.T) {
|
|
t.Skipf("Skipping because of https://github.com/databricks/cli/issues/2010")
|
|
|
|
ctx := context.Background()
|
|
stdout, _ := testcli.RequireSuccessfulRun(t, ctx, "auth", "describe")
|
|
outStr := stdout.String()
|
|
|
|
w, err := databricks.NewWorkspaceClient(&databricks.Config{})
|
|
require.NoError(t, err)
|
|
|
|
require.NotEmpty(t, outStr)
|
|
require.Contains(t, outStr, "Host: "+w.Config.Host)
|
|
|
|
me, err := w.CurrentUser.Me(context.Background())
|
|
require.NoError(t, err)
|
|
require.Contains(t, outStr, "User: "+me.UserName)
|
|
require.Contains(t, outStr, "Authenticated with: "+w.Config.AuthType)
|
|
require.Contains(t, outStr, "Current configuration:")
|
|
require.Contains(t, outStr, "✓ host: "+w.Config.Host)
|
|
require.Contains(t, outStr, "✓ profile: default")
|
|
}
|
|
|
|
func TestAuthDescribeFailure(t *testing.T) {
|
|
t.Skipf("Skipping because of https://github.com/databricks/cli/issues/2010")
|
|
|
|
ctx := context.Background()
|
|
stdout, _ := testcli.RequireSuccessfulRun(t, ctx, "auth", "describe", "--profile", "nonexistent")
|
|
outStr := stdout.String()
|
|
|
|
require.NotEmpty(t, outStr)
|
|
require.Contains(t, outStr, "Unable to authenticate: resolve")
|
|
require.Contains(t, outStr, "has no nonexistent profile configured")
|
|
require.Contains(t, outStr, "Current configuration:")
|
|
|
|
w, err := databricks.NewWorkspaceClient(&databricks.Config{})
|
|
require.NoError(t, err)
|
|
|
|
require.Contains(t, outStr, "✓ host: "+w.Config.Host)
|
|
require.Contains(t, outStr, "✓ profile: nonexistent (from --profile flag)")
|
|
}
|