ruby-ajp 0.1.5

data/test/net/test_ajp13request.rb

@@ -0,0 +1,177 @@
+require 'test/unit'
+require 'stringio'
+require File.dirname(__FILE__) + '/../../lib/net/ajp13'
+
+class Net::AJP13::RequestTest < Test::Unit::TestCase
+  def setup
+    @req = Net::AJP13::Request.new('/')
+  end
+  def test_new
+    assert_kind_of Net::AJP13::Request, @req
+    assert_equal '/', @req.path
+
+    req_with_another_path = Net::AJP13::Request.new('/path/to/somewhere')
+    assert_equal '/path/to/somewhere', req_with_another_path.path
+  end
+
+  def test_header
+    assert_nil @req['accept-language']
+    assert_nil @req['content-type']
+    assert_nil @req['content-length']
+    
+    @req['accept-laguage'] = 'ja,en_US,en'
+    assert_equal 'ja,en_US,en', @req['accept-laguage']
+    @req['AcCePt-lAgUaGe'] = 'fr,de,es'
+    assert_equal 'fr,de,es', @req['accept-laguage']
+  end
+
+  def test_attribute
+    assert_nil @req.get_attributes('query_string')
+    assert_nil @req.get_attributes('something-unrecognized')
+
+    @req.set_attribute('QUERY_STRING', "a=1&b=c")
+    assert_equal ["a=1&b=c"], @req.get_attributes('qUeRy_StRiNg')
+    
+    @req.add_attribute('something-unrecognized', 'abcde')
+    @req.add_attribute('something-unrecognized', 'fghij')
+    assert_equal ['abcde', 'fghij'], @req.get_attributes('something-unrecognized')
+
+    @req.set_attribute('something-unrecognized', 'something else')
+    assert_equal ['something else'], @req.get_attributes('something-unrecognized')
+  end
+
+  def test_new
+    get = Net::AJP13::GetRequest.new('/path/to/somewhere', 
+                                     'Accept' => 'text/xml')
+    assert_equal 'GET', get.method
+    assert_equal '/path/to/somewhere', get.path
+    assert !get.request_body_permitted?
+    assert get.response_body_permitted?
+    assert_equal 'text/xml', get['accept']
+    assert_same false, get.is_ssl?
+
+    post = Net::AJP13::PostRequest.new('/path/to/somewhere/else',
+                                       'Accept' => 'application/x-dvi')
+    assert_equal 'POST', post.method
+    assert_equal '/path/to/somewhere/else', post.path
+    assert post.request_body_permitted?
+    assert post.response_body_permitted?
+    assert_equal 'application/x-dvi', post['accept']
+
+    head = Net::AJP13::HeadRequest.new('/path/to/yet/another/place',
+                                       'Accept' => 'application/x-pdf')
+    assert_equal 'HEAD', head.method
+    assert_equal '/path/to/yet/another/place', head.path
+    assert !head.request_body_permitted?
+    assert !head.response_body_permitted?
+    assert_equal 'application/x-pdf', head['accept']
+  end
+
+  def test_send_to
+    req = Net::AJP13::HeadRequest.new('/?q=something&a=1',
+                                      'Host' => 'localhost')
+    req.protocol = 'HTTP/1.1'
+    req.remote_addr = '127.0.0.1'
+    req.remote_host = 'localhost'
+    req.server_name = 'localhost'
+    req.server_port = 80
+    io = StringIO.new
+    req.send_to(io)
+    io.rewind
+    assert_equal \
+      "\x12\x34\x00\x5c" +          # payload
+      "\x02\x03" +                  # FORWARD_REQUEST HEAD
+      "\x00\x08HTTP/1.1\x00" +      # protocol = HTTP/1.1
+      "\x00\x01/\x00" +             # request_path = /
+      "\x00\x09127.0.0.1\x00" +     # remote_addr = 127.0.0.1
+      "\x00\x09localhost\x00" +     # remote_host = localhost
+      "\x00\x09localhost\x00" +     # server_name = localhost
+      "\x00\x50" +                  # server_port = 80
+      "\x00" +                      # is_ssl = false
+      "\x00\x01" +                  # num_headers = 1
+      "\xA0\x0B" +                  #  Host:
+      "\x00\x09localhost\0" +       #   localhost
+      "\x05" +                      # QUERY_STRING = 
+      "\x00\x0Fq=something&a=1\x00"+#  "q=something&a=1"
+      "\xFF" +                      # TERMINATOR 
+     "",
+      io.read
+  end
+
+  def test_from_io
+    packet =
+      "\x12\x34\x00\x5c" +          # payload
+      "\x02\x03" +                  # FORWARD_REQUEST HEAD
+      "\x00\x08HTTP/1.1\x00" +      # protocol = HTTP/1.1
+      "\x00\x01/\x00" +             # request_path = /
+      "\x00\x09127.0.0.1\x00" +     # remote_addr = 127.0.0.1
+      "\x00\x09localhost\x00" +     # remote_host = localhost
+      "\x00\x09localhost\x00" +     # server_name = localhost
+      "\x00\x50" +                  # server_port = 80
+      "\x00" +                      # is_ssl = false
+      "\x00\x01" +                  # num_headers = 1
+      "\xA0\x0B" +                  #  Host:
+      "\x00\x09localhost\x00" +       #   localhost
+      "\x05" +                      # QUERY_STRING = 
+      "\x00\x0Fq=something&a=1\x00"+#  "q=something&a=1"
+      "\xFF" +                      # TERMINATOR
+      ""
+    io = StringIO.new(packet)
+    assert !io.eof?
+    req = Net::AJP13::Request.from_io(io)
+    assert io.eof?
+    assert_equal "HEAD", req.method
+    assert_equal "/", req.path
+    assert_equal "127.0.0.1", req.remote_addr
+    assert_equal "localhost", req.remote_host
+    assert_equal "localhost", req.server_name
+    assert_equal 80, req.server_port
+    assert_same false, req.is_ssl?
+    assert_equal 1, req.length
+    assert_equal "localhost", req['host']
+
+    File.open(File.dirname(__FILE__) + '/data/ajp13request-data.1') {|f|
+      assert !f.eof?
+      req = Net::AJP13::Request.from_io(f)
+      assert f.eof?
+      assert_equal "GET", req.method
+      assert_equal "HTTP/1.1", req.protocol
+      assert_equal "/admin/", req.path
+      assert_equal "127.0.0.1", req.remote_addr
+      assert_nil req.remote_host
+      assert_equal "localhost", req.server_name
+      assert_equal 80, req.server_port
+      assert_same false, req.is_ssl?
+      assert_equal 11, req.length
+      req.each do |name, value|
+        case name.downcase
+        when 'user-agent'
+          assert_equal "Opera/8.51 (X11; Linux i686; U; ja)", value
+        when 'host'
+          assert_equal "localhost", value
+        when 'accept'
+          assert_equal "text/html, application/xml;q=0.9, application/xhtml+xml, image/png, image/jpeg, image/gif, image/x-xbitmap, */*;q=0.1", value
+        when 'accept-language'
+          assert_equal "ja,en;q=0.9", value
+        when 'accept-charset'
+          assert_equal "shift_jis, utf-8, utf-16, iso-8859-1;q=0.6, *;q=0.1", value
+        when 'accept-encoding'
+          assert_equal "deflate, gzip, x-gzip, identity, *;q=0", value
+        when 'cookie'
+          assert_block {
+            "JSESSIONID=54104A3A77560CEF8967D263F1A7193" == value or
+              "$Version=1" == value
+          }
+        when 'cache-control'
+          assert_equal "no-cache", value
+        when 'connection'
+          assert_equal "Keep-Alive, TE", value
+        when 'TE'
+          assert_equal "deflate, gzip, chunked, identity, trailers", value
+        when 'content-length'
+          assert_equal "0", value
+        end
+      end
+    }
+  end
+end

data/test/net/test_ajp13response.rb

@@ -0,0 +1,162 @@
+require 'test/unit'
+require 'stringio'
+require File.dirname(__FILE__) + '/../../lib/net/ajp13'
+
+class Net::AJP13::ResponseTest < Test::Unit::TestCase
+  def setup
+    @res = Net::AJP13::Response.new(301)
+  end
+
+  def test_new
+    assert_equal 301, @res.status
+    assert_equal "301", @res.code
+    assert_equal 'Moved Permanently', @res.message
+    assert_equal 'Moved Permanently', @res.reason_phrase
+    assert_same @res.message, @res.reason_phrase
+    assert_nil @res.body
+    assert_nil @res.body_stream
+  end
+
+  def test_from_io
+    packet = 
+      "\x41\x42\x00\x2e" +         # "AB" length
+      "\x04" +                     # SEND_HEADERS
+      "\x00\xC8\x00\x02OK\x00" +   # 200 OK
+      "\x00\x02" +                 # num_headers = 2
+      "\xA0\x01" +                 # Content-Type:
+      "\x00\x18text/html;" +
+        " charset=UTF-8\x00" +     #  text/html; charset=UTF-8
+      "\xA0\x03" +                 # Content-Length:
+      "\x00\x0241\x00" +           #   41
+      ""
+    io = StringIO.new(packet)
+    assert !io.eof?
+    res = Net::AJP13::Response.from_io(io)
+    assert io.eof?
+    assert_equal 200, res.status
+    assert_equal '200', res.code
+    assert_equal 'OK', res.message
+    assert !res.key?('Exprires')
+    assert_equal 'text/html; charset=UTF-8', res['content-type']
+    assert_equal '41', res['Content-Length']
+    assert_nil res.body
+    assert_nil res.body_stream
+
+    packet = 
+      "\x41\x42\x00\x11" +         # "AB" length
+      "\x04" +                     # SEND_HEADERS
+      "\x00\xC8\x00\x02OK\x00" +   # 200 OK
+      "\x00\x01" +                 # num_headers = 1
+      "\xA0\x3c" +                 # UNRECOGNIZED HEADER CODE
+      "\x00\x0241\x00" +           #   41
+      ""
+    assert_raise(Net::AJP13::AJPPacketError) {
+      res = Net::AJP13::Response.from_io(StringIO.new(packet))
+    }
+      
+   
+    
+    # WebDAV response without body
+    File.open(File.dirname(__FILE__) + '/data/ajp13response-webdav.1') { |f|
+      assert !f.eof?
+      res = Net::AJP13::Response.from_io(f)
+      assert !f.eof? # because of END_RESPONSE packet
+
+      assert_equal 200, res.status
+      assert_equal "200", res.code
+      assert_equal "OK", res.reason_phrase
+      assert_equal "1,2", res['dav']
+      assert_equal "OPTIONS, GET, HEAD, POST, DELETE, TRACE, PROPPATCH, COPY, MOVE, LOCK, UNLOCK, PROPFIND", res['ALLOW']
+      assert_equal "DAV", res['MS-Author-Via']
+      assert_equal 0, res.content_length
+      assert_nil res['no-such-header']
+      assert_nil res.body
+      assert_nil res.body_stream
+    }
+
+
+    # WebDAV response with body
+    File.open(File.dirname(__FILE__) + '/data/ajp13response-webdav.2') { |f|
+      assert !f.eof?
+      res = Net::AJP13::Response.from_io(f)
+      assert !f.eof? # because of BODY_CHUNK packets and END_RESPONSE packet.
+
+      assert_equal 207, res.status
+      assert_equal 'Multi-Status', res.message
+      assert_equal 'text/xml;charset=UTF-8', res['content-type']
+      assert_equal '390', res['content-length']
+      assert_equal 390, res.content_length
+      assert_nil res['no-such-header']
+      assert_nil res.body
+      assert_nil res.body_stream
+      #assert_equal %Q(<?xml version="1.0" encoding="utf-8" ?>\n<multistatus xmlns="DAV:"><response><href>/webdav/</href>\n<propstat><prop><resourcetype><collection/></resourcetype>\n</prop>\n<status>HTTP/1.1 200 OK</status>\n</propstat>\n<propstat><prop><getcontentlength/><getlastmodified/><executable/><checked-in/><checked-out/></prop>\n<status>HTTP/1.1 404 Not Found</status>\n</propstat>\n</response>\n</multistatus>\n), res.body
+    }
+  end
+
+
+  def test_message
+    res = Net::AJP13::Response.new(500, :reason_phrase => 'Internal Servlet Container Error :P')
+    assert_equal "Internal Servlet Container Error :P", res.message
+    assert_equal 500, res.status
+    assert_equal "500", res.code
+  end
+
+  def test_body
+    assert_nil @res.body
+    @res.body = "test body"
+    assert_equal "test body", @res.body
+  end
+
+  def test_body_stream
+    assert_nil @res.body
+    assert_nil @res.body_stream
+    @res.body_stream = io = StringIO.new('test body stream')
+    assert_same io, @res.body_stream
+    assert_nil @res.body
+  end
+
+  def test_send_to
+    res = Net::AJP13::Response.new(200, :reason_phrase => 'Yeah!')
+    res['content-length'] = '30'
+
+    io = StringIO.new
+    res.send_to(io)
+    io.rewind
+    assert_equal "" +
+      "\x41\x42\x00\x14" +         # "AB" length
+      "\x04" +                     # SEND_HEADERS
+      "\x00\xC8\x00\x05Yeah!\x00" +# 200 Yeah!
+      "\x00\x01" +                 # num_headers = 1
+      "\xA0\x03" +                 # Content-Length
+      "\x00\x0230\x00" +           #   30
+      "",
+      io.read
+  end
+
+  def test_send_to_and_from_io
+    res = Net::AJP13::Response.new(500, :reason_phrase => 'Something in the Servlet is wrong')
+    res['content-length'] = '50'
+    res['X-Framework'] = 'No such Framework T_T'
+    res['ConTent-tYpe'] = 'text/html; charset=BIG-5'
+    res['Content-Language'] = 'zh-CN'
+    res['Servlet-Engine'] = 'No such engine :P'
+
+    io = StringIO.new
+    res.send_to(io)
+    io.rewind
+
+    assert !io.eof?
+    res = Net::AJP13::Response.from_io(io)
+    assert io.eof?
+    assert_equal 500, res.status
+    assert_equal '500', res.code
+    assert_equal '50', res['Content-Length']
+    assert_equal 50, res.content_length
+    assert_equal 'zh-CN', res['Content-Language']
+    assert_equal 'text/html; charset=BIG-5', res['Content-Type']
+    assert_equal 'No such engine :P', res['Servlet-Engine']
+    assert_equal 'No such Framework T_T', res['X-Framework']
+    assert_nil res['No-Such-Header']
+  end
+end
+