"""企业工商信息查询测试"""
import os
from query import BusinessQuery


def test_query():
    """测试企业工商信息查询"""
    api_key = os.getenv("CHINAZ_API_KEY", "")
    
    if not api_key:
        print("错误: 请设置 CHINAZ_API_KEY 环境变量")
        return
    
    query = BusinessQuery(api_key)
    
    # 测试查询
    print("测试查询: 杭州数脉科技有限公司")
    result = query.query_business_info("杭州数脉科技有限公司")
    
    print(f"\nHTTP状态码: {result.get('status_code')}")
    print(f"查询成功: {result.get('success')}")
    
    if result.get('success'):
        print(f"返回数据: {result.get('data')}")
    else:
        print(f"错误信息: {result.get('error')}")


if __name__ == "__main__":
    test_query()
