前書き:LLM の登場で自作 OSS の E2E テストを書く機会が増えた

LLM が登場してから、E2E(End to End) テストを書くのが楽になりました。維持は相変わらず大変ですが。

具体例としては、業務では k1LoW/runn を利用して API をシナリオベースで E2E したり、プライベートでは shellspec/shellspec を使って CLI のブラックボックステストしています。LLM に自然言語でテスト観点やユースケースを書けば、テストを実行するための spec ファイルを生成してくれます。いい時代になりました。

大前提として、runn は API サーバー寄りのテスト、ShellSpec はシェルスクリプト寄りのテストに最適化されたテスティングフレームワークと言えるでしょう。どちらも素晴らしいツールであり、CLI のテストにも利用できます。私は、ShellSpec を気に入りすぎて、Software Design 誌で ShellSpec に関する記事を寄稿しました。

しかし、これらのツールでは私のユースケースに合わない部分がありました。私が重要視したのは、以下のポイントです。

  • ターゲットは、サーバーやシェルスクリプトではなく、ターミナルで普段使いするコマンド
  • 対話型 CLI がテストできること
  • PR 作成時に CI しても気にならないほど、実行が高速であること
  • 画像、PDF、金融特有フォーマットなどの入出力を取り扱えること
  • クロスプラットフォームで実行しやすいこと
  • 特定の言語限定のツールでないこと

私は幅広い領域の CLI を作りがちであり、Linux/macOS/Windows 環境で動作することを重要視しています。また、私は CI 待ちが嫌いな性格です。私のユースケースに合うテスティングツールを探すより、自作した方が話が早いと結論づけました。私が OSS 開発を続ける限り、テスティングツールを使い続けるので余裕でペイできると判断しました。

このような流れで開発したのが nao1215/atago です。本記事では、v0.4.0時点の機能を一部紹介します。

demo


atago の特徴

atago は、コマンドを実際に実行し、YAML で定義したテストケースにもとづいて stdout / stderr / exit code / 生成ファイル / snapshot / 対話操作まで検証します。CLI 向けブラックボックス E2E テストランナーです。

意図的に runn と ShellSpec と競合しないようにしました。つまり、シナリオベースの API サーバーをテストしたい場合は runn を採用し、シェルやシェルスクリプトのテストをしたい場合は ShellSpec を採用してください。atago は、ターミナルで普段使いするコマンドをメインターゲットにしています。

以下、atago の特徴(抜粋)です。

  • クロスプラットフォーム対応
  • YAML による宣言的なテストケース記述(複雑度を下げるため、意図的に式を不採用)
  • デフォルト並列実行による高速な E2E
  • 実バイナリをそのまま対象にしたブラックボックステスト
  • atago record:テスト対象コマンドの実行結果からテストケースの雛形生成
  • atago snapshot:Golden Test ファイル(期待値データ)の生成
  • 対話型 CLI および TUI の操作フロー検証
  • 標準出力・終了コード・生成ファイル・ディレクトリ差分・画像・PDFなどをチェック
  • Mock Server / DB / SSH など、CLI が依存する外部環境をテスト実行時に用意
  • CI ログやスナップショットへのクレデンシャル流出を抑えるマスキング機構
  • YAML ファイルを Markdown ドキュメントに変換
  • GitHub Actions でビルド済み atago を取得するnao1215/setup-atago を提供

現在は機能追加だけでなく、atago 自身を atago で E2E し、ドッグフーディングしながら品質を上げるフェーズに入っています。


実行例

atago は、実行時に指定したディレクトリ以下を再帰的に探索し、YAML をテストケース定義ファイルとみなしてテストします。以下が実行例です。ドットがスコスコするのが好きなので、スコスコさせました。

$ atago run ./specs
...............................................

PASSED  47 scenarios: 47 passed, 0 failed, 0 errored, 0 skipped (1.2s)

テスト失敗時は、以下の出力になります。 unified diff 形式で差分を表示するため、出力と期待値のズレが分かりやすくなっています。UX を高めるために、想定原因の Hint も出力しています。

FAILED: demo / greeting matches its golden

Step:
  assert stdout snapshot

Diff (-expected +actual):
  --- snapshot (golden)
  +++ actual
  @@ -1,3 +1,3 @@
   hello
  -WORLD
  +world
   bye

Hint:
  stdout did not match snapshot "snaps/greeting.txt" (update with --update-snapshots if intended)

YAML(テストケース)サンプル

Exampleから、atago の守備範囲が伝わりやすいものを 3 つだけ抜粋します。atago は、基本的に機能を追加するたびに Example コードを用意し、その Example コードが実装と乖離しないように CI で確認しています。

終了コード・標準出力・標準エラー出力の検証

version: "1"          # YAML スキーマのバージョン。現状は "1" 固定

suite:
  name: run and assert  # テストスイート名。レポートやログの見出しに使われる

scenarios:
  - name: one assert can check several targets at once  # シナリオ名(1テストケース)
    steps:
      - run:
          shell: true          # シェル経由で実行(パイプやリダイレクトが使える)
          command: echo hello atago  # 実際に実行するコマンド
      # 1つの assert に複数ターゲットを並べると、全て満たされて初めて成功する
      - assert:
          exit_code: 0         # 終了コードが 0 であること
          stdout:
            contains: hello    # 標準出力に hello を含むこと
          stderr:
            empty: true        # 標準エラー出力が空であること

対話型 CLI の expect / send 検証

version: "1"

suite:
  name: pty

scenarios:
  - name: drive an interactive session with expect and send
    skip:
      os: windows          # pty は POSIX 専用。Windows ではこのシナリオを飛ばす
    steps:
      - pty:                # 疑似端末(PTY)上でコマンドを実行し、対話操作する
          command: cat      # cat は入力行をそのまま返すミニ REPL として利用している
          timeout: 10s      # セッション全体のタイムアウト(無応答でハングせず失敗)
          session:          # send/expect を上から順に実行する対話手順
            - send: "hello interactive world\n"  # 端末へ入力(\n で Enter 相当)
            - expect: "hello interactive world"  # この文字列(正規表現)が出るまで待つ
            - send: ""      # 空送信は EOF(Ctrl-D)。cat を終了させる
      - assert:
          exit_code: 0                        # 正常終了したこと
          stdout:
            contains: hello interactive world  # 端末のエコーを含む出力を検証

Mock Server を用いた API クライアント CLI 検証

version: "1"

suite:
  name: mock http server

scenarios:
  - name: the CLI posts a report and the mock records exactly what was sent
    skip:
      os: windows
    mock_servers:              # ステップ実行前にスタブ HTTP サーバーを起動する
      - name: api              # 参照名。${api.url} / ${api.port} で使える
        routes:                # method + path が一致したリクエストへ返す定義
          - method: POST
            path: /v1/reports
            status: 201        # 返す HTTP ステータスコード
            json: { id: "r-1", ok: true }  # 返すレスポンスボディ(JSON)
    steps:
      - run:
          shell: true
          command: >-          # ${api.url} にモックサーバーのアドレスが入る
            curl -sf -X POST -H 'Authorization: Bearer tok-123'
            -d '{"title":"report"}' ${api.url}/v1/reports
      - assert:
          exit_code: 0
          stdout:
            json: { path: "$.id", equals: "r-1" }  # JSONPath で応答を検証
      # モックが記録した「CLI が実際に送った内容」を検証する
      - assert:
          mock:
            name: api          # 検証対象のモックサーバー名
            path: /v1/reports  # 受け取ったリクエストの path
            method: POST       # 受け取ったリクエストの method
            count: 1           # 一致するリクエストが 1 回だけ来たこと
            header: { name: Authorization, matches: "^Bearer " }  # ヘッダを正規表現で検証
            body:
              json: { path: "$.title", equals: "report" }        # 送信ボディを検証

record 機能

atago record は、CLI を一度だけ実行し、その結果からテストケースの雛形を生成する機能です。多くの人が YAML を手で書くのが苦痛だと思われたので、用意しました。

例えば atago record --out mytool.atago.yaml -- mytool convert input.txt のように実行すると、終了コード、標準出力、標準エラー出力、生成されたファイルを記録し、記録をもとに atago が YAML を出力します。対話型 CLI および TUI 向けには atago record --pty も用意しています。pseudo terminal 上で実際にコマンドを操作するため、対話形式の CLI や TTY 前提で挙動が変わるコマンドでも、最初の YAML を手で全部書かなくて済みます。

なお、atago initで YAML テンプレートファイルを出力する方法もあります。


snapshot 機能(Golden Test)

atago snapshot は、CLI の出力を Golden File(期待値データ)として保存し、以後の実行結果と比較する機能です。containsequals で細かく assert を積み上げる代わりに、「この出力全体が期待値である」と丸ごと固定できます。

特に、help メッセージ、整形済みテキスト、レポート出力、Markdown 生成結果のように、部分一致より全体一致で見たいケースと相性が良いです。差分が出た場合は unified diff 形式で確認できます。また、snapshot は単純な文字列比較ではありません。ANSI カラーコード、一時ディレクトリ、UUID、タイムスタンプ、ポート番号、CRLF などを正規化して比較するため、環境差分で壊れにくいようにしています。

更新時は、以下のコマンドで Golden File を再生成します。

atago snapshot update spec.atago.yaml

atago snapshot は、YAML(テストケース)に書かれているテストをそのまま実行し、snapshot: を使っている assert に到達したら、比較の代わりにその時点の出力を Golden File へ書き込みます。

snapshot を利用した YAML の例を以下に示します。

version: "1"

suite:
  name: snapshot

scenarios:
  - name: stdout matches the committed golden file
    steps:
      - run:
          shell: true
          command: echo stable greeting
      - assert:
          stdout:
            snapshot: snapshots/greeting.txt  # 出力全体をこの Golden File と比較(無ければ生成)

snapshot


サンドボックスによる環境隔離・高速化および Flaky テストの発見

各テストは、サンドボックス(一時ディレクトリ。プロセスレベルの隔離はしていない)の中で実行されます。

テスト対象が生成するファイル、$HOMEや設定ディレクトリへの書き込みは、サンドボックス内で実行されます。テストを実行しても環境が汚れません。sandbox_home を指定すれば、~/.config~/.cache を触るコマンドも、サンドボックス内へ隔離できます。副作用が別テストに漏れないので、並列実行でき、高速化に繋がっています。

Flaky テスト向けの機能もあります。atago run --repeat 100 で同じシナリオを繰り返して不安定さをあぶり出したり、--retry-failed 2 で再試行しつつ「一度落ちて回復したものは Flaky として報告する(隠蔽しない)」ようにしています。


Assert の具体例

Assert の中で、便利に使えるものを紹介します。

jsonは、JSONPath と数値の範囲で検証できます。jq などの外部コマンドを準備する必要がありません。

- assert:
    exit_code: { in: [0, 2] }        # 終了コードが 0 または 2 のいずれか(許容集合)
    stdout:
      json: { path: "$.count", gte: 1 }  # JSONPath $.count の値が 1 以上

changesは、コマンド実行後に作成・変更・削除したファイルをチェックします。不要なファイル改変が行われていないことをテストできます。

- assert:
    exit_code: 0
    changes:              # 直前の run/pty が作業ディレクトリに与えた差分を厳密に検証
      created:            # 新規作成されたファイル(glob 可)。これ以外が作られたら失敗
        - site/index.html
        - site/assets/*.css
      modified:           # 変更されたファイル
        - config.yaml
      deleted:            # 削除されたファイル
        - stale.html

storeで値の保存、signalでシグナルを送り、fileでファイル内容をチェックします。以下の例は、サーバーにシグナルを送り、graceful shutdown に成功したかどうかをチェックしています。

services:            # ステップ実行前に起動し、シナリオ終了時に自動で片付ける常駐プロセス
  - name: server
    command: ./mysrv # サーバー起動
    ready: { file: addr.txt, store: addr, timeout: 5s }   # 待受アドレスを ${addr} へ保存
steps:
  - run: { shell: true, command: "mycli ping ${addr}" }  
  - assert: { stdout: { contains: pong } }
  # atago が起動したサービスへ名前指定でシグナルを送る(--parallel でも競合しない)
  - signal: { service: server, signal: TERM, wait: { timeout: 5s } }  # TERM 送信後、終了を待つ
  - assert:
      file: { path: server.log, contains: "graceful shutdown complete" }  # ログに完了痕跡があるか

GitHub Actions 対応

nao1215/setup-atago を利用すると、事前にビルドされたリリースバイナリをダウンロードしてセットアップします。Linux、macOS、Windows (amd64 / arm64)向けのバイナリを自動で準備します 。バージョンをピン留めする機能もあります。

以下が参考例です。

name: behavior-specs
on: [push, pull_request]
jobs:
  atago:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v7
      - uses: nao1215/setup-atago@v0   # ビルド済み atago を取得・セットアップ
      # --ci で色なし出力、--report gha で GitHub Actions 向けレポートを出力
      - run: atago run --ci --report gha ./specs   # ./specs 以下の YAML を再帰実行

Go 製 CLI ではユニットテストと E2E テストのカバレッジを合算できる

テスト対象のコマンドが Go 製の場合、ユニットテストのカバレッジと、E2E テストで実際に CLI を実行した経路のカバレッジを合算できます。

Go 1.20以降では、go build -cover によりカバレッジ計測用のバイナリを生成できます。 このバイナリを GOCOVERDIR を指定して実行すると、通常の go test だけでなく、CLI として実行された経路のカバレッジも収集できます。atago は Go 製なので、この仕組みを使ってユニットテストと E2E テストのカバレッジを合算しています(参考:Makefileカバレッジ合算スクリプト

この機能は atago が提供しているものではありません。Go の仕組みが素晴らしい、という話です。


atago で E2E テストした OSS

atago の信頼性を高めるために、atago 自身 / 私の OSS(8個) / 有名な OSS(29個) を atago で E2E しました。テスト対象 OSS を網羅的にテストしているわけではなく、基本機能だけピックアップでテストしています。

下表にテスト対象の OSS を示します。Specs はテストケース集へのリンク、Docs はテストケースから変換した Markdown ドキュメントへのリンクです。

私が開発した OSS 向けのテスト:

ツール名機能SpecsDocs
atago本記事のツールspecsdocs
gup$GOBIN 内の Go 製コマンドラインツールの更新・管理specsdocs
sqlyCSV/TSV/LTSV/JSON/Parquet/Excel/ACH/Fedwire に対する SQL 実行specsdocs
truss画像変換(形式変換、リサイズ、再エンコード)specsdocs
iso8583toolISO 8583 決済メッセージのデバッグ・解析specsdocs
joseJOSE による署名・暗号化specsdocs
career単一 YAML から履歴書・経歴書 PDF を生成specsdocs
mimixbox多数の Unix コマンドを 1 つの BusyBox 風バイナリに同梱specsdocs
mobilepkgiOS / Android パッケージのメタデータ・セキュリティの解析specsdocs

有名な OSS 向けのテスト:

ツール名機能LicenseSpecsDocs
gitバージョン管理GPL-2.0specsdocs
jqJSON 処理MITspecsdocs
fzf対話型ファジーファインダMITspecsdocs
redisインメモリデータストアRSALv2/SSPLv1specsdocs
hugo静的サイトジェネレータApache-2.0specsdocs
openssl暗号ツールキットApache-2.0specsdocs
sqlite3埋め込みデータベースPublic Domainspecsdocs
caddyWeb サーバーApache-2.0specsdocs
corednsDNS サーバーApache-2.0specsdocs
giteaGit ホスティングサービスMITspecsdocs
gotify通知サーバーMITspecsdocs
grafana可観測性プラットフォームAGPL-3.0specsdocs
mailpitメールテストサーバーMITspecsdocs
minioS3 互換オブジェクトストレージAGPL-3.0specsdocs
aws-cliAWS 向け CLIApache-2.0specsdocs
python3プログラミング言語処理系PSF-2.0specsdocs
ssh-keygenSSH 鍵生成ツールBSDspecsdocs
ffmpeg動画・音声変換ツールLGPL/GPLspecsdocs
pandocドキュメント変換ツールGPL-2.0+specsdocs
terraformIaC ツールBUSL-1.1specsdocs
ageファイル暗号化ツールBSD-3-Clausespecsdocs
natsメッセージングシステムApache-2.0specsdocs
ntfyPush 通知サービスApache-2.0specsdocs
prometheus監視システムApache-2.0specsdocs
pushgatewayメトリクス受け渡しゲートウェイApache-2.0specsdocs
rcloneファイル同期ツールMITspecsdocs
resticバックアップツールBSD-2-Clausespecsdocs
transfer.shファイル共有サービスMITspecsdocs
webhookWebhook 受信サーバーMITspecsdocs

atago の由来

atago は、愛宕(あたご)から名付けました。愛宕は、防火の神様です。IT 業界では炎上プロジェクトと呼ばれる人災があり、「atago が炎上を食い止める存在まで成長すると嬉しい」という思いを込めました。atago は Go 製ツールなので、末尾に"go"が付いているのもポイント高かったです。

パブリック公開直前まで名前が決まらず、“b3spec(Black Box Behavior Spec)“として開発していました。「長年使うツールなのに、b3spec はなんか味気ないな」と思い、ChatGPT と壁打ちしながら決めました。他の候補は、hegi(新潟のへぎそばに使う箱)、nagisa(落ち着いたテストツール感)、amon(隠されたものという意味。SHADOW HEARTS から借用)でした。


最後に:テスティングツール再び

atago は、約3年前のリベンジでもあります。

私は2023年頃に Go 向けの nao1215/spectestnao1215/hottest を作り、API E2E テスト、テスト結果の Markdown ドキュメント化、テスト実行中のドットスコスコを実現していました。

spectest は steinfletcher/apitest をフォークして改良していたこともあり、実装の困難さを感じていました。他人の書いたコードを修正するのは難しいものです。さらに、「Go でしか使えない」「E2E ライブラリは製品コードに埋め込みづらいし、onsi/ginkgo が似たポジションで知名度があった」「k1LoW/runn の方向性が自分好みで、runn の設計センスに勝てる気がしない(k1LoW 氏が開発するツールはどれも設計センスが良いですよね)」と、複数の課題を感じて開発を中断していました。当時の自分には、テストツールとしての責務の切り分けや、既存ツールとの差別化を十分に整理できていませんでした。

開発停止から2年強ほどの期間が経過し、「runn(+ ShellSpec)と競う必要はない」「自分の課題を解決するツールであればよい」と割り切れるぐらいには大人になりました。atago は完全に自分向けです。正直、CLI を E2E したい開発者は少数派だと考えています。しかし、自分には必要でした。

私が CLI の E2E にこだわる理由は、過去の経験が関係しています。昔、「出力するファイルの形式を変えただろう。外部仕様を勝手に変えるな。ファイルもインターフェースで、他のアプリが使う」と怒られたことがあり、あらゆる出力をテストしたい気持ちが強くなりました。

それに、自分向けにツールを作ると、自分の需要を確実に満たすので「良いツール作ったな!」と自己肯定感を高められてオススメです。atago も、まずは自分の OSS 群を安心して変更し続けるためのツールとして育てていきます。


余談

atago をパブリック公開する前に、知り合い(spectest 作った頃の同僚)が存在を速攻で検知してて、「GitHub を SNS みたいに使っている人だ」と思わずにいられませんでした。