added and fixed tests

This commit is contained in:
Andrew Nester 2024-09-23 12:29:11 +02:00
parent bb58bd1341
commit 67557b7c84
No known key found for this signature in database
GPG Key ID: 12BC628A44B7DA57
3 changed files with 13 additions and 2 deletions

View File

@ -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") {

View File

@ -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)

View File

@ -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),