unicorn 0.1.0

data/test/unit/test_upload.rb

@@ -0,0 +1,151 @@
+# Copyright (c) 2009 Eric Wong
+require 'test/test_helper'
+
+include Unicorn
+
+class UploadTest < Test::Unit::TestCase
+
+  def setup
+    @addr = ENV['UNICORN_TEST_ADDR'] || '127.0.0.1'
+    @port = unused_port
+    @hdr = {'Content-Type' => 'text/plain', 'Content-Length' => '0'}
+    @bs = 4096
+    @count = 256
+    @server = nil
+
+    # we want random binary data to test 1.9 encoding-aware IO craziness
+    @random = File.open('/dev/urandom','rb')
+    @sha1 = Digest::SHA1.new
+    @sha1_app = lambda do |env|
+      input = env['rack.input']
+      resp = { :pos => input.pos, :size => input.stat.size }
+      begin
+        loop { @sha1.update(input.sysread(@bs)) }
+      rescue EOFError
+      end
+      resp[:sha1] = @sha1.hexdigest
+      [ 200, @hdr.merge({'X-Resp' => resp.inspect}), [] ]
+    end
+  end
+
+  def teardown
+    redirect_test_io { @server.stop(true) } if @server
+    @random.close
+  end
+
+  def test_put
+    start_server(@sha1_app)
+    sock = TCPSocket.new(@addr, @port)
+    sock.syswrite("PUT / HTTP/1.0\r\nContent-Length: #{length}\r\n\r\n")
+    @count.times do
+      buf = @random.sysread(@bs)
+      @sha1.update(buf)
+      sock.syswrite(buf)
+    end
+    read = sock.read.split(/\r\n/)
+    assert_equal "HTTP/1.1 200 OK", read[0]
+    resp = eval(read.grep(/^X-Resp: /).first.sub!(/X-Resp: /, ''))
+    assert_equal length, resp[:size]
+    assert_equal 0, resp[:pos]
+    assert_equal @sha1.hexdigest, resp[:sha1]
+  end
+
+
+  def test_put_keepalive_truncates_small_overwrite
+    start_server(@sha1_app)
+    sock = TCPSocket.new(@addr, @port)
+    to_upload = length + 1
+    sock.syswrite("PUT / HTTP/1.0\r\nContent-Length: #{to_upload}\r\n\r\n")
+    @count.times do
+      buf = @random.sysread(@bs)
+      @sha1.update(buf)
+      sock.syswrite(buf)
+    end
+    sock.syswrite('12345') # write 4 bytes more than we expected
+    @sha1.update('1')
+
+    read = sock.read.split(/\r\n/)
+    assert_equal "HTTP/1.1 200 OK", read[0]
+    resp = eval(read.grep(/^X-Resp: /).first.sub!(/X-Resp: /, ''))
+    assert_equal to_upload, resp[:size]
+    assert_equal 0, resp[:pos]
+    assert_equal @sha1.hexdigest, resp[:sha1]
+  end
+
+  def test_put_excessive_overwrite_closed
+    start_server(lambda { |env| [ 200, @hdr, [] ] })
+    sock = TCPSocket.new(@addr, @port)
+    buf = ' ' * @bs
+    sock.syswrite("PUT / HTTP/1.0\r\nContent-Length: #{length}\r\n\r\n")
+    @count.times { sock.syswrite(buf) }
+    assert_raise(Errno::ECONNRESET, Errno::EPIPE) do
+      ::Unicorn::Const::CHUNK_SIZE.times { sock.syswrite(buf) }
+    end
+  end
+
+  def test_put_handler_closed_file
+    nr = '0'
+    start_server(lambda { |env|
+      env['rack.input'].close
+      resp = { :nr => nr.succ! }
+      [ 200, @hdr.merge({ 'X-Resp' => resp.inspect}), [] ]
+    })
+    sock = TCPSocket.new(@addr, @port)
+    buf = ' ' * @bs
+    sock.syswrite("PUT / HTTP/1.0\r\nContent-Length: #{length}\r\n\r\n")
+    @count.times { sock.syswrite(buf) }
+    read = sock.read.split(/\r\n/)
+    assert_equal "HTTP/1.1 200 OK", read[0]
+    resp = eval(read.grep(/^X-Resp: /).first.sub!(/X-Resp: /, ''))
+    assert_equal '1', resp[:nr]
+
+    # server still alive?
+    sock = TCPSocket.new(@addr, @port)
+    sock.syswrite("GET / HTTP/1.0\r\n\r\n")
+    read = sock.read.split(/\r\n/)
+    assert_equal "HTTP/1.1 200 OK", read[0]
+    resp = eval(read.grep(/^X-Resp: /).first.sub!(/X-Resp: /, ''))
+    assert_equal '2', resp[:nr]
+  end
+
+  def test_renamed_file_not_closed
+    start_server(lambda { |env|
+      new_tmp = Tempfile.new('unicorn_test')
+      input = env['rack.input']
+      File.rename(input.path, new_tmp.path)
+      resp = {
+        :inode => input.stat.ino,
+        :size => input.stat.size,
+        :new_tmp => new_tmp.path,
+        :old_tmp => input.path,
+      }
+      [ 200, @hdr.merge({ 'X-Resp' => resp.inspect}), [] ]
+    })
+    sock = TCPSocket.new(@addr, @port)
+    buf = ' ' * @bs
+    sock.syswrite("PUT / HTTP/1.0\r\nContent-Length: #{length}\r\n\r\n")
+    @count.times { sock.syswrite(buf) }
+    read = sock.read.split(/\r\n/)
+    assert_equal "HTTP/1.1 200 OK", read[0]
+    resp = eval(read.grep(/^X-Resp: /).first.sub!(/X-Resp: /, ''))
+    new_tmp = File.open(resp[:new_tmp])
+    assert_equal resp[:inode], new_tmp.stat.ino
+    assert_equal length, resp[:size]
+    assert ! File.exist?(resp[:old_tmp])
+    assert_equal resp[:size], new_tmp.stat.size
+  end
+
+  private
+
+  def length
+    @bs * @count
+  end
+
+  def start_server(app)
+    redirect_test_io do
+      @server = HttpServer.new(app, :listeners => [ "#{@addr}:#{@port}" ] )
+      @server.start
+    end
+  end
+
+end

data/unicorn.gemspec

@@ -0,0 +1,35 @@
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{unicorn}
+  s.version = "0.1.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Eric Wong"]
+  s.date = %q{2009-03-10}
+  s.default_executable = %q{unicorn}
+  s.description = %q{A small fast HTTP library and server for Rack applications.}
+  s.email = %q{normalperson@yhbt.net}
+  s.executables = ["unicorn"]
+  s.extensions = ["ext/unicorn/http11/extconf.rb"]
+  s.extra_rdoc_files = ["CHANGELOG", "LICENSE", "README", "TODO", "bin/unicorn", "ext/unicorn/http11/ext_help.h", "ext/unicorn/http11/extconf.rb", "ext/unicorn/http11/http11.c", "ext/unicorn/http11/http11_parser.c", "ext/unicorn/http11/http11_parser.h", "ext/unicorn/http11/http11_parser.rl", "ext/unicorn/http11/http11_parser_common.rl", "lib/unicorn.rb", "lib/unicorn/configurator.rb", "lib/unicorn/const.rb", "lib/unicorn/http_request.rb", "lib/unicorn/http_response.rb", "lib/unicorn/socket.rb", "lib/unicorn/util.rb"]
+  s.files = [".document", ".gitignore", "CHANGELOG", "CONTRIBUTORS", "DESIGN", "GNUmakefile", "LICENSE", "Manifest", "README", "Rakefile", "SIGNALS", "TODO", "bin/unicorn", "ext/unicorn/http11/ext_help.h", "ext/unicorn/http11/extconf.rb", "ext/unicorn/http11/http11.c", "ext/unicorn/http11/http11_parser.c", "ext/unicorn/http11/http11_parser.h", "ext/unicorn/http11/http11_parser.rl", "ext/unicorn/http11/http11_parser_common.rl", "lib/unicorn.rb", "lib/unicorn/configurator.rb", "lib/unicorn/const.rb", "lib/unicorn/http_request.rb", "lib/unicorn/http_response.rb", "lib/unicorn/socket.rb", "lib/unicorn/util.rb", "setup.rb", "test/aggregate.rb", "test/benchmark/previous.rb", "test/benchmark/simple.rb", "test/benchmark/utils.rb", "test/exec/README", "test/exec/test_exec.rb", "test/test_helper.rb", "test/tools/trickletest.rb", "test/unit/test_configurator.rb", "test/unit/test_http_parser.rb", "test/unit/test_response.rb", "test/unit/test_server.rb", "test/unit/test_upload.rb", "unicorn.gemspec"]
+  s.has_rdoc = true
+  s.homepage = %q{http://unicorn.bogomips.org}
+  s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Unicorn", "--main", "README"]
+  s.require_paths = ["lib", "ext"]
+  s.rubyforge_project = %q{unicorn}
+  s.rubygems_version = %q{1.3.1}
+  s.summary = %q{A small fast HTTP library and server for Rack applications.}
+  s.test_files = ["test/unit/test_configurator.rb", "test/unit/test_response.rb", "test/unit/test_upload.rb", "test/unit/test_http_parser.rb", "test/unit/test_server.rb"]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 2
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+    else
+    end
+  else
+  end
+end

metadata

@@ -0,0 +1,122 @@
+--- !ruby/object:Gem::Specification 
+name: unicorn
+version: !ruby/object:Gem::Version 
+  version: 0.1.0
+platform: ruby
+authors: 
+- Eric Wong
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-03-10 00:00:00 -07:00
+default_executable: 
+dependencies: []
+
+description: A small fast HTTP library and server for Rack applications.
+email: normalperson@yhbt.net
+executables: 
+- unicorn
+extensions: 
+- ext/unicorn/http11/extconf.rb
+extra_rdoc_files: 
+- CHANGELOG
+- LICENSE
+- README
+- TODO
+- bin/unicorn
+- ext/unicorn/http11/ext_help.h
+- ext/unicorn/http11/extconf.rb
+- ext/unicorn/http11/http11.c
+- ext/unicorn/http11/http11_parser.c
+- ext/unicorn/http11/http11_parser.h
+- ext/unicorn/http11/http11_parser.rl
+- ext/unicorn/http11/http11_parser_common.rl
+- lib/unicorn.rb
+- lib/unicorn/configurator.rb
+- lib/unicorn/const.rb
+- lib/unicorn/http_request.rb
+- lib/unicorn/http_response.rb
+- lib/unicorn/socket.rb
+- lib/unicorn/util.rb
+files: 
+- .document
+- .gitignore
+- CHANGELOG
+- CONTRIBUTORS
+- DESIGN
+- GNUmakefile
+- LICENSE
+- Manifest
+- README
+- Rakefile
+- SIGNALS
+- TODO
+- bin/unicorn
+- ext/unicorn/http11/ext_help.h
+- ext/unicorn/http11/extconf.rb
+- ext/unicorn/http11/http11.c
+- ext/unicorn/http11/http11_parser.c
+- ext/unicorn/http11/http11_parser.h
+- ext/unicorn/http11/http11_parser.rl
+- ext/unicorn/http11/http11_parser_common.rl
+- lib/unicorn.rb
+- lib/unicorn/configurator.rb
+- lib/unicorn/const.rb
+- lib/unicorn/http_request.rb
+- lib/unicorn/http_response.rb
+- lib/unicorn/socket.rb
+- lib/unicorn/util.rb
+- setup.rb
+- test/aggregate.rb
+- test/benchmark/previous.rb
+- test/benchmark/simple.rb
+- test/benchmark/utils.rb
+- test/exec/README
+- test/exec/test_exec.rb
+- test/test_helper.rb
+- test/tools/trickletest.rb
+- test/unit/test_configurator.rb
+- test/unit/test_http_parser.rb
+- test/unit/test_response.rb
+- test/unit/test_server.rb
+- test/unit/test_upload.rb
+- unicorn.gemspec
+has_rdoc: true
+homepage: http://unicorn.bogomips.org
+post_install_message: 
+rdoc_options: 
+- --line-numbers
+- --inline-source
+- --title
+- Unicorn
+- --main
+- README
+require_paths: 
+- lib
+- ext
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "1.2"
+  version: 
+requirements: []
+
+rubyforge_project: unicorn
+rubygems_version: 1.3.1
+signing_key: 
+specification_version: 2
+summary: A small fast HTTP library and server for Rack applications.
+test_files: 
+- test/unit/test_configurator.rb
+- test/unit/test_response.rb
+- test/unit/test_upload.rb
+- test/unit/test_http_parser.rb
+- test/unit/test_server.rb