func TestCallersMulti(t *testing.T) {
m := stack.CallersMulti(0)
const expected = "stack_test.go:35 TestCallersMulti"
first := m.Stacks()[0][0].String()
if !strings.HasSuffix(first, expected) {
t.Fatalf(`expected suffix "%s" got "%s"`, expected, first)
}
}
func TestCallersMultiWithTwo(t *testing.T) {
m := stack.CallersMulti(0)
m.AddCallers(0)
matches := []string{
"stack_test.go:44 +TestCallersMultiWithTwo$",
"",
"",
`\(Stack 2\)$`,
"stack_test.go:46 +TestCallersMultiWithTwo$",
}
match(t, m.String(), matches)
}
// WrapSkip may wrap the error and return an augmented error depending on the
// configuration in the context. The defaults result in the error being
// returned as is. It also handles nils correctly, returning a nil in that
// case. If the given error is already wrapped, it returns it as-is.
func WrapSkip(ctx context.Context, err error, skip int) error {
switch err := err.(type) {
case nil:
return nil
case *singleFrameError:
return err
case *singleStackError:
return err
case *multiStackError:
err.multiStack.AddCallers(skip + 1)
return err
}
config := ContextConfig(ctx)
switch config.StackMode {
case StackModeSingleFrame:
return &singleFrameError{
config: config,
underlying: err,
frame: stack.Caller(skip + 1),
}
case StackModeSingleStack:
return &singleStackError{
config: config,
underlying: err,
stack: stack.Callers(skip + 1),
}
case StackModeMultiStack:
return &multiStackError{
config: config,
underlying: err,
multiStack: stack.CallersMulti(skip + 1),
}
}
return err
}
func BenchmarkCallersMulti(b *testing.B) {
for i := 0; i < b.N; i++ {
stack.CallersMulti(0)
}
}