lib/amazon_auth/client.rb in amazon_auth-0.1.3 vs lib/amazon_auth/client.rb in amazon_auth-0.1.4

@@ -16,10 +16,11 @@
           Converter.decode(amazon_password_code)
         else
           raise('AMAZON_PASSWORD_CODE is required.')
         end
       end
+      @driver = options.fetch(:driver, :selenium)
     end
 
     def sign_in
       session.visit initial_url
       session.first('#a-autoid-0-announce').click
@@ -61,10 +62,10 @@
     def image_recognition_displayed?
       session.has_selector?('#auth-captcha-image-container')
     end
 
     def session
-      @session ||= Capybara::Session.new(:selenium)
+      @session ||= Capybara::Session.new(@driver)
     end
 
     def driver
       session.driver
     end