Fix OpenAI or Azure returns an undefined error message return (#283)

* Fix OpenAI or Azure returns an undefined error message return * Fix: OpenAI or Azure returns an undefined error message return #280 * Fix: OpenAI or Azure returns an undefined error message return #280 * Fix: OpenAI or Azure returns an undefined error message return #280 Handle long line * Fix: OpenAI or Azure returns an undefined error message return #280 Add unit test ErrorResponse nil pointer check * Fix: OpenAI or Azure returns an undefined error message return #280 Add unit test ErrorResponse nil pointer check --------- Co-authored-by: zhangxm <xuanming_zhang@qingsongchou.com>
2023-05-01 05:21:15 +08:00
parent cacd976a0c
commit 67aef196d2
2 changed files with 83 additions and 0 deletions
--- a/client_test.go
+++ b/client_test.go
@@ -2,7 +2,9 @@ package openai //nolint:testpackage // testing private field

 import (
 	"bytes"
+	"fmt"
 	"io"
+	"net/http"
 	"testing"
 )

@@ -57,3 +59,81 @@ func TestDecodeResponse(t *testing.T) {
 		})
 	}
 }
+
+func TestHandleErrorResp(t *testing.T) {
+	// var errRes *ErrorResponse
+	var errRes ErrorResponse
+	var reqErr RequestError
+	t.Log(errRes, errRes.Error)
+	if errRes.Error != nil {
+		reqErr.Err = errRes.Error
+	}
+	t.Log(fmt.Errorf("error, %w", &reqErr))
+	t.Log(errRes.Error, "nil pointer check Pass")
+
+	const mockToken = "mock token"
+	client := NewClient(mockToken)
+
+	testCases := []struct {
+		name     string
+		httpCode int
+		body     io.Reader
+		expected string
+	}{
+		{
+			name:     "401 Invalid Authentication",
+			httpCode: http.StatusUnauthorized,
+			body: bytes.NewReader([]byte(
+				`{
+					"error":{
+						"message":"You didn't provide an API key. ....",
+						"type":"invalid_request_error",
+						"param":null,
+						"code":null
+					}
+				}`,
+			)),
+			expected: "error, status code: 401, message: You didn't provide an API key. ....",
+		},
+		{
+			name:     "401 Azure Access Denied",
+			httpCode: http.StatusUnauthorized,
+			body: bytes.NewReader([]byte(
+				`{
+					"error":{
+						"code":"AccessDenied",
+						"message":"Access denied due to Virtual Network/Firewall rules."
+					}
+				}`,
+			)),
+			expected: "error, Access denied due to Virtual Network/Firewall rules.",
+		},
+		{
+			name:     "503 Model Overloaded",
+			httpCode: http.StatusServiceUnavailable,
+			body: bytes.NewReader([]byte(`
+				{
+					"error":{
+						"message":"That model...",
+						"type":"server_error",
+						"param":null,
+						"code":null
+					}
+				}`)),
+			expected: "error, status code: 503, message: That model...",
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			testCase := &http.Response{}
+			testCase.StatusCode = tc.httpCode
+			testCase.Body = io.NopCloser(tc.body)
+			err := client.handleErrorResp(testCase)
+			if err.Error() != tc.expected {
+				t.Errorf("Unexpected error: %v , expected: %s", err, tc.expected)
+				t.Fail()
+			}
+		})
+	}
+}