Use defer in non-loop

This commit is contained in:
CatherineF-dev 2021-11-12 23:03:38 +00:00
parent a30af261f1
commit 49d341aa2b

View File

@ -33,6 +33,7 @@ import (
func TestRecordOperation(t *testing.T) { func TestRecordOperation(t *testing.T) {
// Use local registry // Use local registry
var registry = compbasemetrics.NewKubeRegistry() var registry = compbasemetrics.NewKubeRegistry()
defer registry.Reset()
registry.MustRegister(metrics.RuntimeOperations) registry.MustRegister(metrics.RuntimeOperations)
registry.MustRegister(metrics.RuntimeOperationsDuration) registry.MustRegister(metrics.RuntimeOperationsDuration)
registry.MustRegister(metrics.RuntimeOperationsErrors) registry.MustRegister(metrics.RuntimeOperationsErrors)
@ -64,8 +65,6 @@ func TestRecordOperation(t *testing.T) {
assert.HTTPBodyContains(t, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { assert.HTTPBodyContains(t, http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
mux.ServeHTTP(w, r) mux.ServeHTTP(w, r)
}), "GET", prometheusURL, nil, runtimeOperationsDurationExpected) }), "GET", prometheusURL, nil, runtimeOperationsDurationExpected)
registry.Reset()
} }
func TestInstrumentedVersion(t *testing.T) { func TestInstrumentedVersion(t *testing.T) {