Skip to content

Commit

Permalink
reorders the defer to execute the close
Browse files Browse the repository at this point in the history
Signed-off-by: Afzal Ansari <afzal442@gmail.com>
  • Loading branch information
afzal442 committed Jul 27, 2023
1 parent a0dcb35 commit a8b1c3e
Showing 1 changed file with 10 additions and 9 deletions.
19 changes: 10 additions & 9 deletions plugin/metrics/prometheus/metricsstore/reader_test.go
Expand Up @@ -83,9 +83,9 @@ func TestNewMetricsReaderValidAddress(t *testing.T) {
ServerURL: "http://localhost:1234",
ConnectTimeout: defaultTimeout,
}, logger, tracer)
defer closer()
require.NoError(t, err)
assert.NotNil(t, reader)
defer closer()
}

func TestNewMetricsReaderInvalidAddress(t *testing.T) {
Expand All @@ -95,10 +95,10 @@ func TestNewMetricsReaderInvalidAddress(t *testing.T) {
ServerURL: "\n",
ConnectTimeout: defaultTimeout,
}, logger, tracer)
defer closer()
require.Error(t, err)
assert.Contains(t, err.Error(), "failed to initialize prometheus client")
assert.Nil(t, reader)
defer closer()
}

func TestGetMinStepDuration(t *testing.T) {
Expand All @@ -113,12 +113,12 @@ func TestGetMinStepDuration(t *testing.T) {
ServerURL: "http://" + listener.Addr().String(),
ConnectTimeout: defaultTimeout,
}, logger, tracer)
defer closer()
require.NoError(t, err)

minStep, err := reader.GetMinStepDuration(context.Background(), &params)
require.NoError(t, err)
assert.Equal(t, time.Millisecond, minStep)
defer closer()
}

func TestMetricsServerError(t *testing.T) {
Expand Down Expand Up @@ -148,6 +148,7 @@ func TestMetricsServerError(t *testing.T) {
ServerURL: "http://" + address,
ConnectTimeout: defaultTimeout,
}, logger, tracer)
defer closer()
require.NoError(t, err)

m, err := reader.GetCallRates(context.Background(), &params)
Expand All @@ -156,7 +157,6 @@ func TestMetricsServerError(t *testing.T) {
assert.Contains(t, err.Error(), "failed executing metrics query")
assert.NotEmpty(t, exp.GetSpans(), "Expected spans are recorded")
assert.Len(t, exp.GetSpans(), 1, "HTTP request was traced and span reported")
defer closer()
}

func TestGetLatencies(t *testing.T) {
Expand Down Expand Up @@ -254,11 +254,12 @@ func TestGetLatencies(t *testing.T) {
defer mockPrometheus.Close()

m, err := reader.GetLatencies(context.Background(), &params)
defer closer()
require.NoError(t, err)
assert.NotEmpty(t, exp.GetSpans(), "Spans recorded during the test.")
assert.Len(t, exp.GetSpans(), 1, "HTTP request was traced and span reported")
assertMetrics(t, m, tc.wantLabels, tc.wantName, tc.wantDescription)
defer closer()

})
}
}
Expand Down Expand Up @@ -355,11 +356,11 @@ func TestGetCallRates(t *testing.T) {
defer mockPrometheus.Close()

m, err := reader.GetCallRates(context.Background(), &params)
defer closer()
require.NoError(t, err)
assert.NotEmpty(t, exp.GetSpans(), "Spans recorded during the test.")
assert.Len(t, exp.GetSpans(), 1, "HTTP request was traced and span reported")
assertMetrics(t, m, tc.wantLabels, tc.wantName, tc.wantDescription)
defer closer()
})
}
}
Expand Down Expand Up @@ -481,11 +482,11 @@ func TestGetErrorRates(t *testing.T) {
defer mockPrometheus.Close()

m, err := reader.GetErrorRates(context.Background(), &params)
defer closer()
require.NoError(t, err)
assert.NotEmpty(t, exp.GetSpans(), "Spans recorded during the test.")
assert.Len(t, exp.GetSpans(), 1, "HTTP request was traced and span reported")
assertMetrics(t, m, tc.wantLabels, tc.wantName, tc.wantDescription)
defer closer()
})
}
}
Expand Down Expand Up @@ -515,11 +516,11 @@ func TestWarningResponse(t *testing.T) {
defer mockPrometheus.Close()

m, err := reader.GetErrorRates(context.Background(), &params)
defer closer()
require.NoError(t, err)
assert.NotEmpty(t, exp.GetSpans(), "Spans recorded during the test.")
assert.Len(t, exp.GetSpans(), 1, "HTTP request was traced and span reported")
assert.NotNil(t, m)
defer closer()
}

func TestGetRoundTripperTLSConfig(t *testing.T) {
Expand Down Expand Up @@ -622,9 +623,9 @@ func TestInvalidCertFile(t *testing.T) {
CAPath: "foo",
},
}, logger, tracer)
defer closer()
require.Error(t, err)
assert.Nil(t, reader)
defer closer()
}

func startMockPrometheusServer(t *testing.T, wantPromQlQuery string, wantWarnings []string) *httptest.Server {
Expand Down

0 comments on commit a8b1c3e

Please sign in to comment.