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

merge

This commit is contained in:
Shreyas Goenka 2024-12-20 11:38:17 +05:30
parent a4f5d89298
commit 6580869117
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
internal/telemetry_test.go → integration/libs/telemetry/telemetry_test.go
View File

@ -6,7 +6,7 @@ import (
"testing" "testing"
"time" "time"
"github.com/databricks/cli/internal/acc" "github.com/databricks/cli/integration/internal/acc"
"github.com/databricks/cli/libs/telemetry" "github.com/databricks/cli/libs/telemetry"
"github.com/databricks/databricks-sdk-go/client" "github.com/databricks/databricks-sdk-go/client"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
@ -22,14 +22,14 @@ type apiClientWrapper struct {
func (wrapper *apiClientWrapper) Do(ctx context.Context, method, path string, func (wrapper *apiClientWrapper) Do(ctx context.Context, method, path string,
headers map[string]string, request, response any, headers map[string]string, request, response any,
visitors ...func(*http.Request) error) error { visitors ...func(*http.Request) error,
) error {
err := wrapper.apiClient.Do(ctx, method, path, headers, request, response, visitors...) err := wrapper.apiClient.Do(ctx, method, path, headers, request, response, visitors...)
wrapper.response = response.(*telemetry.ResponseBody) wrapper.response = response.(*telemetry.ResponseBody)
return err return err
} }
func TestAccTelemetryLogger(t *testing.T) { func TestTelemetryLogger(t *testing.T) {
ctx, w := acc.WorkspaceTest(t) ctx, w := acc.WorkspaceTest(t)
ctx = telemetry.NewContext(ctx) ctx = telemetry.NewContext(ctx)