Sinatra + omniauth-twitter + RSpecでTwitterログインをテストする

SinatraでTwitterログインを下記みたいに実装しました。stubの使い方に少し悩んだ。

view

<a href="javascript:void(0);" onclick="$('#login_form').submit();">Twitterでログイン</a>

</form>

hiddenパラメータのback_urlにログイン後に戻りたい画面のURLを保存してます。ログインをaタグにしているのはCSSの都合なので、submitボタンでもOK。

main.rb

post "/login" do

# 値を引き回すため、画面から受け取ったback_urlをcookieに保存。

response.set_cookie("back_url", params[:back_url])

# /auth/twitterへリダイレクト

redirect "/auth/twitter"

end

# 認証後の処理

get "/auth/:provider/callback" do

auth = request.env["omniauth.auth"]

# 認証情報をsessionにセット

session[:twitter] = {

:uid => auth[:uid],

:name => auth[:info][:name],

:image => auth[:info][:image],

}

back_url = request.cookies['back_url'] || "/"

redirect back_url

end

Requestの中身をstubにしたい

認証後にsessionに正しく情報がセットされたかどうかを検証したいので、Request.envに値が入っている必要があります。
これは実際にTwitterログインを行わないと値が入ってこないので、Request.envをstubにしちゃいます。

main_spec.rb

describe "get /auth/:provider/callback" do

before do

# Twitterログイン認証情報のフェイク

@auth = {"omniauth.auth" => {

:uid => "testuid",

:info => {

:name => nil,

:image => nil,

}

}}

# Request.envをstubにする

# requestの中身をstubにしたいので、Sinatra::Request.any_instance〜という構文で

# 全てのSinatra::Requestのインスタンスに対してstubできる

Sinatra::Request.any_instance.stub(:env).and_return(@auth)

end

it "sessionにTwitterアカウント情報がセットされること" do

get "/auth/:provider/callback"

session[:twitter][:uid].should == @auth["omniauth.auth"][:uid]

end

やっぱりRSpecは書きやすいなー。
contextで分岐を表現できたりするのでそれぞれのテストメソッドが担当する箇所が明確にしやすい感じ。

この記事を書いた人
最新の記事

原田敦

日本CAWのエンジニア。もふもふ部の部長。得意分野はRuby on Railsを使った小規模WEBアプリケーションを高速で開発すること。週末の楽しみは一人お菓子パーティー。三度の飯より小動物をもふもふするのが好きです。

Requestの中身をstubにしたい

原田 敦

最新記事 by 原田 敦 (全て見る)

原田敦

最新記事 by 原田敦 (全て見る)