- 真野隼記
- 約 2,000 文字
- 800 View
はじめに
TIG真野です。
バックエンドのアプリケーションの上で、メール送信するコードがある場合の単体テストをどう実現するか悩みました。
メールには、タイトル・本文・From・TO・CC・BCCなど複数の設定値がありますし、SMTPサーバの接続情報もあります。これらを表現する構造体のモデルだけに絞った検証に留めることは、気が進みませんでした。時代はインフラレベルでダミーサーバを動かしモックする方向で動いています。SMTPでメール送信し、その送信結果をテストコード上で取得&検証する一連の流れを行って動作を確かめたいと思いました。
方法として、澁川さんのMailSlurperを使って6桁のコードの送信コードのテストをするで紹介されたMailSlurperを使うか迷いましたが、以下の点で牛刀だなと感じました。
- メール送信するのはごく一部の機能(私の場合は1機能。今後増える見込みは現時点で見えなかった)
- MailSlurperの管理画面アクセス機能などは不要
- Testcontainersで呼び出すとはいえ、テストで依存するコンテナが増えることに抵抗感
httptest.NewServer() は既存のテストコードでの利用頻度が高かったため、同じメンタルモデルで対応できると嬉しいということで、Goで実装されたSMTPサーバの実装を利用する方向にしました。
go-smtp-mock
※(ロゴがPlaywrightに見えますね)
GoでSMTPのモックサーバの中でおそらく最も有名なmocktools/go-smtp-mockを利用して、SMTPサーバのモックとします。
テストコードのイメージとしては次のようになります。
package example |
サーバの起動は以下のコードの部分です。smtpmock.ConfigurationAttr は未指定ですが、configuringを見れば様々な設定が可能です。利用するポート番号が未指定であれば、動的に空きポートを選択してくれるそうです(テストの安定性的には未指定がベターでしょう)。
// SMTPのモックサーバの初期化 |
続いて、起動したgo-smtp-mockサーバのエンドポイントは testSMTPServer.PortNumber() の部分で取得できます。テスト対象のController(ああるいは、Usecase, Serviceなどの制御層)に渡します。今回の例では、HandlerがMailConfigという構造体を受け取るとします。
h := NewExampleMailSendHandler(db, MailConfig{ |
検証対象の機能を呼び出した後は、以下の部分でメールを受信します。受信とPurge(削除)を同時に行うため、複数のテストを連続して動かしても問題なしです。
msgs := testSMTPServer.MessagesAndPurge() |
検証は想定結果のファイルを次のような形式で用意し確認しました。メール本文だけ切り出して検証するか迷いましたが、From, To, Cc, Bcc, Subject, 本文が全て同じファイルにあったほうが検証しやすかったので、go-smtp-mockのメッセージ形式を採用しています。
From:future-sandbox@example.com |
使用時の注意点
便利なgo-smtp-mockですが、ドキュメントのfeaturesに次のように書かれています。
No authentication support
認証のサポートがない
そのため、認証付きのメール送信を行う場合はそのままだと使えません。例えば、次のようにユーザ名・パスワードが必要とします。
import ( |
そのため、テスト実行時はメール認証を無しにするような分岐が何かしら必要です。
例えば次のようなヘルパー関数を追加します(本当は ENABLE_NO_SMTP_AUTH=true のような環境変数で切り替えたほうが良い気もします)。
func (h ExampleMailSendHandler) SMTPAuth() smtp.Auth { |
これを先程のメール送信部分から読み込ませます。これによりgo-smtp-mockにメール送信ができました。
func (h ExampleMailSendHandler)SendMain(ctx context.Context, targetID string) error { |
使ってみて感想
認証部分は多少のハックが必要という少しだけ点は残念かもしれませんが、net/smtp パッケージ経由でメール送信して成功しているという状態を作れるので非常に安心できました。
go-smpt-mockはPure Go実装ですし、立ち上げも早く待ち時間を感じません。テストを何度も繰り返し実行するに当たりストレスはあまり感じませんでした。ライトなメール送信機能のテストを書く別の機会があれば、また利用すると思います。
READMEにあるサンプルコードのように、以下のオプションを入れるとgo-smtp-mockの動作が細かくログ出力されるため、疎通もさほど困りませんでした。初めて利用される方はこのオプションを有効にすることをお勧めします。
server := smtpmock.New(smtpmock.ConfigurationAttr{ |
まとめ
go-smtp-mockでSMTPのモックサーバと動かせ、単体テストに利用しました。
認証機能など不足している部分もあるが、その他ハマりどころは少なく利用できました。
それではハッピーなメールライフを!