mirror of https://github.com/pulumi/pulumi.git
115 lines
3.4 KiB
Go
115 lines
3.4 KiB
Go
// Copyright 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.
|
|
|
|
package lifecycletest
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/blang/semver"
|
|
"github.com/stretchr/testify/assert"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/credentials/insecure"
|
|
|
|
"github.com/pulumi/pulumi/pkg/v3/codegen/schema"
|
|
. "github.com/pulumi/pulumi/pkg/v3/engine" //nolint:revive
|
|
lt "github.com/pulumi/pulumi/pkg/v3/engine/lifecycletest/framework"
|
|
"github.com/pulumi/pulumi/pkg/v3/resource/deploy/deploytest"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/resource/plugin"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/util/contract"
|
|
"github.com/pulumi/pulumi/sdk/v3/go/common/util/rpcutil"
|
|
codegenrpc "github.com/pulumi/pulumi/sdk/v3/proto/go/codegen"
|
|
)
|
|
|
|
func TestLoader(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
expectedSpec := schema.PackageSpec{
|
|
Name: "pkgA",
|
|
Version: "1.0.0",
|
|
Resources: map[string]schema.ResourceSpec{
|
|
"pkgA:index:Resource": {
|
|
ObjectTypeSpec: schema.ObjectTypeSpec{
|
|
Description: "Resource",
|
|
Properties: map[string]schema.PropertySpec{
|
|
"foo": {
|
|
TypeSpec: schema.TypeSpec{Type: "string"},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
loaders := []*deploytest.ProviderLoader{
|
|
deploytest.NewProviderLoader("pkgA", semver.MustParse("1.0.0"), func() (plugin.Provider, error) {
|
|
return &deploytest.Provider{
|
|
GetSchemaF: func(ctx context.Context, req plugin.GetSchemaRequest) (plugin.GetSchemaResponse, error) {
|
|
bytes, err := json.Marshal(expectedSpec)
|
|
if err != nil {
|
|
contract.Failf("marshal schema: %v", err)
|
|
}
|
|
|
|
return plugin.GetSchemaResponse{
|
|
Schema: bytes,
|
|
}, nil
|
|
},
|
|
}, nil
|
|
}),
|
|
}
|
|
|
|
programF := deploytest.NewLanguageRuntimeF(func(info plugin.RunInfo, monitor *deploytest.ResourceMonitor) error {
|
|
// Check we can connect to the schema loader and query a schema.
|
|
conn, err := grpc.NewClient(
|
|
info.LoaderAddress,
|
|
grpc.WithTransportCredentials(insecure.NewCredentials()),
|
|
rpcutil.GrpcChannelOptions(),
|
|
)
|
|
if err != nil {
|
|
return fmt.Errorf("could not connect to resource monitor: %w", err)
|
|
}
|
|
defer conn.Close()
|
|
loader := codegenrpc.NewLoaderClient(conn)
|
|
|
|
ctx := context.Background()
|
|
resp, err := loader.GetSchema(ctx, &codegenrpc.GetSchemaRequest{
|
|
Package: "pkgA",
|
|
})
|
|
if err != nil {
|
|
return fmt.Errorf("could not get schema: %w", err)
|
|
}
|
|
|
|
var actualSpec schema.PackageSpec
|
|
err = json.Unmarshal(resp.Schema, &actualSpec)
|
|
if err != nil {
|
|
return fmt.Errorf("could not unmarshal schema: %w", err)
|
|
}
|
|
|
|
// Check the schema fields match
|
|
assert.Equal(t, expectedSpec.Name, actualSpec.Name)
|
|
|
|
return nil
|
|
})
|
|
hostF := deploytest.NewPluginHostF(nil, nil, programF, loaders...)
|
|
|
|
p := <.TestPlan{
|
|
Options: lt.TestUpdateOptions{T: t, HostF: hostF, SkipDisplayTests: true},
|
|
}
|
|
_, err := lt.TestOp(Update).RunStep(p.GetProject(), p.GetTarget(t, nil), p.Options, false, p.BackendClient, nil, "0")
|
|
assert.NoError(t, err)
|
|
}
|