test: check status code before reading the body

Signed-off-by: Paul Holzinger <[email protected]>
containers · Jan 20, 2025 · ebb19b0 · ebb19b0
1 parent afa68c7
commit ebb19b0
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test/e2e/libhvee_test.go b/test/e2e/libhvee_test.go
@@ -31,6 +31,9 @@ func get(endpoint string) ([]byte, error) {
 	if err != nil {
 		return nil, err
 	}
+	if resp.StatusCode != http.StatusOK {
+		Fail(fmt.Sprintf("get %s: status code: %d", endpoint, resp.StatusCode))
+	}
 	defer resp.Body.Close()
 	body, err := io.ReadAll(resp.Body)
 	return body, err
@@ -46,6 +49,9 @@ func pullWithProgress(endpoint string, dst *os.File) error {
 	if err != nil {
 		Fail(err.Error())
 	}
+	if resp.StatusCode != http.StatusOK {
+		Fail(fmt.Sprintf("get %s: status code: %d", endpoint, resp.StatusCode))
+	}
 	defer resp.Body.Close()
 	return doWithProgress("downloading", resp.Body, dst, resp.ContentLength)
 }