e6a02cb90d
* init * enrich doc * init * init * init * doc * chore: remove deprecated email field from pyproject.toml --------- Co-authored-by: zhengweijun <weijun.zheng@aminer.cn>
24 lines
473 B
Python
24 lines
473 B
Python
import jwt
|
|
import pytest
|
|
|
|
from zai.core._jwt_token import generate_token
|
|
|
|
|
|
def test_token() -> None:
|
|
# Generate token
|
|
token = generate_token('12345678.abcdefg')
|
|
assert token is not None
|
|
|
|
# Parse token
|
|
payload = jwt.decode(
|
|
token,
|
|
'abcdefg',
|
|
algorithms='HS256',
|
|
options={'verify_signature': False},
|
|
)
|
|
assert payload is not None
|
|
assert payload.get('api_key') == '12345678'
|
|
|
|
apikey = 'invalid_api_key'
|
|
with pytest.raises(Exception):
|
|
generate_token(apikey)
|