Clean up resources in test

hashicorp · Sep 14, 2023 · a1aa591 · a1aa591
1 parent 658c27a
commit a1aa591
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agent/agent_test.go b/agent/agent_test.go
@@ -354,10 +354,10 @@ func TestAgent_HTTPMaxHeaderBytes(t *testing.T) {
 			require.NoError(t, err)
 
 			a, err := New(bd)
+			require.NoError(t, err)
 			mockDelegate := delegateMock{}
 			mockDelegate.On("LicenseCheck").Return()
 			a.delegate = &mockDelegate
-			require.NoError(t, err)
 
 			a.startLicenseManager(testutil.TestContext(t))
 
@@ -392,6 +392,8 @@ func TestAgent_HTTPMaxHeaderBytes(t *testing.T) {
 				resp, err := client.Do(req.WithContext(ctx))
 				require.NoError(t, err)
 				require.Equal(t, tt.expectedHTTPResponse, resp.StatusCode, "expected a '%d' http response, got '%d'", tt.expectedHTTPResponse, resp.StatusCode)
+				resp.Body.Close()
+				s.Shutdown(ctx)
 			}
 		})
 	}