Fix prometheus test (#49)

Co-authored-by: Arun Gopalpuri <[email protected]>
labstack · Apr 6, 2021 · 9d2f96a · 9d2f96a
1 parent 304bbdf
commit 9d2f96a
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/prometheus/prometheus_test.go b/prometheus/prometheus_test.go
@@ -127,8 +127,10 @@ func TestMetricsPushGateway(t *testing.T) {
 	p := NewPrometheus("echo", nil)
 	p.Use(e)
 
-	result := p.getMetrics()
-
-	assert.Contains(t, string(result), fmt.Sprintf("%s_request_duration", p.Subsystem))
+	g := gofight.New()
+	g.GET(p.MetricsPath).Run(e, func(r gofight.HTTPResponse, rq gofight.HTTPRequest) {
+		assert.Equal(t, http.StatusOK, r.Code)
+		assert.NotContains(t, r.Body.String(), fmt.Sprintf("%s_request_duration", p.Subsystem))
+	})
 	unregister(p)
 }