From fa240aac739b4d4c40cade005f4cfbafe0103db7 Mon Sep 17 00:00:00 2001
From: anirudhwarrier <12178754+anirudhwarrier@users.noreply.github.com>
Date: Wed, 27 Mar 2024 20:35:34 +0400
Subject: [PATCH] address review comments

---
 .../load/automationv2_1/automationv2_1_test.go     | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/integration-tests/load/automationv2_1/automationv2_1_test.go b/integration-tests/load/automationv2_1/automationv2_1_test.go
index 2c1205bf420..ccac4c0af12 100644
--- a/integration-tests/load/automationv2_1/automationv2_1_test.go
+++ b/integration-tests/load/automationv2_1/automationv2_1_test.go
@@ -4,6 +4,7 @@ import (
 	"context"
 	"encoding/hex"
 	"fmt"
+	"io"
 	"math"
 	"math/big"
 	"net/http"
@@ -143,9 +144,20 @@ func setUpDataStreamsWireMock(url string) error {
 		return err
 	}
 	resp, err := http.Post(fmt.Sprintf("%s/__admin/mappings/save", url), "application/json", nil)
-	if err != nil || resp.StatusCode != 200 {
+	if err != nil {
+		return errors.New("error saving wiremock mappings")
+	}
+	defer func(Body io.ReadCloser) {
+		err := Body.Close()
+		if err != nil {
+			panic(err)
+		}
+	}(resp.Body)
+
+	if resp.StatusCode != 200 {
 		return errors.New("error saving wiremock mappings")
 	}
+
 	return nil
 }