Function test_get_api_key

test_performance.py:62–97

测试获取API密钥的性能

(source_name: str, request_id: int)

        return available_sources

def test_get_api_key(source_name: str, request_id: int):
    """测试获取API密钥的性能"""
    start_time = time.time()
    try:
        response = requests.post(
            f"{API_KEY_MANAGER_URL}/get_apikey",
            json={"source_name": source_name},
            timeout=10
        )
        end_time = time.time()
        
        if response.status_code == 200:
            return {
                "request_id": request_id,
                "source_name": source_name,
                "success": True,
                "response_time": end_time - start_time,
                "api_key_preview": response.json().get("api_key", "")[:8] + "..."
            }
        else:
            return {
                "request_id": request_id,
                "source_name": source_name,
                "success": False,
                "response_time": end_time - start_time,
                "error": f"HTTP {response.status_code}: {response.text[:100]}"
            }
    except Exception as e:
        end_time = time.time()
        return {
            "request_id": request_id,
            "source_name": source_name,
            "success": False,
            "response_time": end_time - start_time,
            "error": str(e)
        }

def test_concurrent_performance():
    """测试并发性能"""