mirror of https://github.com/databricks/cli.git
added and fixed tests
This commit is contained in:
parent
bb58bd1341
commit
67557b7c84
|
@ -67,7 +67,12 @@ func rewriteComputeIdToClusterId(v dyn.Value, p dyn.Path) (dyn.Value, diag.Diagn
|
||||||
// Drop the "compute_id" key.
|
// Drop the "compute_id" key.
|
||||||
vout, err := dyn.Walk(nv, func(p dyn.Path, v dyn.Value) (dyn.Value, error) {
|
vout, err := dyn.Walk(nv, func(p dyn.Path, v dyn.Value) (dyn.Value, error) {
|
||||||
switch len(p) {
|
switch len(p) {
|
||||||
case 0, 1:
|
case 0:
|
||||||
|
return v, nil
|
||||||
|
case 1:
|
||||||
|
if p[0] == dyn.Key("compute_id") {
|
||||||
|
return v, dyn.ErrDrop
|
||||||
|
}
|
||||||
return v, nil
|
return v, nil
|
||||||
case 2:
|
case 2:
|
||||||
if p[1] == dyn.Key("compute_id") {
|
if p[1] == dyn.Key("compute_id") {
|
||||||
|
|
|
@ -41,8 +41,13 @@ func TestComputeIdToClusterIdInTargetOverride(t *testing.T) {
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
diags := bundle.Apply(context.Background(), b, bundle.Seq(mutator.ComputeIdToClusterId(), mutator.SelectTarget("dev")))
|
diags := bundle.Apply(context.Background(), b, mutator.ComputeIdToClusterId())
|
||||||
assert.NoError(t, diags.Error())
|
assert.NoError(t, diags.Error())
|
||||||
|
assert.Empty(t, b.Config.Targets["dev"].ComputeId)
|
||||||
|
|
||||||
|
diags = diags.Extend(bundle.Apply(context.Background(), b, mutator.SelectTarget("dev")))
|
||||||
|
assert.NoError(t, diags.Error())
|
||||||
|
|
||||||
assert.Equal(t, "compute-id-dev", b.Config.Bundle.ClusterId)
|
assert.Equal(t, "compute-id-dev", b.Config.Bundle.ClusterId)
|
||||||
assert.Empty(t, b.Config.Bundle.ComputeId)
|
assert.Empty(t, b.Config.Bundle.ComputeId)
|
||||||
|
|
||||||
|
|
|
@ -71,6 +71,7 @@ func TestConvertCluster(t *testing.T) {
|
||||||
"availability": "SPOT",
|
"availability": "SPOT",
|
||||||
},
|
},
|
||||||
"data_security_mode": "USER_ISOLATION",
|
"data_security_mode": "USER_ISOLATION",
|
||||||
|
"no_wait": true,
|
||||||
"node_type_id": "m5.xlarge",
|
"node_type_id": "m5.xlarge",
|
||||||
"autoscale": map[string]any{
|
"autoscale": map[string]any{
|
||||||
"min_workers": int64(1),
|
"min_workers": int64(1),
|
||||||
|
|
Loading…
Reference in New Issue