Claude Code Hooks怎么用自动门禁让AI边写测试边自动修？（下）

# .claude/hooks/check_assertions.sh
#!/bin/bash
FILE=$1
# 只检查测试文件if [[ ! "$FILE" == tests/*.py ]]; then  exit 0fi
# 跳过不包含测试类或函数的文件if ! grep -qE "def test_" "$FILE"; then  exit 0fi
# 统计文件中的测试函数数量TEST_COUNT=$(grep -cE "def test_" "$FILE")
# 统计断言模式的数量ASSERTION_COUNT=$(grep -cE "assert |expect\(|should\.|to_be_|to_have_|to_contain_" "$FILE")
if [ "$ASSERTION_COUNT" -eq 0 ]; then  echo ""  echo "QA GATE FAILED: No assertions found in $FILE"  echo "Every test must verify at least one expected outcome."  echo "Examples:"  echo "  assert page.url.endswith('/dashboard')"  echo "  expect(page.locator(\"[data-testid='success-msg']\")).to_be_visible()"  exit 1fi
# 如果测试函数数量远多于断言数量，发出警告if [ "$TEST_COUNT" -gt "$ASSERTION_COUNT" ]; then  echo ""  echo "QA GATE WARNING: $FILE has $TEST_COUNT test functions but only $ASSERTION_COUNT assertions."  echo "Some tests may be missing meaningful verifications."fi
exit0

{  "hooks": {    "PostToolUse": [      {        "matcher": "Edit|Write",        "hooks": [          {            "type": "command",            "command": "bash .claude/hooks/check_assertions.sh $CLAUDE_TOOL_INPUT_FILE_PATH"          }        ]      }    ]  }}

QA GATE FAILED: No assertions found in tests/e2e/test_profile.py
Every test must verify at least one expected outcome.
Examples:  assert page.url.endswith('/dashboard')  expect(page.locator("[data-testid='success-msg']")).to_be_visible()

# .claude/hooks/qa_gate.sh
#!/bin/bash
FILE=$1VIOLATIONS=0
# 只检查Python测试文件和页面对象if [[ ! "$FILE" == tests/*.py ]] && [[ ! "$FILE" == pages/*.py ]]; then  exit 0fi
echo "Running QA gate checks on $FILE..."
# 检查1：硬编码等待if grep -nE "time\.sleep\(|waitForTimeout\(|Thread\.sleep\(" "$FILE"; then  echo ""  echo "FAILED [Hardcoded Wait]: Replace with an explicit wait condition."  echo "  Playwright: page.wait_for_selector(), expect(locator).to_be_visible()"  echo "  Selenium: WebDriverWait(driver, timeout).until(...)"  VIOLATIONS=$((VIOLATIONS + 1))fi
# 检查2：禁止的选择器if grep -nE "By\.XPATH|find_element\(By\.ID|find_element\(By\.CLASS_NAME|css=\'" "$FILE"; then  echo ""  echo "FAILED [Forbidden Selector]: Use data-testid attributes only."  echo "  Correct: page.locator(\"[data-testid='submit-btn']\")"  echo "  Forbidden: By.XPATH, By.ID, By.CLASS_NAME, css selectors"  VIOLATIONS=$((VIOLATIONS + 1))fi
# 检查3：缺少断言（仅测试文件）if [[ "$FILE" == tests/*.py ]]; then  if grep -qE "def test_" "$FILE"; then    ASSERTION_COUNT=$(grep -cE "assert |expect\(|should\.|to_be_|to_have_|to_contain_" "$FILE")    TEST_COUNT=$(grep -cE "def test_" "$FILE")
    if [ "$ASSERTION_COUNT" -eq 0 ]; then      echo ""      echo "FAILED [Missing Assertions]: Every test must verify at least one outcome."      VIOLATIONS=$((VIOLATIONS + 1))    elif [ "$TEST_COUNT" -gt "$ASSERTION_COUNT" ]; then      echo ""      echo "WARNING [Assertion Coverage]: $TEST_COUNT tests but only $ASSERTION_COUNT assertions found."      echo "Some tests may be missing meaningful verifications."    fi  fifi
# 总结if [ $VIOLATIONS -gt 0 ]; then  echo ""  echo "QA gate failed with $VIOLATIONS violation(s) in $FILE"  echo "Claude Code will attempt to fix the issues above before continuing."  exit 1fi
echo "QA gate passed."exit 0

{  "hooks": {    "PostToolUse": [      {        "matcher": "Edit|Write",        "hooks": [          {            "type": "command",            "command": "bash .claude/hooks/qa_gate.sh $CLAUDE_TOOL_INPUT_FILE_PATH"          }        ]      }    ]  }}

Running QA gate checks on tests/e2e/test_checkout.py...
tests/e2e/test_checkout.py:14: time.sleep(2)
FAILED [Hardcoded Wait]: Replace with an explicit wait condition.  Playwright: page.wait_for_selector(), expect(locator).to_be_visible()  Selenium: WebDriverWait(driver, timeout).until(...)
tests/e2e/test_checkout.py:22: self.driver.find_element(By.ID, "confirm-btn").click()
FAILED [Forbidden Selector]: Use data-testid attributes only.  Correct: page.locator("[data-testid='confirm-btn']")  Forbidden: By.XPATH, By.ID, By.CLASS_NAME, css selectors
QA gate failed with 2 violation(s) in tests/e2e/test_checkout.py
Claude Code will attempt to fix the issues above before continuing.