mirror of
https://github.com/cnlimiter/codex-register.git
synced 2026-05-06 20:02:51 +08:00
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
from types import SimpleNamespace
|
|
|
|
from src.core.upload import newapi_upload
|
|
|
|
|
|
class FakeResponse:
|
|
def __init__(self, status_code=200, payload=None, text=""):
|
|
self.status_code = status_code
|
|
self._payload = payload
|
|
self.text = text
|
|
|
|
def json(self):
|
|
if self._payload is None:
|
|
raise ValueError("no json payload")
|
|
return self._payload
|
|
|
|
|
|
def test_build_headers_rejects_non_ascii_api_key():
|
|
try:
|
|
newapi_upload._build_headers("系统访问令牌 (System Access Token)")
|
|
except ValueError as exc:
|
|
assert str(exc) == "Authorization Token 包含非 ASCII 字符，请确认填写的是实际令牌而不是中文说明"
|
|
else:
|
|
raise AssertionError("expected ValueError")
|
|
|
|
|
|
def test_upload_to_newapi_uses_ascii_authorization_header(monkeypatch):
|
|
calls = []
|
|
|
|
def fake_post(url, **kwargs):
|
|
calls.append({"url": url, "kwargs": kwargs})
|
|
return FakeResponse(status_code=201)
|
|
|
|
monkeypatch.setattr(newapi_upload.cffi_requests, "post", fake_post)
|
|
|
|
success, message = newapi_upload.upload_to_newapi(
|
|
account=SimpleNamespace(email="tester@example.com", access_token="access-token"),
|
|
api_url="https://newapi.example.com/",
|
|
api_key="token-123",
|
|
)
|
|
|
|
assert success is True
|
|
assert message == "上传成功"
|
|
assert calls[0]["url"] == "https://newapi.example.com/api/channel/"
|
|
assert calls[0]["kwargs"]["headers"]["Authorization"] == "Bearer token-123"
|
|
assert calls[0]["kwargs"]["headers"]["Content-Type"] == "application/json; charset=utf-8"
|
|
assert calls[0]["kwargs"]["data"].startswith(b"{")
|
|
|
|
|
|
def test_upload_to_newapi_returns_clear_error_for_non_ascii_api_key():
|
|
success, message = newapi_upload.upload_to_newapi(
|
|
account=SimpleNamespace(email="tester@example.com", access_token="access-token"),
|
|
api_url="https://newapi.example.com/",
|
|
api_key="系统访问令牌 (System Access Token)",
|
|
)
|
|
|
|
assert success is False
|
|
assert message == "上传异常: Authorization Token 包含非 ASCII 字符，请确认填写的是实际令牌而不是中文说明"
|