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