pulumi/sdk/go/pulumi/internals/outputs_test.go

66 lines
1.7 KiB
Go

// Copyright 2016-2022, 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.
package internals
import (
"context"
"errors"
"testing"
"github.com/pulumi/pulumi/sdk/v3/go/pulumi"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func await(out pulumi.Output) (interface{}, bool, bool, []pulumi.Resource, error) {
result, err := UnsafeAwaitOutput(context.Background(), out)
return result.Value, result.Known, result.Secret, result.Dependencies, err
}
func TestBasicOutputs(t *testing.T) {
t.Parallel()
ctx, err := pulumi.NewContext(context.Background(), pulumi.RunInfo{
Project: "proj",
Stack: "stack",
})
require.NoError(t, err)
// Just test basic resolve and reject functionality.
{
out, resolve, _ := ctx.NewOutput()
go func() {
resolve(42)
}()
v, known, secret, deps, err := await(out)
assert.Nil(t, err)
assert.True(t, known)
assert.False(t, secret)
assert.Nil(t, deps)
assert.NotNil(t, v)
assert.Equal(t, 42, v.(int))
}
{
out, _, reject := ctx.NewOutput()
go func() {
reject(errors.New("boom"))
}()
v, _, _, _, err := await(out)
assert.NotNil(t, err)
assert.Nil(t, v)
}
}