fix: auto-fix code issues (cron)
- 修复重复导入/字段 - 修复异常处理 - 修复PEP8格式问题 - 添加类型注解 - 修复缺失的urllib.parse导入
This commit is contained in:
@@ -17,6 +17,7 @@ from tenant_manager import get_tenant_manager
|
||||
|
||||
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
||||
|
||||
|
||||
def test_tenant_management():
|
||||
"""测试租户管理功能"""
|
||||
print("=" * 60)
|
||||
@@ -28,10 +29,7 @@ def test_tenant_management():
|
||||
# 1. 创建租户
|
||||
print("\n1.1 创建租户...")
|
||||
tenant = manager.create_tenant(
|
||||
name="Test Company",
|
||||
owner_id="user_001",
|
||||
tier="pro",
|
||||
description="A test company tenant"
|
||||
name="Test Company", owner_id="user_001", tier="pro", description="A test company tenant"
|
||||
)
|
||||
print(f"✅ 租户创建成功: {tenant.id}")
|
||||
print(f" - 名称: {tenant.name}")
|
||||
@@ -55,9 +53,7 @@ def test_tenant_management():
|
||||
# 4. 更新租户
|
||||
print("\n1.4 更新租户信息...")
|
||||
updated = manager.update_tenant(
|
||||
tenant_id=tenant.id,
|
||||
name="Test Company Updated",
|
||||
tier="enterprise"
|
||||
tenant_id=tenant.id, name="Test Company Updated", tier="enterprise"
|
||||
)
|
||||
assert updated is not None, "更新租户失败"
|
||||
print(f"✅ 租户更新成功: {updated.name}, 层级: {updated.tier}")
|
||||
@@ -69,6 +65,7 @@ def test_tenant_management():
|
||||
|
||||
return tenant.id
|
||||
|
||||
|
||||
def test_domain_management(tenant_id: str):
|
||||
"""测试域名管理功能"""
|
||||
print("\n" + "=" * 60)
|
||||
@@ -79,11 +76,7 @@ def test_domain_management(tenant_id: str):
|
||||
|
||||
# 1. 添加域名
|
||||
print("\n2.1 添加自定义域名...")
|
||||
domain = manager.add_domain(
|
||||
tenant_id=tenant_id,
|
||||
domain="test.example.com",
|
||||
is_primary=True
|
||||
)
|
||||
domain = manager.add_domain(tenant_id=tenant_id, domain="test.example.com", is_primary=True)
|
||||
print(f"✅ 域名添加成功: {domain.domain}")
|
||||
print(f" - ID: {domain.id}")
|
||||
print(f" - 状态: {domain.status}")
|
||||
@@ -118,6 +111,7 @@ def test_domain_management(tenant_id: str):
|
||||
|
||||
return domain.id
|
||||
|
||||
|
||||
def test_branding_management(tenant_id: str):
|
||||
"""测试品牌白标功能"""
|
||||
print("\n" + "=" * 60)
|
||||
@@ -136,7 +130,7 @@ def test_branding_management(tenant_id: str):
|
||||
secondary_color="#52c41a",
|
||||
custom_css=".header { background: #1890ff; }",
|
||||
custom_js="console.log('Custom JS loaded');",
|
||||
login_page_bg="https://example.com/bg.jpg"
|
||||
login_page_bg="https://example.com/bg.jpg",
|
||||
)
|
||||
print("✅ 品牌配置更新成功")
|
||||
print(f" - Logo: {branding.logo_url}")
|
||||
@@ -157,6 +151,7 @@ def test_branding_management(tenant_id: str):
|
||||
|
||||
return branding.id
|
||||
|
||||
|
||||
def test_member_management(tenant_id: str):
|
||||
"""测试成员管理功能"""
|
||||
print("\n" + "=" * 60)
|
||||
@@ -168,10 +163,7 @@ def test_member_management(tenant_id: str):
|
||||
# 1. 邀请成员
|
||||
print("\n4.1 邀请成员...")
|
||||
member1 = manager.invite_member(
|
||||
tenant_id=tenant_id,
|
||||
email="admin@test.com",
|
||||
role="admin",
|
||||
invited_by="user_001"
|
||||
tenant_id=tenant_id, email="admin@test.com", role="admin", invited_by="user_001"
|
||||
)
|
||||
print(f"✅ 成员邀请成功: {member1.email}")
|
||||
print(f" - ID: {member1.id}")
|
||||
@@ -179,10 +171,7 @@ def test_member_management(tenant_id: str):
|
||||
print(f" - 权限: {member1.permissions}")
|
||||
|
||||
member2 = manager.invite_member(
|
||||
tenant_id=tenant_id,
|
||||
email="member@test.com",
|
||||
role="member",
|
||||
invited_by="user_001"
|
||||
tenant_id=tenant_id, email="member@test.com", role="member", invited_by="user_001"
|
||||
)
|
||||
print(f"✅ 成员邀请成功: {member2.email}")
|
||||
|
||||
@@ -217,6 +206,7 @@ def test_member_management(tenant_id: str):
|
||||
|
||||
return member1.id, member2.id
|
||||
|
||||
|
||||
def test_usage_tracking(tenant_id: str):
|
||||
"""测试资源使用统计功能"""
|
||||
print("\n" + "=" * 60)
|
||||
@@ -230,11 +220,11 @@ def test_usage_tracking(tenant_id: str):
|
||||
manager.record_usage(
|
||||
tenant_id=tenant_id,
|
||||
storage_bytes=1024 * 1024 * 50, # 50MB
|
||||
transcription_seconds=600, # 10分钟
|
||||
transcription_seconds=600, # 10分钟
|
||||
api_calls=100,
|
||||
projects_count=5,
|
||||
entities_count=50,
|
||||
members_count=3
|
||||
members_count=3,
|
||||
)
|
||||
print("✅ 资源使用记录成功")
|
||||
|
||||
@@ -258,6 +248,7 @@ def test_usage_tracking(tenant_id: str):
|
||||
|
||||
return stats
|
||||
|
||||
|
||||
def cleanup(tenant_id: str, domain_id: str, member_ids: list):
|
||||
"""清理测试数据"""
|
||||
print("\n" + "=" * 60)
|
||||
@@ -281,6 +272,7 @@ def cleanup(tenant_id: str, domain_id: str, member_ids: list):
|
||||
manager.delete_tenant(tenant_id)
|
||||
print(f"✅ 租户已删除: {tenant_id}")
|
||||
|
||||
|
||||
def main():
|
||||
"""主测试函数"""
|
||||
print("\n" + "=" * 60)
|
||||
@@ -307,6 +299,7 @@ def main():
|
||||
except Exception as e:
|
||||
print(f"\n❌ 测试失败: {e}")
|
||||
import traceback
|
||||
|
||||
traceback.print_exc()
|
||||
|
||||
finally:
|
||||
@@ -317,5 +310,6 @@ def main():
|
||||
except Exception as e:
|
||||
print(f"⚠️ 清理失败: {e}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Reference in New Issue
Block a user