mirror of https://github.com/databricks/cli.git
fix tests
This commit is contained in:
parent
aec6b06cbe
commit
a5bbd1f4f4
|
@ -48,7 +48,7 @@ func TestTelemetryUploadRetriesOnPartialSuccess(t *testing.T) {
|
||||||
err := Upload(ctx, &config.Config{
|
err := Upload(ctx, &config.Config{
|
||||||
Host: server.URL,
|
Host: server.URL,
|
||||||
Token: "token",
|
Token: "token",
|
||||||
})
|
}, protos.ExecutionContext{})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Equal(t, 2, count)
|
assert.Equal(t, 2, count)
|
||||||
}
|
}
|
||||||
|
@ -96,7 +96,7 @@ func uploadRetriesFor(t *testing.T, statusCode int) {
|
||||||
err := Upload(ctx, &config.Config{
|
err := Upload(ctx, &config.Config{
|
||||||
Host: server.URL,
|
Host: server.URL,
|
||||||
Token: "token",
|
Token: "token",
|
||||||
})
|
}, protos.ExecutionContext{})
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Equal(t, 2, count)
|
assert.Equal(t, 2, count)
|
||||||
}
|
}
|
||||||
|
@ -143,7 +143,7 @@ func TestTelemetryUploadMaxRetries(t *testing.T) {
|
||||||
err := Upload(ctx, &config.Config{
|
err := Upload(ctx, &config.Config{
|
||||||
Host: server.URL,
|
Host: server.URL,
|
||||||
Token: "token",
|
Token: "token",
|
||||||
})
|
}, protos.ExecutionContext{})
|
||||||
assert.EqualError(t, err, "failed to upload telemetry logs after three attempts")
|
assert.EqualError(t, err, "failed to upload telemetry logs after three attempts")
|
||||||
assert.Equal(t, 3, count)
|
assert.Equal(t, 3, count)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue