Contents

require 'rack/test'

module Capybara::RackTestJson
  class Client
    include Rack::Test::Methods

    attr_accessor :app, :options

    def initialize(app, options)
      @app, @options = app, options
    end

    %w[ get post put patch delete ].each do |method|
      module_eval <<-DEF, __FILE__, __LINE__+1
        def #{method}(uri, params = {}, env = {}, &block)
          env = env.merge(:method => "#{method.upcase}", :params => params)
          if options[:follow_redirect]
            request_with_follow_redirect(uri, env, &block)
          else
            request(uri, env)
          end
        end
      DEF
    end

    def request_with_follow_redirect(uri, env)
      request(uri, env)
      (5+1).times do
        if last_response.redirect?
          request(last_response["Location"], env.merge(:method => "GET"))
        else
          return
        end
      end

      # 6 times redirect cause InfiniteRedirectError
      raise Capybara::InfiniteRedirectError,
      "redirected more than 5 times, check for infinite redirects."
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
capybara-json-0.4.0	lib/capybara/rack_test_json/client.rb