From b8a96380f1805b2180a36d0edf30206d00f1848f Mon Sep 17 00:00:00 2001 From: josh <josh@elixxir.io> Date: Wed, 23 Sep 2020 08:49:25 -0700 Subject: [PATCH] Allow for multiple testing signatures in NewManagerTesting --- connect/manager.go | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/connect/manager.go b/connect/manager.go index 69e66d3..817041c 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() } -- GitLab