PART 01 — 性能要件定義書
レスポンスタイム・スループット・同時接続数・容量設計で定義すべきこと

"""
Locustを使った負荷テストシナリオ。性能要件の数値を合否基準として検証する。
pip install locust

実行: locust -f locustfile.py --host=http://localhost:8000
     --users 30 --spawn-rate 3 --run-time 5m --headless
"""
from locust import HttpUser, task, between, events
import json
import time


class AppUser(HttpUser):
    """通常ユーザーの操作シナリオ"""
    wait_time = between(1, 3)  # 操作間隔: 1〜3秒

    def on_start(self):
        """セッション開始: ログイン"""
        resp = self.client.post("/api/v1/auth/login/", json={
            "email": "testuser@example.com",
            "password": "testpass123"
        })
        self.token = resp.json().get("access_token", "")

    @property
    def auth_headers(self):
        return {"Authorization": f"Bearer {self.token}"}

    @task(5)  # 重み: 5（最も頻繁）
    def view_product_list(self):
        """商品一覧表示（参照系 - 最頻アクセス）"""
        with self.client.get(
            "/api/v1/products/",
            headers=self.auth_headers,
            name="/api/v1/products/ [一覧]",
            catch_response=True
        ) as resp:
            if resp.elapsed.total_seconds() > 3.0:
                # 性能要件: 3秒以内
                resp.failure(f"レスポンスタイム超過: {resp.elapsed.total_seconds():.2f}s")
            elif resp.status_code != 200:
                resp.failure(f"HTTP {resp.status_code}")

    @task(2)
    def view_product_detail(self):
        """商品詳細表示"""
        product_id = 1  # テストデータのID
        with self.client.get(
            f"/api/v1/products/{product_id}/",
            headers=self.auth_headers,
            name="/api/v1/products/{id}/ [詳細]",
            catch_response=True
        ) as resp:
            if resp.elapsed.total_seconds() > 3.0:
                resp.failure(f"レスポンスタイム超過: {resp.elapsed.total_seconds():.2f}s")

    @task(1)
    def create_product(self):
        """商品登録（更新系 - 低頻度）"""
        payload = {
            "product_code": f"LOAD-{int(time.time() * 1000) % 100000:05d}",
            "product_name": "負荷テスト商品",
            "category_id": 1,
            "unit_price": 1000,
            "tax_type": 10,
        }
        with self.client.post(
            "/api/v1/products/",
            json=payload,
            headers=self.auth_headers,
            name="/api/v1/products/ [登録]",
            catch_response=True
        ) as resp:
            if resp.elapsed.total_seconds() > 5.0:
                # 性能要件: 5秒以内（更新系）
                resp.failure(f"レスポンスタイム超過: {resp.elapsed.total_seconds():.2f}s")


@events.quitting.add_listener
def on_quitting(environment, **kwargs):
    """テスト終了時に性能要件に対する合否を判定"""
    stats = environment.runner.stats.total
    print(f"\n=== 性能テスト結果 ===")
    print(f"総リクエスト数: {stats.num_requests}")
    print(f"失敗率: {stats.fail_ratio:.1%}")
    print(f"平均レスポンスタイム: {stats.avg_response_time:.0f}ms")
    print(f"95%tile レスポンスタイム: {stats.get_response_time_percentile(0.95):.0f}ms")

    # 合否判定
    if stats.avg_response_time > 1000:  # 平均1秒以内
        print("❌ 性能要件NG: 平均レスポンスタイムが1秒を超えています")
        environment.process_exit_code = 1
    elif stats.get_response_time_percentile(0.95) > 3000:  # 95%tile 3秒以内
        print("❌ 性能要件NG: 95%tileレスポンスタイムが3秒を超えています")
        environment.process_exit_code = 1
    else:
        print("✅ 性能要件OK")