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 timeout cleanup

This commit is contained in:
Shreyas Goenka 2024-11-29 02:04:42 +01:00
parent a9c53acdff
commit 6cde085b22
No known key found for this signature in database
GPG Key ID: 92A07DF49CCB0622
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/telemetry_test.go
View File

@ -36,7 +36,7 @@ func TestAccTelemetryLogger(t *testing.T) {
// Extend the maximum wait time for the telemetry flush just for this test. // Extend the maximum wait time for the telemetry flush just for this test.
telemetry.MaxAdditionalWaitTime = 1 * time.Hour telemetry.MaxAdditionalWaitTime = 1 * time.Hour
t.Cleanup(func() { t.Cleanup(func() {
telemetry.MaxAdditionalWaitTime = 2 telemetry.MaxAdditionalWaitTime = 2 * time.Second
}) })
// Log some events. // Log some events.

3
libs/telemetry/logger_test.go
View File

@ -5,6 +5,7 @@ import (
"fmt" "fmt"
"net/http" "net/http"
"testing" "testing"
"time"
"github.com/databricks/cli/cmd/root" "github.com/databricks/cli/cmd/root"
"github.com/stretchr/testify/assert" "github.com/stretchr/testify/assert"
@ -78,7 +79,7 @@ func TestTelemetryLoggerFlushExitsOnTimeout(t *testing.T) {
// Set the maximum additional wait time to 0 to ensure that the Flush method times out immediately. // Set the maximum additional wait time to 0 to ensure that the Flush method times out immediately.
MaxAdditionalWaitTime = 0 MaxAdditionalWaitTime = 0
t.Cleanup(func() { t.Cleanup(func() {
MaxAdditionalWaitTime = 2 MaxAdditionalWaitTime = 2 * time.Second
}) })
mockClient := &mockDatabricksClient{ mockClient := &mockDatabricksClient{