73c8cc3ec5
This commit adds a new language option for Python generation to specify the package name instead of using `pulumi_x` where x is the name defined in the schema. A new test is added, and this has also been shown to produce no diff when run against `pulumi-eks`.
117 lines
3.1 KiB
Go
117 lines
3.1 KiB
Go
package python
|
|
|
|
import (
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/pulumi/pulumi/pkg/v3/codegen/internal/test"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
var pathTests = []struct {
|
|
input string
|
|
expected string
|
|
}{
|
|
{".", "."},
|
|
{"", "."},
|
|
{"../", ".."},
|
|
{"../..", "..."},
|
|
{"../../..", "...."},
|
|
{"something", ".something"},
|
|
{"../parent", "..parent"},
|
|
{"../../module", "...module"},
|
|
}
|
|
|
|
func TestRelPathToRelImport(t *testing.T) {
|
|
for _, tt := range pathTests {
|
|
t.Run(tt.input, func(t *testing.T) {
|
|
result := relPathToRelImport(tt.input)
|
|
if result != tt.expected {
|
|
t.Errorf("expected \"%s\"; got \"%s\"", tt.expected, result)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGeneratePackage(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
schemaDir string
|
|
expectedFiles []string
|
|
}{
|
|
{
|
|
"Simple schema with local resource properties",
|
|
"simple-resource-schema",
|
|
[]string{
|
|
filepath.Join("pulumi_example", "resource.py"),
|
|
filepath.Join("pulumi_example", "other_resource.py"),
|
|
filepath.Join("pulumi_example", "arg_function.py"),
|
|
},
|
|
},
|
|
{
|
|
"Simple schema with local resource properties and custom Python package name",
|
|
"simple-resource-schema-custom-pypackage-name",
|
|
[]string{
|
|
filepath.Join("custom_py_package", "resource.py"),
|
|
filepath.Join("custom_py_package", "other_resource.py"),
|
|
filepath.Join("custom_py_package", "arg_function.py"),
|
|
},
|
|
},
|
|
{
|
|
"External resource schema",
|
|
"external-resource-schema",
|
|
[]string{
|
|
filepath.Join("pulumi_example", "_inputs.py"),
|
|
filepath.Join("pulumi_example", "arg_function.py"),
|
|
filepath.Join("pulumi_example", "cat.py"),
|
|
filepath.Join("pulumi_example", "component.py"),
|
|
filepath.Join("pulumi_example", "workload.py"),
|
|
},
|
|
},
|
|
{
|
|
"Simple schema with enum types",
|
|
"simple-enum-schema",
|
|
[]string{
|
|
filepath.Join("pulumi_plant", "_enums.py"),
|
|
filepath.Join("pulumi_plant", "_inputs.py"),
|
|
filepath.Join("pulumi_plant", "outputs.py"),
|
|
filepath.Join("pulumi_plant", "__init__.py"),
|
|
filepath.Join("pulumi_plant", "tree", "__init__.py"),
|
|
filepath.Join("pulumi_plant", "tree", "v1", "_enums.py"),
|
|
filepath.Join("pulumi_plant", "tree", "v1", "__init__.py"),
|
|
filepath.Join("pulumi_plant", "tree", "v1", "rubber_tree.py"),
|
|
filepath.Join("pulumi_plant", "tree", "v1", "nursery.py"),
|
|
},
|
|
},
|
|
{
|
|
"Simple schema with plain properties",
|
|
"simple-plain-schema",
|
|
[]string{
|
|
filepath.Join("pulumi_example", "_inputs.py"),
|
|
filepath.Join("pulumi_example", "component.py"),
|
|
filepath.Join("pulumi_example", "outputs.py"),
|
|
},
|
|
},
|
|
}
|
|
|
|
testDir := filepath.Join("..", "internal", "test", "testdata")
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
files, err := test.GeneratePackageFilesFromSchema(
|
|
filepath.Join(testDir, tt.schemaDir, "schema.json"), GeneratePackage)
|
|
assert.NoError(t, err)
|
|
|
|
dir := filepath.Join(testDir, tt.schemaDir)
|
|
lang := "python"
|
|
|
|
test.RewriteFilesWhenPulumiAccept(t, dir, lang, files)
|
|
|
|
expectedFiles, err := test.LoadFiles(dir, lang, tt.expectedFiles)
|
|
assert.NoError(t, err)
|
|
|
|
test.ValidateFileEquality(t, files, expectedFiles)
|
|
})
|
|
}
|
|
}
|