Locust MCP Server

🚀 ⚡️ Locust-MCP-Server

このサーバーは、Locustの負荷テスト機能をAI駆動の開発環境とシームレスに統合する、モデルコンテキストプロトコル（MCP）サーバーの実装です。

🚀 クイックスタート

1. Locustテストスクリプト（例：hello.py）を作成します。

from locust import HttpUser, task, between

class QuickstartUser(HttpUser):
    wait_time = between(1, 5)

    @task
    def hello_world(self):
        self.client.get("/hello")
        self.client.get("/world")

    @task(3)
    def view_items(self):
        for item_id in range(10):
            self.client.get(f"/item?id={item_id}", name="/item")
            time.sleep(1)

    def on_start(self):
        self.client.post("/login", json={"username":"foo", "password":"bar"})

2. お好きなMCPクライアント（Claude Desktop、Cursor、Windsurfなど）でMCPサーバーを以下のように設定します。

{
  "mcpServers": {
    "locust": {
      "command": "/Users/naveenkumar/.local/bin/uv",
      "args": [
        "--directory",
        "/Users/naveenkumar/Gits/locust-mcp-server",
        "run",
        "locust_server.py"
      ]
    }
  }
}

3. ここで、LLMを使ってテストを実行します。例えば、hello.pyを分析するためにLocustテストを実行するには、以下のようにします。Locust MCPサーバーは以下のツールを使ってテストを開始します。

• run_locust：設定オプション付きのテストランナー（ヘッドレスモード、ホスト、実行時間、ユーザー数、スポーンレート）

✨ 主な機能

• モデルコンテキストプロトコルフレームワークへの簡単な統合
• ヘッドレスモードとUIモードのサポート
• テストパラメータ（ユーザー数、スポーンレート、実行時間）の設定可能
• Locust負荷テストを実行するための使いやすいAPI
• テスト実行のリアルタイム出力
• HTTP/HTTPSプロトコルのネイティブサポート
• カスタムタスクシナリオのサポート

📦 インストール

1. リポジトリをクローンします。

git clone https://github.com/qainsights/locust-mcp-server.git

2. 必要な依存関係をインストールします。

uv pip install -r requirements.txt

3. 環境変数を設定します（オプション）。プロジェクトのルートディレクトリに.envファイルを作成します。

LOCUST_HOST=http://localhost:8089  # テストのデフォルトホスト
LOCUST_USERS=3                     # デフォルトのユーザー数
LOCUST_SPAWN_RATE=1               # ユーザーのデフォルトスポーンレート
LOCUST_RUN_TIME=10s               # デフォルトのテスト実行時間

📚 ドキュメント

📝 API 参考

Locustテストの実行

run_locust(
    test_file: str,
    headless: bool = True,
    host: str = "http://localhost:8089",
    runtime: str = "10s",
    users: int = 3,
    spawn_rate: int = 1
)

パラメータ：

• test_file：Locustテストスクリプトのパス
• headless：ヘッドレスモード（True）またはUI付き（False）
• host：負荷テストの対象ホスト
• runtime：テストの実行時間（例：“30s”、“1m”、“5m”）
• users：シミュレートする同時ユーザー数
• spawn_rate：ユーザーのスポーンレート

✨ 使用例

基本的な使用法

# Locustテストスクリプトの作成
from locust import HttpUser, task, between

class QuickstartUser(HttpUser):
    wait_time = between(1, 5)

    @task
    def hello_world(self):
        self.client.get("/hello")
        self.client.get("/world")

    @task(3)
    def view_items(self):
        for item_id in range(10):
            self.client.get(f"/item?id={item_id}", name="/item")
            time.sleep(1)

    def on_start(self):
        self.client.post("/login", json={"username":"foo", "password":"bar"})

高度な使用法

# LLMを使ったテストの実行
# ここでは、LLMが特定のテストを実行するための指示を出すシナリオを想定
# 例えば、特定のテストスクリプトを指定して実行する

✨ 使用シナリオ

• LLM駆動の結果分析
• LLMを活用した効果的なデバッグ

🔧 技術詳細

始める前に、以下のものをインストールしてください。

• Python 3.13以上
• uvパッケージマネージャー（インストールガイド）

🤝 貢献

貢献を歓迎します！自由にPull Requestを送信してください。

📄 ライセンス

このプロジェクトはMITライセンスの下で提供されています。詳細については、LICENSEファイルを参照してください。