fix linting

coinbase · Nov 27, 2024 · 637e34a · 637e34a
1 parent 6b23db7
commit 637e34a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 2 deletions.
diff --git a/headerforwarder/forwarder_test.go b/headerforwarder/forwarder_test.go
@@ -64,7 +64,14 @@ type mockInvoker struct {
 	err           error
 }
 
-func (m *mockInvoker) Invoke(ctx context.Context, method string, req, reply interface{}, cc *grpc.ClientConn, opts ...grpc.CallOption) error {
+func (m *mockInvoker) Invoke(
+	ctx context.Context,
+	method string,
+	req,
+	reply interface{},
+	cc *grpc.ClientConn,
+	opts ...grpc.CallOption,
+) error {
 	m.LastRequestMD, _ = metadata.FromOutgoingContext(ctx)
 
 	// add response metadata to pointer from call option
@@ -212,9 +219,11 @@ func TestRoundTrip(t *testing.T) {
 
 			testReq := newRequest(ctx, nil)
 
-			_, err := hf.RoundTrip(testReq)
+			response, err := hf.RoundTrip(testReq)
 			assert.Equal(t, test.ExpectedError, err)
 
+			defer response.Body.Close()
+
 			outgoingReq := mockTransport.(*fakeTransport).LastRequest
 			capturedHeaders, _ := hf.GetResponseHeaders(requestID)
 

diff --git a/headerforwarder/response_writer_test.go b/headerforwarder/response_writer_test.go
@@ -43,6 +43,7 @@ func TestWriteHeader(t *testing.T) {
 	hfrw.WriteHeader(200)
 
 	result := writer.Result()
+	defer result.Body.Close()
 	assert.Equal(t, 200, result.StatusCode)
 }
 
@@ -53,6 +54,7 @@ func TestWriteHeaderAddsHeaders(t *testing.T) {
 	hfrw.WriteHeader(200)
 
 	result := writer.Result()
+	defer result.Body.Close()
 	assert.Equal(t, 200, result.StatusCode)
 	assert.Equal(t, mockHeader.Values("test-header"), result.Header.Values("test-header"))
 }
@@ -66,6 +68,7 @@ func TestWriteAddsHeaders(t *testing.T) {
 	assert.Nil(t, err)
 
 	result := writer.Result()
+	defer result.Body.Close()
 	assert.Equal(t, 200, result.StatusCode)
 	assert.Equal(t, mockHeader.Values("test-header"), result.Header.Values("test-header"))
 }