pulumi/pkg/codegen/dotnet/gen_test.go
Komal a259d59624
[codegen/dotnet,nodejs] - Improve makeSafeEnumName (#5711)
* Improve makeSafeEnumName

* Improve nodejs enumName generator

* Fix comment

* PR feedback
2020-11-09 11:33:22 -08:00

81 lines
2 KiB
Go

package dotnet
import (
"path/filepath"
"testing"
"github.com/pulumi/pulumi/pkg/v2/codegen/internal/test"
"github.com/stretchr/testify/assert"
)
func TestGeneratePackage(t *testing.T) {
tests := []struct {
name string
schemaDir string
expectedFiles []string
}{
{
"Simple schema with local resource properties",
"simple-resource-schema",
[]string{
"Resource.cs",
"OtherResource.cs",
"ArgFunction.cs",
},
},
{
"Simple schema with enum types",
"simple-enum-schema",
[]string{
"Tree/V1/RubberTree.cs",
"Tree/V1/Enums.cs",
"Enums.cs",
"Inputs/ContainerArgs.cs",
"Outputs/Container.cs",
},
},
}
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), "dotnet", tt.expectedFiles)
assert.NoError(t, err)
test.ValidateFileEquality(t, files, expectedFiles)
})
}
}
func TestMakeSafeEnumName(t *testing.T) {
tests := []struct {
input string
expected string
wantErr bool
}{
{"+", "", true},
{"*", "Asterisk", false},
{"0", "Zero", false},
{"Microsoft-Windows-Shell-Startup", "Microsoft_Windows_Shell_Startup", false},
{"Microsoft.Batch", "Microsoft_Batch", false},
{"readonly", "@Readonly", false},
{"SystemAssigned, UserAssigned", "SystemAssigned_UserAssigned", false},
{"Dev(NoSLA)_Standard_D11_v2", "Dev_NoSLA_Standard_D11_v2", false},
{"Standard_E8as_v4+1TB_PS", "Standard_E8as_v4_1TB_PS", false},
}
for _, tt := range tests {
t.Run(tt.input, func(t *testing.T) {
got, err := makeSafeEnumName(tt.input)
if (err != nil) != tt.wantErr {
t.Errorf("makeSafeEnumName() error = %v, wantErr %v", err, tt.wantErr)
return
}
if got != tt.expected {
t.Errorf("makeSafeEnumName() got = %v, want %v", got, tt.expected)
}
})
}
}