Spaces:
Paused
Paused
File size: 2,888 Bytes
99a7ebb 2a025c0 99a7ebb 2a025c0 99a7ebb 2a025c0 99a7ebb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | import base64
import json
import unittest
import zipfile
from io import BytesIO
def jwt_with_payload(payload: dict) -> str:
def encode(value: dict) -> str:
raw = json.dumps(value, separators=(",", ":")).encode("utf-8")
return base64.urlsafe_b64encode(raw).decode("ascii").rstrip("=")
return f"{encode({'alg': 'none'})}.{encode(payload)}.signature"
class CpaExportServiceTests(unittest.TestCase):
def test_build_cpa_payload_matches_single_file_shape(self) -> None:
from services.cpa_export_service import build_cpa_payload
token = jwt_with_payload({"exp": 1779622898, "iat": 1778758898})
payload = build_cpa_payload(
{
"access_token": token,
"email": "deandrea.northey@outlook.com",
"user_id": "user-example",
"status": "禁用",
"refresh_token": "rt-example",
"id_token": "id-example",
"password": "P@ssw0rd!",
}
)
self.assertEqual(payload["access_token"], token)
self.assertEqual(payload["refresh_token"], "rt-example")
self.assertEqual(payload["id_token"], "id-example")
self.assertEqual(payload["email"], "deandrea.northey@outlook.com")
self.assertEqual(payload["account_id"], "user-example")
self.assertEqual(payload["password"], "P@ssw0rd!")
self.assertTrue(payload["disabled"])
self.assertEqual(payload["type"], "codex")
self.assertTrue(payload["expired"].startswith("2026-05-24T"))
self.assertTrue(payload["last_refresh"].startswith("2026-05-14T"))
def test_build_cpa_payload_password_defaults_empty(self) -> None:
from services.cpa_export_service import build_cpa_payload
payload = build_cpa_payload({"access_token": "token-no-pwd", "email": "x@example.com"})
self.assertEqual(payload["password"], "")
def test_build_cpa_zip_writes_email_named_json_files(self) -> None:
from services.cpa_export_service import build_cpa_zip
payload = build_cpa_zip(
[
{"access_token": "token-one", "email": "one@example.com", "status": "正常"},
{"access_token": "token-two", "email": "two@example.com", "status": "异常"},
]
)
with zipfile.ZipFile(BytesIO(payload), "r") as archive:
names = sorted(archive.namelist())
self.assertEqual(names, ["one@example.com.json", "two@example.com.json"])
first = json.loads(archive.read("one@example.com.json").decode("utf-8"))
second = json.loads(archive.read("two@example.com.json").decode("utf-8"))
self.assertEqual(first["access_token"], "token-one")
self.assertFalse(first["disabled"])
self.assertTrue(second["disabled"])
if __name__ == "__main__":
unittest.main()
|