76 lines
2.0 KiB
Python
76 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
测试 OpenClaw Auth Runtime
|
||
"""
|
||
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
# 添加当前目录到路径
|
||
sys.path.insert(0, str(Path(__file__).parent))
|
||
|
||
from auth_runtime import (
|
||
create_env_config,
|
||
get_access_token,
|
||
request_api_with_auto_refresh,
|
||
load_client_key_from_env,
|
||
)
|
||
|
||
|
||
def main():
|
||
print("=" * 60)
|
||
print("OpenClaw Auth Runtime 测试")
|
||
print("=" * 60)
|
||
|
||
# 方式 1: 从 .env 加载 CLIENT_KEY
|
||
try:
|
||
client_key = load_client_key_from_env()
|
||
print(f"✓ 从 .env 文件加载 CLIENT_KEY: {client_key[:10]}...")
|
||
except ValueError as e:
|
||
print(f"❌ {e}")
|
||
print("\n请在 .env 文件中设置 CLIENT_KEY")
|
||
print("或设置环境变量：export CLIENT_KEY=your-key")
|
||
return 1
|
||
|
||
# 创建配置
|
||
config = create_env_config()
|
||
print(f"✓ 配置创建成功")
|
||
print(f" AUTH_BASE: {config.auth_base}")
|
||
print(f" AUTH_CACHE_DIR: {config.auth_cache_dir}")
|
||
|
||
# 获取访问令牌
|
||
print("\n【测试 1】获取访问令牌...")
|
||
try:
|
||
token = get_access_token(dry_run=False, config=config)
|
||
print(f"✓ 获取成功：{token[:20]}...")
|
||
except Exception as e:
|
||
print(f"❌ 失败：{e}")
|
||
return 1
|
||
|
||
# 测试 API 请求
|
||
print("\n【测试 2】测试 API 请求...")
|
||
try:
|
||
# 示例：请求认证端点
|
||
response = request_api_with_auto_refresh(
|
||
method="POST",
|
||
url=f"{config.auth_base}/auth/skill-credit/session",
|
||
dry_run=False,
|
||
config=config,
|
||
body={"clientKey": config.client_key},
|
||
)
|
||
print(f"✓ 响应状态：{response.status}")
|
||
print(f" 响应体：{response.body[:100]}...")
|
||
except Exception as e:
|
||
print(f"❌ 失败：{e}")
|
||
|
||
print("\n" + "=" * 60)
|
||
print("✅ 测试完成")
|
||
print("=" * 60)
|
||
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|