亚马逊Nova Act自动化测试第三部分团队管理与指标

import os
import requests
from typing import Dict, Any, List

class TestSuiteClassifier:
    def __init__(self, api_key: str, base_url: str = "https://api.n1n.ai/v1"):
        self.api_key = api_key
        self.base_url = base_url

    def classify_test_case(self, test_script: str) -> Dict[str, Any]:
        headers = {
            "Authorization": f"Bearer {self.api_key}",
            "Content-Type": "application/json"
        }
        payload = {
            "model": "amazon-nova-act-pro",
            "messages": [
                {
                    "role": "system",
                    "content": "你是一位企业级QA架构师。分析提供的测试脚本，并按风险级别（高、中、低）、目标域和执行优先级对其进行分类。"
                },
                {
                    "role": "user",
                    "content": f"分析此测试脚本：\n<script>\n{test_script}\n</script>"
                }
            ],
            "temperature": 0.1
        }
        
        response = requests.post(f"{self.base_url}/chat/completions", json=payload, headers=headers)
        response.raise_for_status()
        return response.json()

# 示例用法
classifier = TestSuiteClassifier(api_key=os.getenv("N1N_API_KEY"))
sample_script = """
async function testCheckoutFlow() {
    await page.goto('https://checkout.example.com');
    await page.fill('#credit-card', '4242424242424242');
    await page.click('#submit-order');
    await expect(page.locator('.success-banner')).toBeVisible();
}
"""
classification_result = classifier.classify_test_case(sample_script)
print("分类结果：", classification_result)