7b1d6ec1ac
These changes support arbitrary combinations of input + plain types within a schema. Handling plain types at the property level was not sufficient to support such combinations. Reifying these types required updating quite a bit of code. This is likely to have caused some temporary complications, but should eventually lead to substantial simplification in the SDK and program code generators. With the new design, input and optional types are explicit in the schema type system. Optionals will only appear at the outermost level of a type (i.e. Input<Optional<>>, Array<Optional<>>, etc. will not occur). In addition to explicit input types, each object type now has a "plain" shape and an "input" shape. The former uses only plain types; the latter uses input shapes wherever a plain type is not specified. Plain types are indicated in the schema by setting the "plain" property of a type spec to true.
112 lines
3.2 KiB
Go
112 lines
3.2 KiB
Go
// Copyright 2016-2020, 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 test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/pulumi/pulumi/pkg/v3/codegen/schema"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
// GenPkgSignature corresponds to the shape of the codegen GeneratePackage functions.
|
|
type GenPkgSignature func(string, *schema.Package, map[string][]byte) (map[string][]byte, error)
|
|
|
|
// GeneratePackageFilesFromSchema loads a schema and generates files using the provided GeneratePackage function.
|
|
func GeneratePackageFilesFromSchema(schemaPath string, genPackageFunc GenPkgSignature) (map[string][]byte, error) {
|
|
// Read in, decode, and import the schema.
|
|
schemaBytes, err := ioutil.ReadFile(schemaPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var pkgSpec schema.PackageSpec
|
|
err = json.Unmarshal(schemaBytes, &pkgSpec)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
pkg, err := schema.ImportSpec(pkgSpec, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return genPackageFunc("test", pkg, nil)
|
|
}
|
|
|
|
// LoadFiles loads the provided list of files from a directory.
|
|
func LoadFiles(dir, lang string, files []string) (map[string][]byte, error) {
|
|
result := map[string][]byte{}
|
|
for _, file := range files {
|
|
fileBytes, err := ioutil.ReadFile(filepath.Join(dir, lang, file))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
result[file] = fileBytes
|
|
}
|
|
|
|
return result, nil
|
|
}
|
|
|
|
// ValidateFileEquality compares maps of files for equality.
|
|
func ValidateFileEquality(t *testing.T, actual, expected map[string][]byte) {
|
|
for name, file := range expected {
|
|
assert.Contains(t, actual, name)
|
|
assert.Equal(t, string(file), string(actual[name]), name)
|
|
}
|
|
for name := range actual {
|
|
if _, ok := expected[name]; !ok {
|
|
t.Logf("missing data for %s", name)
|
|
}
|
|
}
|
|
}
|
|
|
|
// If PULUMI_ACCEPT is set, writes out actual output to th expected
|
|
// file set, so we can continue enjoying golden tests without manually
|
|
// modifying the expected output.
|
|
func RewriteFilesWhenPulumiAccept(t *testing.T, dir, lang string, actual map[string][]byte) {
|
|
if os.Getenv("PULUMI_ACCEPT") != "" {
|
|
for file, bytes := range actual {
|
|
err := ioutil.WriteFile(filepath.Join(dir, lang, file), bytes, 0600)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// CheckAllFilesGenerated ensures that the set of expected and actual files generated
|
|
// are exactly equivalent.
|
|
func CheckAllFilesGenerated(t *testing.T, actual, expected map[string][]byte) {
|
|
seen := map[string]bool{}
|
|
for x := range expected {
|
|
seen[x] = true
|
|
}
|
|
for a := range actual {
|
|
assert.Contains(t, seen, a, "Unexpected file generated: %s", a)
|
|
if seen[a] {
|
|
delete(seen, a)
|
|
}
|
|
}
|
|
|
|
for s := range seen {
|
|
assert.Fail(t, "No content generated for expected file %s", s)
|
|
}
|
|
}
|