測試基礎：用 pytest 讓你敢放心改程式碼

沒有測試，每次改功能都可能不小心壞掉另一個地方，而你不會知道，直到使用者回報。這篇帶你用 pytest 寫出第一批測試，學會用 Mock 隔離外部 API，再用 FastAPI 的 TestClient 直接測試端點行為，讓 AI 幫你重構程式碼的時候，你也能放心。

為什麼要寫測試

一個簡單的思考框架：

• 不想被叫起床修 bug，就要有測試
• 想讓 AI 放心地幫你重構，就要有測試
• 想讓客戶相信你的交付品質，就要有測試
• 想三個月後還能改自己的程式碼，就要有測試

pytest 快速上手

# tests/test_weather.py
import pytest
from app.weather import parse_weather_response

def test_parse_temperature_celsius():
    """溫度應該正確解析為攝氏"""
    response = {"main": {"temp": 301.15}, "weather": [{"description": "clear"}]}
    result = parse_weather_response(response)
    assert result["temperature"] == 28

def test_parse_missing_main_raises():
    """API 回傳缺少 main 欄位時應拋出 ValueError"""
    with pytest.raises(ValueError):
        parse_weather_response({})  # 缺少 main 欄位

pytest                        # 跑所有測試
pytest tests/test_weather.py  # 跑特定檔案
pytest -v                     # 顯示每個測試的名稱
pytest --cov=app              # 顯示覆蓋率

Mock：隔離外部依賴

測試不應該真的呼叫 OpenWeatherMap API（太慢、有費用、網路可能不通），用 Mock 替換外部呼叫。

from unittest.mock import patch, MagicMock

def test_get_weather_success(mocker):
    # 替換 httpx.get，讓它回傳假的 API 回應
    mock_response = MagicMock()
    mock_response.json.return_value = {
        "main": {"temp": 301.15},
        "weather": [{"description": "clear sky"}]
    }
    mocker.patch("httpx.get", return_value=mock_response)

    from app.weather import get_weather
    result = get_weather("Taipei")
    assert result["temperature"] == 28
    assert "description" in result

FastAPI 測試客戶端

from fastapi.testclient import TestClient
from app.main import app

client = TestClient(app)

def test_weather_endpoint_returns_200(mocker):
    # Mock 外部 API
    mocker.patch("app.weather.fetch_weather", return_value={
        "temperature": 28, "description": "晴天"
    })
    response = client.get("/weather/Taipei",
        headers={"Authorization": "Bearer test-token"})
    assert response.status_code == 200
    assert response.json()["temperature"] == 28

def test_weather_endpoint_requires_auth():
    # 沒有 token 應該回傳 401
    response = client.get("/weather/Taipei")
    assert response.status_code == 401

測試策略：寫哪些測試