Add RecordCustomMetric to *Application

nrmock/application.go

@@ -14,7 +14,8 @@ type Application struct {
 	AllTransactions   []*Transaction
 	LatestTransaction *Transaction
 
-	CustomEvents map[string][]map[string]interface{}
+	CustomEvents  map[string][]map[string]interface{}
+	CustomMetrics map[string][]float64
 }
 
 // NewApplication creates a new *Application
@@ -53,6 +54,21 @@ func (a *Application) RecordCustomEvent(eventType string, params map[string]inte
 	return nil
 }
 
+// RecordCustomMetric records a custom metric.
+func (a *Application) RecordCustomMetric(name string, value float64) error {
+	if a.CustomMetrics == nil {
+		a.CustomMetrics = map[string][]float64{}
+	}
+
+	if a.CustomMetrics[name] == nil {
+		a.CustomMetrics[name] = []float64{}
+	}
+
+	a.CustomMetrics[name] = append(a.CustomMetrics[name], value)
+
+	return nil
+}
+
 // WaitForConnection does nothing
 func (a *Application) WaitForConnection(timeout time.Duration) error {
 	return nil

nrmock/application_test.go

@@ -38,6 +38,23 @@ func TestRecordCustomEvent(t *testing.T) {
 	}
 }
 
+func TestRecordCustomMetric(t *testing.T) {
+	app := &Application{}
+
+	app.RecordCustomMetric("foo", 12.3)
+	app.RecordCustomMetric("foo", 45.6)
+
+	foo := app.CustomMetrics["foo"]
+
+	if got, want := len(foo), 2; got != want {
+		t.Fatalf("len(foo) = %d, want %d", got, want)
+	}
+
+	if got, want := foo[1], 45.6; got != want {
+		t.Fatalf("foo[1] = %v, want %v", got, want)
+	}
+}
+
 func TestWaitForConnection(t *testing.T) {
 	app := &Application{}