require File.dirname(__FILE__) + '/test_helper'

class ServerTest < Test::Unit::TestCase
  def setup
    @handler = TestHandler.new
    
    @socket = stub_everything
    TCPServer.stubs(:new).returns(@socket)
    
    @server = Thin::Server.new('0.0.0.0', 3000, @handler)
    @server.silent = true
  end
  
  def test_ok
    request "GET / HTTP/1.1\r\nHost: localhost:3000\r\n\r\n"
    
    assert_response '', :status => 200
  end
  
  def test_bad_request
    request "FUCKED / STUFF/1.1\r\nnononon"
    
    assert_response 'Bad request', :status => 400
  end
  
  def test_not_found
    @handler.stubs(:process).returns(false)
    request "GET / HTTP/1.1\r\nHost: localhost:3000\r\n\r\n"
    
    assert_response 'Page not found', :status => 404    
  end
  
  def test_ok_with_body
    request "POST / HTTP/1.1\r\nHost: localhost:3000\r\nContent-Length: 12\r\n\r\nmore cowbell"

    assert_response 'more cowbell', :status => 200
  end
  
  def test_invalid_content_length
    request "POST / HTTP/1.1\r\nHost: localhost:3000\r\nContent-Length: 324623\r\n\r\nmore cowbell"
    
    assert_response 'more cowbell', :status => 200
  end
  
  def test_stop
    @server.start!
    @socket.expects(:close)
    @server.stop
  end
  
  private
    def request(body)
      @client = StringIO.new(body)
      @client.instance_eval do
        alias :readpartial :read
      end
      @response = StringIO.new
      @socket.stubs(:accept).returns(@client)
      @socket.stubs(:closed?).returns(false).then.returns(true)
      
      @client.stubs(:peeraddr).returns(['127.0.0.1'])
      
      @client.stubs(:write).with { |o| @response << o }

      @server.start!
    end
      
    def assert_response(body, options={})
      @response.rewind
      response = @response.read
      status = options.delete(:status) || 200
      assert_match "HTTP/1.1 #{status} #{Thin::HTTP_STATUS_CODES[status]}", response
      assert_match body, response
    end
end