Rails開発でWebMockを使ってAPIアクセスをスタブ化する

外部API呼び出しのあるアプリケーションを開発する場合、テストやローカル環境での動作確認まで本物のAPIを呼び出していると、時間がかかり開発の生産性が下がるのでAPIモックを使いたくなります。
そこでWebMockというGemを使うと、外部へのHTTPリクエストをスタブ化してくれるので、開発が非常にやりやすくなります。
今回はRailsアプリケーションでのWebMockの使い方を説明します。

インストール方法

Gemfileに書いてbundlerでインストールします。

gem 'webmock'

bundle install --path vendor/bundle

設定方法

事前に自分の欲しいレスポンスを返却できるようにスタブを登録します。
(ここではレスポンスデータを事前に作成して、そのファイルを読み込むように指定しています)

require 'webmock'

WebMock.enable! 

# GETリクエストのスタブ登録
WebMock.stub_request(:get, "http://www.example.com").to_return(
  body: File.read("#{Rails.root}/test/fixtures/stub_api_response.json"),
  status: 200,
  headers: { 'Content-Type' =>  'application/json' })

# PUTリクエストのスタブ登録  
WebMock.stub_request(:put, "http://www.example.com").to_return(
  body: '{"key": "value"}',
  status: 200,
  headers: { 'Content-Type' =>  'application/json' })

# 同じように他HTTPメソッドのリクエストもスタブ登録できます

これでスタブの設定はできたので、該当のAPIを呼び出すと設定したレスポンスデータが返却されるのですが、WebMockはデフォルト設定だとスタブ登録していないAPIのアクセスを許可していません。
スタブ化していないAPIのアクセスを許可するには以下の記述を追加します。

require 'webmock'
WebMock.allow_net_connect!(:net_http_connect_on_start => true)

RSpec

テストでも外部APIへのリクエストをスタブ化します。同じようにスタブ登録するだけです

require 'rails_helper'
require 'webmock/rspec'

describe ApiTest
  before do
    WebMock.enable!

  end

  describe '#get api' do
    context '200 status' do
      before do
        # ここでテスト対象コードの中に実装されている外部API呼び出しをstub登録します
        WebMock.stub_request(:get, "http://www.example.com").to_return(
          body: File.read("#{Rails.root}/test/fixtures/stub_api_response.json"),
          status: 200,
          headers: { 'Content-Type' =>  'application/json' })
      end

      it 'returns 200 status' do
        # ここにテストコードが入ります
      end
    end
  end
end

アプリケーションへの組み込み

次はRailsアプリケーションへWebMockを組み込む方法を説明します。
WebMockをアプリケーションコードに組み込むときに以下のような制約があるとします。

• development, test環境のみ使用する。staging、production環境ではWebMockをインストールしたくない
• development環境でも、WebMockを使用する場合と使用しない場合で使い分けたい
• 外部API呼び出しの実装はModule化している

この場合のWebMockの組み込みの一例を書いてみます

まず、Gemfileのdevelopment, test環境にのみ以下の記述を追加します。
require falseのオプションを追加して明示的に読み込むようにします。

group :development, :test do
  gem 'webmock', require: false
end

次はdevelopment.rbとtest.rbにWebMockを読み込むように記述します。
今回は以下のように設計します。

• development環境でWebMockを使用する場合は環境変数に値を設定する
• test/stub/api_stub.rbファイルに外部API呼び出しをstub登録するコードを実装する

development.rb

  if ENV["APIMODE"] == "stub" # 環境変数の有無を判定します
    require 'webmock'　# ここで明示的に読み込みます
    WebMock.allow_net_connect!(:net_http_connect_on_start => true)
    require_relative '../../test/stub/api_stub'
  end

test.rb

  require 'webmock'
  WebMock.allow_net_connect!(:net_http_connect_on_start => true)

stub登録の実装はtestディレクトリのファイルなので、ここで環境変数が設定されていない場合や、staging、production環境の場合は読み込まれません。

stub登録の実装は今までの説明と同じ内容ですが、stub登録のコードを実行後に本来の外部API呼び出しのコードを実行するため、最後にsuperの記述を追加しています。

api_stub.rb

module ApiStub
  WebMock.enable!

  # アプリケーションコードに`call_get_api`という名称の外部API呼び出し実装を含むメソッドがあることして、ここで同名のメソッドを作成してstub登録します。  
  def call_get_api
    WebMock.stub_request(:get, "http://www.example.com").to_return(
      body: File.read("#{Rails.root}/test/fixtures/stub_api_response.json"),
      status: 200,
      headers: { 'Content-Type' =>  'application/json' })　　

    super # アプリケーションコードの実装を呼び出す
  end
end

次はアプリケーションコードからこのstub登録をするファイルを読み込むように設定します。
WebMockの使用有無を判定する環境変数とRAILS_ENVの二つを確認してstub登録の実装を読み込んでいます。

module ApiModule
  prepend ApiStub if ENV["APIMODE"] == "mock" && Rails.env.development?

  def call_get_api
    # ここに本来の外部API呼び出しの実装が入る
  end
end

ここではprependでスタブ実装のmoduleを読み込んでいます。そのためこのモジュールを読み込んだクラスの継承ツリーは以下のようになります。

1. ApiModule(外部API呼び出し実装)
2. ApiStub(stub登録コード)
3. モジュールを読み込んだクラス

stub登録の実装で最後にsuperと記述しているのは、この継承ツリーが作られることを前提に実装しているからです。

最後に外部API呼び出しのmoduleを使用するクラスですが、通常通りmoduleをincludeするだけです。

api_client.rb

class ApiClient
  include ApiModule
  # ここに実装が入ります
end

まとめ

WebMockを使えば外部API呼び出しのmockを自分で実装しなくても簡単にstub登録できます。
外部API呼び出しのあるRailsアプリケーションを開発する場合はWebMockの使用を検討してみてください。