ruby-ajp 0.1.5

data/test/net/test_ajp13client.rb

@@ -0,0 +1,353 @@
+require 'test/unit'
+require File.dirname(__FILE__) + '/../../lib/net/ajp13client'
+
+class Net::AJP13::ClientTest < Test::Unit::TestCase
+  def setup
+    begin
+      sock = TCPSocket.new('localhost', 8009)
+    rescue
+      warn "#{__FILE__}:#{__LINE__}: This test case requires that" +
+        " a servlet container is listening localhost:8009." +
+	" Almost all tests in #{self.class} will be ignored."
+      @ignored = true
+    else
+      sock.close
+    end
+    @c = Net::AJP13::Client.new('localhost')
+  end
+
+  def test_new
+    assert_equal 'localhost', @c.address
+    assert_equal 8009, @c.port
+    assert !@c.started?
+    assert !@c.assigned?
+    assert !@c.idle?
+    assert_nil @c.state
+    assert_kind_of Integer, @c.open_timeout
+    assert_kind_of Integer, @c.read_timeout
+    assert_kind_of Integer, @c.write_timeout
+
+    c = Net::AJP13::Client.new('localhost', 9008)
+    assert_equal 'localhost', c.address
+    assert_equal 9008, c.port
+    assert !c.started?
+    assert !c.assigned?
+    assert !c.idle?
+    assert_nil c.state
+
+    # Nothing raised because client does not try to connect until #start()ing.
+    assert_nothing_raised { 
+      c = Net::AJP13::Client.new('test.invalid') # unresolvable by RFC2026
+    }
+    assert_equal 'test.invalid', c.address
+    assert_equal 8009, c.port
+    assert !c.started?
+    assert !c.assigned?
+    assert !c.idle?
+    assert_nil c.state
+  end
+
+  def test_start_finish
+    return if @ignored
+
+    assert_nothing_raised { @c.start }
+    assert @c.started?
+    assert @c.idle?
+    assert !@c.assigned?
+    assert_equal :idle, @c.state
+
+    assert_nothing_raised { @c.finish }
+    assert !@c.started?
+    assert !@c.assigned?
+    assert !@c.idle?
+    assert_nil @c.state
+  end
+
+  def test_finish
+    assert_raise(Net::AJP13::AJPStateError) { @c.finish }
+  end
+
+  def test_start_with_block
+    return if @ignored
+
+    expected = Object.new
+    retval = @c.start { |c|
+      assert_same @c, c
+      assert @c.started?
+      assert @c.idle?
+      assert !@c.assigned?
+      assert_equal :idle, @c.state
+
+      expected
+    }
+    assert_same expected, retval
+
+    assert !@c.started?
+    assert !@c.assigned?
+    assert !@c.idle?
+    assert_nil @c.state
+  end
+
+  def test_start_with_block_on_raised
+    return if @ignored
+
+    assert_raise(RuntimeError, 'Error on connection opened') {
+      @c.start { |c|
+        assert_same @c, c
+         assert @c.started?
+	 assert @c.idle?
+	 assert !@c.assigned?
+	 assert_equal :idle, @c.state
+
+         raise 'Error on connection opened'
+      }
+    }
+    assert !@c.started?
+    assert !@c.assigned?
+    assert !@c.idle?
+    assert_nil @c.state
+  end
+
+  def test_finish_only_once
+    return if @ignored
+
+    class << @c
+      alias :orig_finish :finish
+      def finish
+        @finish_count ||= 0
+	@finish_count += 1
+	orig_finish
+      end
+      attr :finish_count
+    end
+    @c.start { |c|
+      assert @c.started?
+      assert @c.idle?
+      assert !@c.assigned?
+
+      c.finish
+
+      assert !@c.started?
+      assert !@c.idle?
+      assert !@c.assigned?
+    }
+    assert_equal 1, @c.finish_count
+    assert !@c.started?
+    assert !@c.idle?
+    assert !@c.assigned?
+  end
+
+
+  def test_restart
+    return if @ignored
+
+    class << @c
+      alias :orig_finish :finish
+      def finish
+        @finish_count ||= 0
+	@finish_count += 1
+	orig_finish
+      end
+      attr :finish_count
+    end
+    @c.start { |c|
+      assert @c.started?
+      assert @c.idle?
+      assert !@c.assigned?
+
+      c.finish
+
+      assert !@c.started?
+      assert !@c.idle?
+      assert !@c.assigned?
+
+      c.start
+
+      assert @c.started?
+      assert @c.idle?
+      assert !@c.assigned?
+    }
+    assert_equal 2, @c.finish_count
+
+    assert !@c.started?
+    assert !@c.assigned?
+    assert !@c.idle?
+  end
+
+  def test_class_start
+    return if @ignored
+
+    c = Net::AJP13::Client.start('localhost')
+    begin
+      assert_equal 'localhost', c.address
+      assert_equal 8009, c.port
+      assert c.started?
+      assert c.idle?
+    return if @ignored
+    return if @ignored
+
+
+      assert !c.assigned?
+      assert_equal :idle, c.state
+    ensure
+      c.finish
+    end
+  end
+
+  def test_class_start_with_block
+    return if @ignored
+
+    created_conn = nil
+    expected = Object.new
+    retval = Net::AJP13::Client.start('localhost') { |c|
+      assert_equal 'localhost', c.address
+      assert_equal 8009, c.port
+      assert c.started?
+      assert c.idle?
+      assert !c.assigned?
+      assert_equal :idle, c.state
+
+      created_conn = c
+      expected
+    }
+    assert_same expected, retval
+
+    assert !created_conn.started?
+    assert !created_conn.assigned?
+    assert !created_conn.idle?
+    assert_nil created_conn.state
+  end
+
+  def test_ask_to_shutdown
+    return if @ignored
+
+    @c.start { |conn|
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+
+      assert_nothing_raised { conn.ask_to_shutdown }
+
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+    }
+  end
+
+  def test_ping
+    return if @ignored
+
+    @c.start { |conn|
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+      
+      conn.ping
+
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+    }
+  end
+
+  def test_request
+    return if @ignored
+
+    req = Object.new
+    class << req
+      def send_to(io)
+        io.write "" +
+	  "\x12\x34\x00\x35" +       # signature length
+          "\x02\x03" +               # FORWARD_REQUEST HEAD
+	  "\x00\x08HTTP/1.0\x00" +   # protocol: HTTP/1.0
+	  "\x00\x01/\x00" +          # req_uri:  /
+	  "\x00\x09127.0.0.1\x00" +  # remote_addr: 127.0.0.1
+	  "\xFF\xFF" +               # remote_host: nil
+	  "\xFF\xFF" +               # server_name: nil
+	  "\x00\x50" +               # server_port: 80
+	  "\x00" +                   # is_ssl : false
+	  "\x00\x01" +               # num_headers = 1
+	  "\xA0\x0B" +               #  Host:
+	  "\x00\x09localhost\x00" +  #   localhost
+	  "\xFF"                     # TERMINATOR
+      end
+      def method_missing(*args)
+        nil
+      end
+    end
+
+    @c.start { |conn|
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+      
+      res = conn.request(req)
+
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+    }
+  end
+
+  def test_request_with_body
+    return if @ignored
+
+    req = Net::AJP13::PostRequest.new('/index.jsp')
+    req.body = (['a=1'] * 10 * 1024).join('&')
+    @c.start { |conn|
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+
+      res = conn.request(req)
+
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+    }
+  end
+
+  def test_request_with_block
+    return if @ignored
+
+    req = Net::AJP13::PostRequest.new('/index.jsp')
+    req.body = (['a=1'] * 10 * 1024).join('&')
+    @c.start { |conn|
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+
+      body = ''
+      res = conn.request(req){ |frag|
+        assert conn.started?
+	assert !conn.idle?
+	assert conn.assigned?
+	assert_kind_of String, frag
+	body << frag
+      }
+      assert_nil res.body
+      assert body.length > 0
+
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+    }
+  end
+
+  def test_get
+    return if @ignored
+
+    @c.start { |conn|
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+      
+      res = conn.get('/', 'User-Agent' => "Ruby/AJP #{__FILE__}")
+      assert res.body.length > 0
+
+      assert conn.started?
+      assert conn.idle?
+      assert !conn.assigned?
+    }
+  end
+end

data/test/net/test_ajp13packet.rb

@@ -0,0 +1,320 @@
+require 'test/unit'
+require 'stringio'
+require File.dirname(__FILE__) + '/../../lib/net/ajp13'
+
+class Net::AJP13::PacketTest < Test::Unit::TestCase
+  def setup
+    @packet = Net::AJP13::Packet.new
+  end
+  def test_direction
+    assert_nil @packet.direction
+
+    @packet.direction = :to_app
+    assert_equal :to_app, @packet.direction
+
+    @packet.direction = :from_app
+    assert_equal :from_app, @packet.direction
+
+    assert_raise(ArgumentError) { @packet.direction = :something_wrong }
+    assert_raise(ArgumentError) { @packet.direction = 'to_app' }
+  end
+
+  def test_length
+    assert_equal 0, @packet.length
+  end
+
+  def test_from_io
+    data = StringIO.new("\x41\x42\x00\x02\x05\x00") # AJP13_END_RESPONSE packet
+    assert_nothing_raised { Net::AJP13::Packet.from_io(data) }
+    assert 4, data.pos
+
+    data = StringIO.new("\x12\x34\x00\x01\x0A") # AJP13_CPING
+    assert_nothing_raised { Net::AJP13::Packet.from_io(data) }
+    assert 4, data.pos
+
+    # packet with unrecognized header
+    data = StringIO.new("\x63\x23\x00\x00\x00\x00\x00") 
+    assert_raise(Net::AJP13::AJPPacketError) { Net::AJP13::Packet.from_io(data) }
+    
+    # packet with mismatched length
+    data = StringIO.new("\x12\x34\x00\x010\x00\x00\x00")
+    # nothing raised until Packet#read_* called
+    assert_nothing_raised { Net::AJP13::Packet.from_io(data) }
+    assert 4, data.pos
+
+    data = StringIO.new("\x12\x34\x00") # packet with imcomplete length field
+    assert_raise(Net::AJP13::AJPPacketError) { Net::AJP13::Packet.from_io(data) }
+    assert 4, data.pos
+  end
+
+  def test_send_to
+    io = StringIO.new
+    assert_raise(Net::AJP13::AJPPacketError, 'Packet direction not specified') {
+      @packet.send_to(io)
+    }
+
+    @packet.direction = :to_app
+    @packet.send_to(io)
+    io.rewind
+    assert_equal "\x12\x34\x00\x00", io.read
+
+    io = StringIO.new
+    @packet.direction = :from_app
+    @packet.send_to(io)
+    io.rewind
+    assert_equal "\x41\x42\x00\x00", io.read
+  end
+
+  def test_append_byte
+    @packet.direction = :from_app
+    @packet.append_byte 0x13
+    @packet.append_byte 0x35
+    assert_raise(ArgumentError) { @packet.append_byte 0x100 }
+    assert_raise(ArgumentError) { @packet.append_byte 0x3400 }
+
+    io = StringIO.new
+    @packet.send_to(io)
+    io.rewind
+    assert_equal "\x41\x42\x00\x02\x13\x35", io.read
+  end
+
+  def test_append_bytes
+    @packet.direction = :from_app
+    @packet.append_bytes "abcdefg\x00\x0C\x0A123\x00"
+    @packet.append_bytes "\x12\x6A"
+
+    io = StringIO.new
+    @packet.send_to(io)
+    io.rewind
+    assert_equal "\x41\x42\x00\x10abcdefg\x00\x0C\x0A123\x00\x12\x6A", io.read
+  end
+
+  def test_append_boolean
+    @packet.direction = :from_app
+    @packet.append_boolean true
+    @packet.append_boolean false
+    @packet.append_boolean nil
+    @packet.append_boolean 0
+    @packet.append_boolean ""
+    @packet.append_boolean "0"
+    @packet.append_boolean "nil"
+    @packet.append_boolean :nil
+    @packet.append_boolean "asdf"
+    @packet.append_boolean nil
+
+    io = StringIO.new
+    @packet.send_to(io)
+    io.rewind
+    assert_equal "\x41\x42\x00\x0A\x01\x00\x00\x01\x01\x01\x01\x01\x01\x00", 
+      io.read
+  end
+
+  def test_append_integer
+    @packet.direction = :from_app
+    @packet.append_integer 0x6789
+    @packet.append_integer 0xAB01
+    assert_raise(ArgumentError) { @packet.append_integer 0x10000 }
+    assert_raise(ArgumentError) { @packet.append_integer 0xFF0000 }
+    
+    io = StringIO.new
+    @packet.send_to(io)
+    io.rewind
+    assert_equal "\x41\x42\x00\x04\x67\x89\xAB\x01", io.read
+  end
+
+  def test_append_string
+    @packet.direction = :from_app
+    @packet.append_string "test str"
+    @packet.append_string "string including NUL(\x00) character"
+    @packet.append_string "string ending with NUL\x00"
+    @packet.append_string "more string"
+
+    io = StringIO.new
+    @packet.send_to(io)
+    io.rewind
+    assert_equal "" +
+      "\x41\x42\x00\x57" +
+      "\x00\x08test str\x00" +
+      "\x00\x21string including NUL(\x00) character\x00" +
+      "\x00\x17string ending with NUL\x00\x00" +
+      "\x00\x0Bmore string\x00",
+      io.read
+  end
+
+  def test_eof
+    data = "\x12\x34\x00\x00"
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert packet.eof?
+
+    data = "\x12\x34\x00\x03\x00\xAF\x00"
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    packet.read_integer; packet.read_byte
+    assert packet.eof?
+
+    data = "\x12\x34\x00\x03\x00\xAF\x00" +
+           "extra bytes"
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    packet.read_integer; packet.read_byte
+    assert packet.eof?
+  end
+
+
+  def test_read_byte
+    data = "\x12\x34\x00\x08\x12\x34\x56\x78\x9A\xBC\xDE\xF0" +
+           "extra bytes"
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_equal 0x12, packet.read_byte
+    assert !packet.eof?
+    assert_equal 0x34, packet.read_byte
+    assert !packet.eof?
+    assert_equal 0x56, packet.read_byte
+    assert !packet.eof?
+    assert_equal 0x78, packet.read_byte
+    assert !packet.eof?
+    assert_equal 0x9A, packet.read_byte
+    assert !packet.eof?
+    assert_equal 0xBC, packet.read_byte
+    assert !packet.eof?
+    assert_equal 0xDE, packet.read_byte
+    assert !packet.eof?
+    assert_equal 0xF0, packet.read_byte
+    assert packet.eof?
+    assert_nil packet.read_byte
+    assert_nil packet.read_byte
+    assert_nil packet.read_byte
+  end
+
+  def test_read_bytes
+    data = "\x12\x34\x00\x08\xFE\xDC\xBA\x98\x76\x54\x32\x10"
+           "extra bytes"
+    data.freeze
+
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_equal "\xFE\xDC\xBA\x98\x76\x54\x32\x10", packet.read_bytes(8)
+    assert packet.eof?
+    assert_nil packet.read_bytes(3)
+    assert_nil packet.read_byte
+    assert_nil packet.read_bytes(10)
+
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_equal "\xFE\xDC\xBA\x98\x76\x54\x32\x10", packet.read_bytes(100)
+    assert packet.eof?
+    assert_nil packet.read_bytes(3)
+    assert_nil packet.read_byte
+    assert_nil packet.read_bytes(10)
+
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_equal "\xFE\xDC\xBA", packet.read_bytes(3)
+    assert !packet.eof?
+    assert_equal "\x98\x76\x54", packet.read_bytes(3)
+    assert !packet.eof?
+    assert_equal "\x32\x10", packet.read_bytes(3)
+    assert packet.eof?
+    assert_nil packet.read_bytes(3)
+    assert_nil packet.read_byte
+    assert_nil packet.read_bytes(10)
+  end
+
+  def test_read_boolean
+    data = "\x41\x42\x00\x05\x01\x01\x00\x01\x03" + "extra bytes"
+
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_same true, packet.read_boolean
+    assert !packet.eof?
+    assert_same true, packet.read_boolean
+    assert !packet.eof?
+    assert_same false, packet.read_boolean
+    assert !packet.eof?
+    assert_same true, packet.read_boolean
+    assert !packet.eof?
+    assert_raise(Net::AJP13::AJPPacketError) { packet.read_boolean }
+    assert packet.eof?
+    assert_nil packet.read_boolean
+    assert_nil packet.read_bytes(10)
+    assert_nil packet.read_byte
+  end
+
+  def test_read_integer
+    data = "\x41\x42\x00\x0B\x00\x01\xCB\x35\x00\x00\x01\x00\xFF\xFF\x00" +
+           "extra bytes"
+    data.freeze
+
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_equal 0x0001, packet.read_integer
+    assert !packet.eof?
+    assert_equal 0xCB35, packet.read_integer
+    assert !packet.eof?
+    assert_equal 0x0000, packet.read_integer
+    assert !packet.eof?
+    assert_equal 0x0100, packet.read_integer
+    assert !packet.eof?
+    assert_equal 0xFFFF, packet.read_integer
+    assert !packet.eof?
+    assert_raise(Net::AJP13::AJPPacketError, "too short to read as integer") { 
+      packet.read_integer
+    }
+    assert !packet.eof?
+    assert_equal 0x00, packet.read_byte
+    assert packet.eof?
+    assert_nil packet.read_integer
+  end
+
+  def test_read_string
+    data = "" + 
+      "\x12\x34\x00\x25" +
+      "\x00\x0Btest string\x00" +
+      "\x00\x0Fstr with NUL(\x00)\x00" +
+      "\xFF\xFF" +                         # treated as NULL string
+      "\x00\x00\x00" +
+      "some extraa bytes"
+    data.freeze
+
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_equal "test string", packet.read_string
+    assert !packet.eof?
+    assert_equal "str with NUL(\x00)", packet.read_string
+    assert !packet.eof?
+    assert_nil packet.read_string
+    assert !packet.eof?
+    assert_equal "", packet.read_string
+    assert packet.eof?
+    assert_nil packet.read_string
+
+
+    data = "" +
+      "\x41\x42\x00\x07" +
+      "\x00\x03str\x00" +
+      "\x00" + # str length incomplete8
+      "some extra bytes"
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_equal "str", packet.read_string
+    assert_raise(Net::AJP13::AJPPacketError) { packet.read_string }
+
+    data = "" +
+      "\x12\x34\x00\x08" +
+      "\x00\x0712345\x00" + # str length too large
+      "some extra bytes"
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_raise(Net::AJP13::AJPPacketError) { packet.read_string }
+
+    data = "" +
+      "\x41\x42\x00\x0B" +
+      "\x00\x08ABCDabcd" + # missing NUL
+      "some extra bytes"
+    packet = Net::AJP13::Packet.from_io(StringIO.new(data))
+    assert !packet.eof?
+    assert_raise(Net::AJP13::AJPPacketError) { packet.read_string }
+    assert packet.eof?
+  end
+end