|
|
|
@@ -206,3 +206,37 @@ func TestMetricsMiddleware_RequestDuration(t *testing.T) { |
|
|
|
|
|
|
|
assert.Equal(t, 1, testutil.CollectAndCount(m.RequestDuration)) |
|
|
|
} |
|
|
|
|
|
|
|
func TestMetricsEndpoint_Format(t *testing.T) { |
|
|
|
handler := NewMetricsHandler() |
|
|
|
|
|
|
|
req := httptest.NewRequest("GET", "/metrics", nil) |
|
|
|
w := httptest.NewRecorder() |
|
|
|
handler.ServeHTTP(w, req) |
|
|
|
|
|
|
|
assert.Equal(t, 200, w.Code) |
|
|
|
assert.Contains(t, w.Header().Get("Content-Type"), "text/plain") |
|
|
|
} |
|
|
|
|
|
|
|
func TestMetricsEndpoint_WithData(t *testing.T) { |
|
|
|
reg := newTestRegistry() |
|
|
|
_ = NewMetrics(reg) |
|
|
|
|
|
|
|
handler := NewMetricsHandlerWithRegistry(reg) |
|
|
|
|
|
|
|
req := httptest.NewRequest("GET", "/metrics", nil) |
|
|
|
w := httptest.NewRecorder() |
|
|
|
handler.ServeHTTP(w, req) |
|
|
|
|
|
|
|
assert.Equal(t, 200, w.Code) |
|
|
|
} |
|
|
|
|
|
|
|
func TestMetricsEndpoint_EmptyNoPanic(t *testing.T) { |
|
|
|
handler := NewMetricsHandler() |
|
|
|
|
|
|
|
req := httptest.NewRequest("GET", "/metrics", nil) |
|
|
|
w := httptest.NewRecorder() |
|
|
|
handler.ServeHTTP(w, req) |
|
|
|
|
|
|
|
assert.Equal(t, 200, w.Code) |
|
|
|
} |