Add T interface #1
3 changed files with 10 additions and 5 deletions
|
@ -3,14 +3,13 @@ package assert
|
|||
import (
|
||||
"reflect"
|
||||
"runtime"
|
||||
"testing"
|
||||
)
|
||||
|
||||
// Assert is a helper for tests
|
||||
type Assert func(bool, ...interface{})
|
||||
|
||||
// New returns a new Assert
|
||||
func New(t *testing.T) Assert {
|
||||
func New(t T) Assert {
|
||||
a := func(ok bool, msg ...interface{}) {
|
||||
if !ok {
|
||||
if msg == nil {
|
||||
|
|
|
@ -5,7 +5,6 @@ import (
|
|||
"reflect"
|
||||
"runtime"
|
||||
"strconv"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func prepMsg(msg []interface{}, format string, args ...interface{}) []interface{} {
|
||||
|
@ -16,9 +15,9 @@ func shell(i int) string {
|
|||
return "\x1B[" + strconv.Itoa(i) + "m"
|
||||
}
|
||||
|
||||
var ts = map[*runtime.Func]*testing.T{}
|
||||
var ts = map[*runtime.Func]T{}
|
||||
|
||||
func t(a Assert) *testing.T {
|
||||
func t(a Assert) T {
|
||||
f := runtime.FuncForPC(reflect.ValueOf(a).Pointer())
|
||||
return ts[f]
|
||||
}
|
||||
|
|
7
type.go
Normal file
7
type.go
Normal file
|
@ -0,0 +1,7 @@
|
|||
package assert
|
||||
|
||||
// T is an interface of what we use from testing.T
|
||||
type T interface {
|
||||
Error(...interface{})
|
||||
Helper()
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue