module Sinatra::Test

def body ; @response.body ; end

def delete(path, *args, &b) ; test_request('DELETE', path, *args, &b) ; end

def env_for(opts={})
  opts = rack_opts(opts)
  Rack::MockRequest.env_for(opts)
end

def follow!
  test_request 'GET', @response.location
end

def get(path, *args, &b)  ; test_request('GET', path, *args, &b) ; end

def head(path, *args, &b) ; test_request('HEAD', path, *args, &b) ; end

def method_missing(name, *args, &block)
  if @response && @response.respond_to?(name)
    @response.send(name, *args, &block)
  else
    super
  end
end

def param_string(hash)
  hash.map { |pair| pair.map{|v|escape(v)}.join('=') }.join('&')
end

def post(path, *args, &b) ; test_request('POST', path, *args, &b) ; end

def put(path, *args, &b)  ; test_request('PUT', path, *args, &b) ; end

def rack_opts(opts)
  opts.inject({}) do |hash,(key,val)|
    key = RACK_OPT_NAMES[key] || key
    hash[key] = val
    hash
  end
end

def respond_to?(symbol, include_private=false)
  super || (@response && @response.respond_to?(symbol, include_private))
end

def should
  @response.should
end

def should
  @response.should
end

def status ; @response.status ; end

def test_request(verb, path, *args)
  @app = Sinatra::Application if @app.nil? && defined?(Sinatra::Application)
  fail "@app not set - cannot make request" if @app.nil?
  @request = Rack::MockRequest.new(@app)
  opts, input =
    case args.size
    when 2 # input, env
      input, env = args
      if input.kind_of?(Hash) # params, env
        [env, param_string(input)]
      else
        [env, input]
      end
    when 1 # params
      if (data = args.first).kind_of?(Hash)
        env = (data.delete(:env) || {})
        [env, param_string(data)]
      else
        [{}, data]
      end
    when 0
      [{}, '']
    else
      raise ArgumentError, "zero, one, or two arguments expected"
    end
  opts = rack_opts(opts)
  opts[:input] ||= input
  yield @request if block_given?
  @response = @request.request(verb, path, opts)
end