# -*- encoding: binary -*-

# Copyright (c) 2009 Eric Wong
# You can redistribute it and/or modify it under the same terms as Ruby 1.8 or
# the GPLv2+ (GPLv3+ preferred)

require './test/test_helper'

include Unicorn

class RequestTest < Test::Unit::TestCase

  class MockRequest < StringIO
    alias_method :readpartial, :sysread
    alias_method :kgio_read!, :sysread
    alias_method :read_nonblock, :sysread
    def kgio_addr
      '127.0.0.1'
    end
  end

  def setup
    @request = HttpRequest.new
    @app = lambda do |env|
      [ 200, { 'Content-Length' => '0', 'Content-Type' => 'text/plain' }, [] ]
    end
    @lint = Rack::Lint.new(@app)
  end

  def test_options
    client = MockRequest.new("OPTIONS * HTTP/1.1\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal '', env['REQUEST_PATH']
    assert_equal '', env['PATH_INFO']
    assert_equal '*', env['REQUEST_URI']
    res = @lint.call(env)
  end

  def test_absolute_uri_with_query
    client = MockRequest.new("GET http://e:3/x?y=z HTTP/1.1\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal '/x', env['REQUEST_PATH']
    assert_equal '/x', env['PATH_INFO']
    assert_equal 'y=z', env['QUERY_STRING']
    res = @lint.call(env)
  end

  def test_absolute_uri_with_fragment
    client = MockRequest.new("GET http://e:3/x#frag HTTP/1.1\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal '/x', env['REQUEST_PATH']
    assert_equal '/x', env['PATH_INFO']
    assert_equal '', env['QUERY_STRING']
    assert_equal 'frag', env['FRAGMENT']
    res = @lint.call(env)
  end

  def test_absolute_uri_with_query_and_fragment
    client = MockRequest.new("GET http://e:3/x?a=b#frag HTTP/1.1\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal '/x', env['REQUEST_PATH']
    assert_equal '/x', env['PATH_INFO']
    assert_equal 'a=b', env['QUERY_STRING']
    assert_equal 'frag', env['FRAGMENT']
    res = @lint.call(env)
  end

  def test_absolute_uri_unsupported_schemes
    %w(ssh+http://e/ ftp://e/x http+ssh://e/x).each do |abs_uri|
      client = MockRequest.new("GET #{abs_uri} HTTP/1.1\r\n" \
                               "Host: foo\r\n\r\n")
      assert_raises(HttpParserError) { @request.read(client, nil) }
    end
  end

  def test_x_forwarded_proto_https
    client = MockRequest.new("GET / HTTP/1.1\r\n" \
                             "X-Forwarded-Proto: https\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal "https", env['rack.url_scheme']
    res = @lint.call(env)
  end

  def test_x_forwarded_proto_http
    client = MockRequest.new("GET / HTTP/1.1\r\n" \
                             "X-Forwarded-Proto: http\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal "http", env['rack.url_scheme']
    res = @lint.call(env)
  end

  def test_x_forwarded_proto_invalid
    client = MockRequest.new("GET / HTTP/1.1\r\n" \
                             "X-Forwarded-Proto: ftp\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal "http", env['rack.url_scheme']
    res = @lint.call(env)
  end

  def test_rack_lint_get
    client = MockRequest.new("GET / HTTP/1.1\r\nHost: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal "http", env['rack.url_scheme']
    assert_equal '127.0.0.1', env['REMOTE_ADDR']
    res = @lint.call(env)
  end

  def test_no_content_stringio
    client = MockRequest.new("GET / HTTP/1.1\r\nHost: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal StringIO, env['rack.input'].class
  end

  def test_zero_content_stringio
    client = MockRequest.new("PUT / HTTP/1.1\r\n" \
                             "Content-Length: 0\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal StringIO, env['rack.input'].class
  end

  def test_real_content_not_stringio
    client = MockRequest.new("PUT / HTTP/1.1\r\n" \
                             "Content-Length: 1\r\n" \
                             "Host: foo\r\n\r\n")
    env = @request.read(client, nil)
    assert_equal Unicorn::TeeInput, env['rack.input'].class
  end

  def test_rack_lint_put
    client = MockRequest.new(
      "PUT / HTTP/1.1\r\n" \
      "Host: foo\r\n" \
      "Content-Length: 5\r\n" \
      "\r\n" \
      "abcde")
    env = @request.read(client, nil)
    assert ! env.include?(:http_body)
    res = @lint.call(env)
  end

  def test_rack_lint_big_put
    count = 100
    bs = 0x10000
    buf = (' ' * bs).freeze
    length = bs * count
    client = Tempfile.new('big_put')
    def client.kgio_addr; '127.0.0.1'; end
    def client.kgio_read(*args)
      readpartial(*args)
    rescue EOFError
    end
    def client.kgio_read!(*args)
      readpartial(*args)
    end
    client.syswrite(
      "PUT / HTTP/1.1\r\n" \
      "Host: foo\r\n" \
      "Content-Length: #{length}\r\n" \
      "\r\n")
    count.times { assert_equal bs, client.syswrite(buf) }
    assert_equal 0, client.sysseek(0)
    env = @request.read(client, nil)
    assert ! env.include?(:http_body)
    assert_equal length, env['rack.input'].size
    count.times {
      tmp = env['rack.input'].read(bs)
      tmp << env['rack.input'].read(bs - tmp.size) if tmp.size != bs
      assert_equal buf, tmp
    }
    assert_nil env['rack.input'].read(bs)
    env['rack.input'].rewind
    res = @lint.call(env)
  end
end