mirror of https://github.com/knative/caching.git
105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
/*
|
|
Copyright 2019 The Knative Authors
|
|
|
|
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 helpers
|
|
|
|
import (
|
|
"log"
|
|
"math/rand"
|
|
"strings"
|
|
"time"
|
|
"unicode"
|
|
|
|
"knative.dev/pkg/kmeta"
|
|
"knative.dev/pkg/test"
|
|
)
|
|
|
|
const (
|
|
letterBytes = "abcdefghijklmnopqrstuvwxyz"
|
|
randSuffixLen = 8
|
|
sep = '-'
|
|
testNamePrefix = "Test"
|
|
)
|
|
|
|
func init() {
|
|
// Properly seed the random number generator so RandomString() is actually random.
|
|
// Otherwise, rerunning tests will generate the same names for the test resources, causing conflicts with
|
|
// already existing resources.
|
|
seed := time.Now().UTC().UnixNano()
|
|
log.Printf("Using '%d' to seed the random number generator", seed)
|
|
rand.Seed(seed)
|
|
}
|
|
|
|
// ObjectPrefixForTest returns the name prefix for this test's random names.
|
|
func ObjectPrefixForTest(t test.T) string {
|
|
return MakeK8sNamePrefix(strings.TrimPrefix(t.Name(), testNamePrefix))
|
|
}
|
|
|
|
// ObjectNameForTest generates a random object name based on the test name.
|
|
func ObjectNameForTest(t test.T) string {
|
|
return kmeta.ChildName(ObjectPrefixForTest(t), string(sep)+RandomString())
|
|
}
|
|
|
|
// AppendRandomString will generate a random string that begins with prefix.
|
|
// This is useful if you want to make sure that your tests can run at the same
|
|
// time against the same environment without conflicting.
|
|
// This method will use "-" as the separator between the prefix and
|
|
// the random suffix.
|
|
func AppendRandomString(prefix string) string {
|
|
return strings.Join([]string{prefix, RandomString()}, string(sep))
|
|
}
|
|
|
|
// RandomString will generate a random string.
|
|
func RandomString() string {
|
|
suffix := make([]byte, randSuffixLen)
|
|
|
|
for i := range suffix {
|
|
suffix[i] = letterBytes[rand.Intn(len(letterBytes))]
|
|
}
|
|
return string(suffix)
|
|
}
|
|
|
|
// MakeK8sNamePrefix converts each chunk of non-alphanumeric character into a single dash
|
|
// and also convert camelcase tokens into dash-delimited lowercase tokens.
|
|
func MakeK8sNamePrefix(s string) string {
|
|
var sb strings.Builder
|
|
newToken := false
|
|
for _, c := range s {
|
|
if !(unicode.IsLetter(c) || unicode.IsNumber(c)) {
|
|
newToken = true
|
|
continue
|
|
}
|
|
if sb.Len() > 0 && (newToken || unicode.IsUpper(c)) {
|
|
sb.WriteRune(sep)
|
|
}
|
|
sb.WriteRune(unicode.ToLower(c))
|
|
newToken = false
|
|
}
|
|
return sb.String()
|
|
}
|
|
|
|
// GetBaseFuncName returns the baseFuncName parsed from the fullFuncName.
|
|
// eg. test/e2e.TestMain will return TestMain.
|
|
func GetBaseFuncName(fullFuncName string) string {
|
|
name := fullFuncName
|
|
// Possibly there is no parent package, so only remove it from the name if '/' exists
|
|
if strings.ContainsRune(name, '/') {
|
|
name = name[strings.LastIndex(name, "/")+1:]
|
|
}
|
|
name = name[strings.LastIndex(name, ".")+1:]
|
|
return name
|
|
}
|