Modify testing functions for metric

connect/host.go

@@ -206,7 +206,7 @@ func (h *Host) isExcludedMetricError(err string) bool {
 
 // Sets the host metrics to an arbitrary value. Used for testing
 // purposes only
-func (h *Host) SetMetricsTesting(errCounter uint64, face interface{}) {
+func (h *Host) SetMetricsTesting(m *Metric, face interface{}) {
 	// Ensure that this function is only run in testing environments
 	switch face.(type) {
 	case *testing.T, *testing.M, *testing.B:
@@ -215,7 +215,7 @@ func (h *Host) SetMetricsTesting(errCounter uint64, face interface{}) {
 		panic("SetMetricsTesting() can only be used for testing.")
 	}
 
-	h.metrics.errCounter = &errCounter
+	h.metrics.errCounter = m
 
 }
 

connect/metrics.go

@@ -11,6 +11,7 @@ package connect
 
 import (
 	"sync/atomic"
+	"testing"
 )
 
 type Metric struct {
@@ -27,6 +28,22 @@ func newMetric() *Metric {
 	}
 }
 
+// Creates a metrics object with specified values. Used for testing
+// purposes only
+func NewMetricTesting(errCounter uint64, face interface{}) *Metric  {
+	// Ensure that this function is only run in testing environments
+	switch face.(type) {
+	case *testing.T, *testing.M, *testing.B:
+	break
+	default:
+	panic("SetMetricsTesting() can only be used for testing.")
+	}
+
+	return &Metric{
+		errCounter: &errCounter,
+	}
+}
+
 // Getter for errCounter
 func (m *Metric) GetErrorCounter() uint64 {
 	return atomic.LoadUint64(m.errCounter)