103 lines
2.7 KiB
Go
103 lines
2.7 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"),
|
|
},
|
|
},
|
|
{
|
|
"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)
|
|
|
|
expectedFiles, err := test.LoadFiles(filepath.Join(testDir, tt.schemaDir), "python", tt.expectedFiles)
|
|
assert.NoError(t, err)
|
|
|
|
test.ValidateFileEquality(t, files, expectedFiles)
|
|
})
|
|
}
|
|
}
|