diff --git a/connect/manager.go b/connect/manager.go index 69e66d336f89a95f94a953ae6dad179a4e48bca6..817041ce05ae1a6f25aa6f5899cac502cc4be1ac 100644 --- a/connect/manager.go +++ b/connect/manager.go @@ -33,9 +33,16 @@ func newManager() *Manager { } } -func NewManagerTesting(t *testing.T) *Manager { - if t == nil { - jww.FATAL.Panicf("NewMangerTesting is for testing only") +func NewManagerTesting(i interface{}) *Manager { + switch i.(type) { + case *testing.T: + break + case *testing.M: + break + case *testing.B: + break + default: + jww.FATAL.Panicf("NewManagerTesting is restricted to testing only. Got %T", i) } return newManager() }