pulumi/pkg/codegen/python/python_test.go

package python

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

var pyNameTests = []struct {
	input    string
	expected string
	legacy   string
}{
	{"kubeletConfigKey", "kubelet_config_key", "kubelet_config_key"},
	{"podCIDR", "pod_cidr", "pod_cidr"},
	{"podCidr", "pod_cidr", "pod_cidr"},
	{"podCIDRs", "pod_cidrs", "pod_cid_rs"},
	{"podIPs", "pod_ips", "pod_i_ps"},
	{"nonResourceURLs", "non_resource_urls", "non_resource_ur_ls"},
	{"someTHINGsAREWeird", "some_things_are_weird", "some_thin_gs_are_weird"},
	{"podCIDRSet", "pod_cidr_set", "pod_cidr_set"},
	{"Sha256Hash", "sha256_hash", "sha256_hash"},
	{"SHA256Hash", "sha256_hash", "sha256_hash"},

	// PyName should return the legacy name for these:
	{"openXJsonSerDe", "open_x_json_ser_de", "open_x_json_ser_de"},
	{"GetPublicIPs", "get_public_i_ps", "get_public_i_ps"},
	{"GetUptimeCheckIPs", "get_uptime_check_i_ps", "get_uptime_check_i_ps"},
}

func TestPyName(t *testing.T) {
	for _, tt := range pyNameTests {
		t.Run(tt.input, func(t *testing.T) {
			// TODO[pulumi/pulumi#5201]: Once the assertion has been removed, we can remove this `if` block.
			// Prevent this input from panic'ing.
			if tt.input == "someTHINGsAREWeird" {
				result := pyName(tt.input, false /*legacy*/)
				assert.Equal(t, tt.expected, result)
				return
			}

			result := PyName(tt.input)
			assert.Equal(t, tt.expected, result)
		})
	}
}

func TestPyNameLegacy(t *testing.T) {
	for _, tt := range pyNameTests {
		t.Run(tt.input, func(t *testing.T) {
			result := PyNameLegacy(tt.input)
			assert.Equal(t, tt.legacy, result)
		})
	}
}