stub session in test mode

Author: AlexanderPavlenko

lib/omniauth/strategies/oauth2.rb

@@ -51,6 +51,10 @@ def authorize_params
           options.authorize_params[:state] = 3.times.map{ rand.to_s[2..-1] }.reduce(&:concat)
         end
         params = options.authorize_params.merge(options.authorize_options.inject({}){|h,k| h[k.to_sym] = options[k] if options[k]; h})
+        if OmniAuth.config.test_mode
+          @env ||= {}
+          @env['rack.session'] ||= {}
+        end
         session['omniauth.state'] = params[:state]
         params
       end

spec/omniauth/strategies/oauth2_spec.rb

@@ -4,6 +4,14 @@
   def app; lambda{|env| [200, {}, ["Hello."]]} end
   let(:fresh_strategy){ Class.new(OmniAuth::Strategies::OAuth2) }
 
+  before do
+    OmniAuth.config.test_mode = true
+  end
+
+  after do
+    OmniAuth.config.test_mode = false
+  end
+
   describe '#client' do
     subject{ fresh_strategy }