pulumi/pkg/codegen/docs/utils_test.go

61 lines
1.8 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.
// Pulling out some of the repeated strings tokens into constants would harm readability, so we just ignore the
// goconst linter's warning.
//
//nolint:lll, goconst
package docs
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestWbr(t *testing.T) {
t.Parallel()
assert.Equal(t, wbr(""), "")
assert.Equal(t, wbr("a"), "a")
assert.Equal(t, wbr("A"), "A")
assert.Equal(t, wbr("aa"), "aa")
assert.Equal(t, wbr("AA"), "AA")
assert.Equal(t, wbr("Ab"), "Ab")
assert.Equal(t, wbr("aB"), "a<wbr>B")
assert.Equal(t, wbr("fooBar"), "foo<wbr>Bar")
assert.Equal(t, wbr("fooBarBaz"), "foo<wbr>Bar<wbr>Baz")
}
func TestRemoveLeadingUnderscores(t *testing.T) {
t.Parallel()
tests := []struct {
input string
expected string
}{
{input: "", expected: ""},
{input: "root", expected: "root"},
{input: "pulumi_azure_native", expected: "pulumi_azure_native"},
{input: "_root.FooBuzz", expected: "root.FooBuzz"},
{input: "_pulumi_random.sub_module.Type", expected: "pulumi_random.sub_module.Type"},
{input: "Optional[Sequence[_meta.v1.module_name.FooBar]]", expected: "Optional[Sequence[meta.v1.module_name.FooBar]]"},
}
for _, test := range tests {
result := removeLeadingUnderscores(test.input)
assert.Equal(t, test.expected, result)
}
}