pulumi/tests/integration/partial_state/partial_state_test.go

112 lines
4.4 KiB
Go

// Copyright 2018-2024, Pulumi Corporation.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//go:build (nodejs || all) && !xplatform_acceptance
package ints
import (
"testing"
"github.com/pulumi/pulumi/pkg/v3/resource/deploy/providers"
"github.com/pulumi/pulumi/pkg/v3/testing/integration"
"github.com/pulumi/pulumi/sdk/v3/go/common/resource"
"github.com/stretchr/testify/assert"
)
// TestPartialState tests that the engine persists partial state of a resource if a provider
// provides partial state alongside a resource creation or update error.
//
// The setup of this test uses a dynamic provider that will partially fail if a resource's state
// value is the number 4.
//
//nolint:paralleltest // ProgramTest calls t.Parallel()
func TestPartialState(t *testing.T) {
integration.ProgramTest(t, &integration.ProgramTestOptions{
Dir: "step1",
Dependencies: []string{"@pulumi/pulumi"},
Quick: true,
ExpectFailure: true,
ExtraRuntimeValidation: func(t *testing.T, stackInfo integration.RuntimeValidationStackInfo) {
// The first update tries to create a resource with state 4. This fails partially.
assert.NotNil(t, stackInfo.Deployment)
assert.Equal(t, 3, len(stackInfo.Deployment.Resources))
stackRes := stackInfo.Deployment.Resources[0]
assert.Equal(t, resource.RootStackType, stackRes.URN.Type())
providerRes := stackInfo.Deployment.Resources[1]
assert.True(t, providers.IsProviderType(providerRes.URN.Type()))
a := stackInfo.Deployment.Resources[2]
// We should still have persisted the resource and its outputs to the snapshot
assert.Equal(t, "doomed", a.URN.Name())
assert.Equal(t, 4.0, a.Outputs["state"].(float64))
assert.Equal(t, []string{"state can't be 4"}, a.InitErrors)
},
EditDirs: []integration.EditDir{
{
Dir: "step2",
Additive: true,
ExtraRuntimeValidation: func(t *testing.T, stackInfo integration.RuntimeValidationStackInfo) {
// The next update deletes the resource. We should successfully delete it.
assert.NotNil(t, stackInfo.Deployment)
assert.Equal(t, 1, len(stackInfo.Deployment.Resources))
stackRes := stackInfo.Deployment.Resources[0]
assert.Equal(t, resource.RootStackType, stackRes.URN.Type())
},
},
{
Dir: "step3",
Additive: true,
ExtraRuntimeValidation: func(t *testing.T, stackInfo integration.RuntimeValidationStackInfo) {
// Step 3 creates a resource with state 5, which succeeds.
assert.NotNil(t, stackInfo.Deployment)
assert.Equal(t, 3, len(stackInfo.Deployment.Resources))
stackRes := stackInfo.Deployment.Resources[0]
assert.Equal(t, resource.RootStackType, stackRes.URN.Type())
providerRes := stackInfo.Deployment.Resources[1]
assert.True(t, providers.IsProviderType(providerRes.URN.Type()))
a := stackInfo.Deployment.Resources[2]
assert.Equal(t, "not-doomed", a.URN.Name())
assert.Equal(t, 5.0, a.Outputs["state"].(float64))
assert.Nil(t, nil)
},
},
{
Dir: "step4",
Additive: true,
ExpectFailure: true,
ExtraRuntimeValidation: func(t *testing.T, stackInfo integration.RuntimeValidationStackInfo) {
// Step 4 updates the resource to have state 4, which fails partially.
assert.NotNil(t, stackInfo.Deployment)
assert.Equal(t, 3, len(stackInfo.Deployment.Resources))
stackRes := stackInfo.Deployment.Resources[0]
assert.Equal(t, resource.RootStackType, stackRes.URN.Type())
providerRes := stackInfo.Deployment.Resources[1]
assert.True(t, providers.IsProviderType(providerRes.URN.Type()))
a := stackInfo.Deployment.Resources[2]
// We should have persisted the updated resource's new outputs
// to the snapshot.
assert.Equal(t, "not-doomed", a.URN.Name())
assert.Equal(t, 4.0, a.Outputs["state"].(float64))
assert.Equal(t, []string{"state can't be 4"}, a.InitErrors)
},
},
},
})
}