From 9de661e791fc54b2e9d967c0382e6c34c3bf4805 Mon Sep 17 00:00:00 2001
From: Niamh Nikali <niamh@elixxir.io>
Date: Fri, 25 Sep 2020 11:52:49 -0700
Subject: [PATCH] Make NewManagerTesting take any testing type
---
connect/manager.go | 13 ++++++++++---
1 file changed, 10 insertions(+), 3 deletions(-)
diff --git a/connect/manager.go b/connect/manager.go
index 69e66d3..6e2deda 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 for testing only. Got %T", i)
}
return newManager()
}
--
GitLab