update client_test.go

sashabaranov · Sep 23, 2024 · 9822ff0 · 9822ff0
1 parent 17c79a0
commit 9822ff0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/client.go b/client.go
@@ -288,7 +288,7 @@ func (c *Client) handleErrorResp(resp *http.Response) error {
 	if !strings.HasPrefix(resp.Header.Get("Content-Type"), "application/json") {
 		body, err := io.ReadAll(resp.Body)
 		if err != nil {
-			return err
+			return fmt.Errorf("error, reading response body: %s", err)
 		}
 		return fmt.Errorf("error, status code: %d, status: %s, body: %s", resp.StatusCode, resp.Status, body)
 	}

diff --git a/client_test.go b/client_test.go
@@ -215,6 +215,13 @@ func TestHandleErrorResp(t *testing.T) {
 </body>
 </html>`,
 		},
+		{
+			name:        "errorReader",
+			httpCode:    http.StatusRequestEntityTooLarge,
+			contentType: "text/html",
+			body:        &errorReader{err: errors.New("errorReader")},
+			expected:    "error, reading response body: errorReader",
+		},
 	}
 
 	for _, tc := range testCases {