用 Playwright + LLM 搭建你的第一个测试 Harness：让 AI 看懂代码，自动写白盒测试

// 原逻辑
public boolean needCaptcha(String phone) {
    return true;
}

// MR 新增逻辑
public boolean needCaptcha(String phone) {
    if (phone == null || phone.isEmpty()) {
        return true;
    }
    MemberDto member = memberService.getMember(phone);
    return member == null; // 会员免验证码
}

【方法名】UserService.needCaptcha
【输入】String phone
【输出】boolean
【业务规则】
- 若手机号为空，返回 true（需验证码）
- 若手机号对应已注册会员，返回 false（免验证码）
- 否则返回 true

【依赖服务】
- memberService.getMember(phone): 返回 MemberDto 或 null

# 1. 安装 Playwright
pip install playwright
playwright install chromium

# 2. 安装 Ollama（运行本地大模型）
curl -fsSL https://ollama.com/install.sh | sh

# 3. 拉取 Qwen 模型（中文友好）
ollama run qwen:latest

import requests
import subprocess
import tempfile
import os

# 1. 让 LLM 读代码，生成测试
def generate_test(code_context: str):
    prompt = f"""
    你是一名资深 Java QA 工程师，请为以下方法生成 JUnit 5 单元测试：

    {code_context}

    要求：
    - 覆盖所有分支：空手机号、会员、非会员
    - 使用 Mockito mock memberService
    - 输出完整可运行的 Java 代码，不要解释
    """

    resp = requests.post("http://localhost:11434/api/generate", json={
        "model": "qwen",
        "prompt": prompt,
        "stream": False
    })
    return resp.json()["response"]

# 2. 用 Playwright 执行测试（模拟 CI 环境）
def run_test(java_code: str):
    # 创建临时测试文件
    with tempfile.NamedTemporaryFile(mode='w', suffix='.java', delete=False) as f:
        f.write(java_code)
        test_file = f.name

    try:
        # 编译并运行（简化示例，实际可用 Maven/Gradle）
        compile_result = subprocess.run(
            ["javac", test_file],
            capture_output=True, text=True
        )
        if compile_result.returncode != 0:
            return {"status": "compile_error", "log": compile_result.stderr}

        # 运行测试（此处省略，实际可调用 mvn test）
        return {"status": "success", "message": "Test generated and compiled."}
    finally:
        os.unlink(test_file)

# 3. 主流程
if __name__ == "__main__":
    # 读取代码上下文
    with open("code_context.txt", "r") as f:
        context = f.read()

    # 生成测试
    test_code = generate_test(context)
    print("✅ AI 生成的测试用例：\n")
    print(test_code)

    # 验证（可选）
    result = run_test(test_code)
    print("\n🔍 验证结果:", result["status"])

python whitebox_harness.py

@Test
void needCaptcha_nullPhone_returnsTrue() {
    assertTrue(userService.needCaptcha(null));
}

@Test
void needCaptcha_emptyPhone_returnsTrue() {
    assertTrue(userService.needCaptcha(""));
}

@Test
void needCaptcha_registeredMember_returnsFalse() {
    when(memberService.getMember("13800138000")).thenReturn(new MemberDto());
    assertFalse(userService.needCaptcha("13800138000"));
}

@Test
void needCaptcha_newUser_returnsTrue() {
    when(memberService.getMember("13900139000")).thenReturn(null);
    assertTrue(userService.needCaptcha("13900139000"));
}