mirror of https://github.com/docker/cli.git
72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
/*Package skip provides functions for skipping a test and printing the source code
|
|
of the condition used to skip the test.
|
|
*/
|
|
package skip // import "gotest.tools/skip"
|
|
|
|
import (
|
|
"fmt"
|
|
"path"
|
|
"reflect"
|
|
"runtime"
|
|
"strings"
|
|
|
|
"gotest.tools/internal/format"
|
|
"gotest.tools/internal/source"
|
|
)
|
|
|
|
type skipT interface {
|
|
Skip(args ...interface{})
|
|
Log(args ...interface{})
|
|
}
|
|
|
|
type helperT interface {
|
|
Helper()
|
|
}
|
|
|
|
// BoolOrCheckFunc can be a bool or func() bool, other types will panic
|
|
type BoolOrCheckFunc interface{}
|
|
|
|
// If the condition expression evaluates to true, or the condition function returns
|
|
// true, skip the test.
|
|
// The skip message will contain the source code of the expression.
|
|
// Extra message text can be passed as a format string with args
|
|
func If(t skipT, condition BoolOrCheckFunc, msgAndArgs ...interface{}) {
|
|
if ht, ok := t.(helperT); ok {
|
|
ht.Helper()
|
|
}
|
|
switch check := condition.(type) {
|
|
case bool:
|
|
ifCondition(t, check, msgAndArgs...)
|
|
case func() bool:
|
|
if check() {
|
|
t.Skip(format.WithCustomMessage(getFunctionName(check), msgAndArgs...))
|
|
}
|
|
default:
|
|
panic(fmt.Sprintf("invalid type for condition arg: %T", check))
|
|
}
|
|
}
|
|
|
|
func getFunctionName(function func() bool) string {
|
|
funcPath := runtime.FuncForPC(reflect.ValueOf(function).Pointer()).Name()
|
|
return strings.SplitN(path.Base(funcPath), ".", 2)[1]
|
|
}
|
|
|
|
func ifCondition(t skipT, condition bool, msgAndArgs ...interface{}) {
|
|
if ht, ok := t.(helperT); ok {
|
|
ht.Helper()
|
|
}
|
|
if !condition {
|
|
return
|
|
}
|
|
const (
|
|
stackIndex = 2
|
|
argPos = 1
|
|
)
|
|
source, err := source.FormattedCallExprArg(stackIndex, argPos)
|
|
if err != nil {
|
|
t.Log(err.Error())
|
|
t.Skip(format.Message(msgAndArgs...))
|
|
}
|
|
t.Skip(format.WithCustomMessage(source, msgAndArgs...))
|
|
}
|