测试SDK

glide-the · glide-the · commit f8c2ed33a841 · 2024-05-17T20:21:33.000+08:00
diff --git a/tests/unit_tests/test_http_client.py b/tests/unit_tests/test_http_client.py
@@ -156,7 +156,7 @@ def test_validate_token_headers(self) -> None:
             FinalRequestOptions.construct(method="get", url="/foo")
         )
         from zhipuai.core._jwt_token import generate_token
-        assert request.headers.get("Authorization") == generate_token(api_key)
+        assert request.headers.get("Authorization") == f"Bearer {generate_token(api_key)}"
 
         with pytest.raises(ZhipuAIError):
             client2 = ZhipuAI(base_url=base_url, api_key=None)

Original file line number	Diff line number	Diff line change
`@@ -156,7 +156,7 @@ def test_validate_token_headers(self) -> None:`
`156`	`156`	`FinalRequestOptions.construct(method="get", url="/foo")`
`157`	`157`	`)`
`158`	`158`	`from zhipuai.core._jwt_token import generate_token`
`159`		`- assert request.headers.get("Authorization") == generate_token(api_key)`
	`159`	`+ assert request.headers.get("Authorization") == f"Bearer {generate_token(api_key)}"`
`160`	`160`
`161`	`161`	`with pytest.raises(ZhipuAIError):`
`162`	`162`	`client2 = ZhipuAI(base_url=base_url, api_key=None)`