Matthias Radestock
1374e4c4de
It breaks stack traces on failure. We can just rely on running `go test -timeout ...` instead, which we already do in tools/test.
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package testing
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"reflect"
|
|
"runtime"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func AssertErrorInterface(t *testing.T, expected interface{}, actual error, desc string) {
|
|
require.Error(t, actual, desc)
|
|
require.Implements(t, expected, actual, desc)
|
|
}
|
|
|
|
func AssertErrorType(t *testing.T, expected interface{}, actual error, desc string) {
|
|
require.Error(t, actual, desc)
|
|
// require.IsType doesn't take the pointer element, and doesn't resolve
|
|
// through interfaces, so we have to do this one ourselves.
|
|
actualT, expectedT := reflect.TypeOf(actual), reflect.TypeOf(expected).Elem()
|
|
if actualT != expectedT {
|
|
require.FailNow(
|
|
t,
|
|
fmt.Sprintf("Expected %s but got %s", expectedT.String(), actualT.String()),
|
|
desc,
|
|
)
|
|
}
|
|
}
|
|
|
|
func stackTrace(all bool) string {
|
|
buf := make([]byte, 1<<20)
|
|
stacklen := runtime.Stack(buf, all)
|
|
return string(buf[:stacklen])
|
|
}
|
|
|
|
// TrimTestArgs finds the first -- in os.Args and trim all args before that
|
|
func TrimTestArgs() {
|
|
i, l := 0, len(os.Args)
|
|
for ; i < l; i++ {
|
|
if os.Args[i] == "--" {
|
|
break
|
|
}
|
|
}
|
|
if i == l {
|
|
panic("Specify weave args after --")
|
|
}
|
|
os.Args = append(os.Args[:1], os.Args[i+1:l]...)
|
|
}
|