pulumi/sdk/go/common/tokens/tokens_test.go

109 lines
2.6 KiB
Go

// Copyright 2016-2018, 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.
package tokens
import (
"testing"
"github.com/stretchr/testify/assert"
)
func TestTokens(t *testing.T) {
t.Parallel()
// Package tokens/names.
p := "test/package"
assert.False(t, IsName(p))
assert.True(t, IsQName(p))
pkg := NewPackageToken(PackageName(p))
assert.Equal(t, p, pkg.Name().String())
assert.Equal(t, p, pkg.String())
p2 := "test/my-package"
assert.False(t, IsName(p2))
assert.True(t, IsQName(p2))
pkg2 := NewPackageToken(PackageName(p2))
assert.Equal(t, p2, pkg2.Name().String())
assert.Equal(t, p2, pkg2.String())
// Module tokens/names.
m := "my/module"
assert.False(t, IsName(m))
assert.True(t, IsQName(m))
mod := NewModuleToken(pkg, ModuleName(m))
assert.Equal(t, m, mod.Name().String())
assert.Equal(t, p, mod.Package().Name().String())
assert.Equal(t, p+TokenDelimiter+m, mod.String())
// Module member tokens/names.
mm := "memby"
assert.True(t, IsName(mm))
assert.True(t, IsQName(mm))
modm := NewModuleMemberToken(mod, ModuleMemberName(mm))
assert.Equal(t, mm, modm.Name().String())
assert.Equal(t, m, modm.Module().Name().String())
assert.Equal(t, p, modm.Module().Package().Name().String())
assert.Equal(t, p+TokenDelimiter+m+TokenDelimiter+mm, modm.String())
}
func TestTypeDisplayName(t *testing.T) {
t.Parallel()
tests := []struct {
desc string
give Type
want string
}{
{
desc: "not enough parts",
give: "incomplete",
want: "incomplete",
},
{
desc: "no name",
give: "pkg:mod:",
want: "pkg:mod:",
},
{
desc: "no slash",
give: "pkg:mod:typ",
want: "pkg:mod:typ",
},
{
desc: "bad casing",
give: "pkg:Mod/foo:typ",
want: "pkg:Mod/foo:typ",
},
{
desc: "remove slash",
give: "pkg:mod/foo/bar:Bar",
want: "pkg:mod/foo:Bar",
},
{
desc: "remove up to last slash",
give: "pkg:mod/foo/bar/baz:Baz",
want: "pkg:mod/foo/bar:Baz",
},
}
for _, tt := range tests {
tt := tt
t.Run(tt.desc, func(t *testing.T) {
t.Parallel()
assert.Equal(t, tt.want, tt.give.DisplayName())
})
}
}