emonti-rbkb 0.6.2.1 → 0.6.6

data/test/test_cli_urlenc.rb

@@ -0,0 +1,44 @@
+require File.join(File.dirname(__FILE__), "test_cli_helper.rb")
+require 'rbkb/cli/urlenc'
+
+class TestCliUrlenc < Test::Unit::TestCase
+  include CliTest
+
+  def setup
+    @cli_class = Rbkb::Cli::Urlenc
+    super()
+  end
+
+  def test_basic_string_arg
+    assert_equal 0, run_with_args(["foo"])
+    assert_equal "foo\n", @stdout_io.string
+  end
+
+  def test_basic_string_arg_space_default
+    assert_equal 0, run_with_args(["fo o"])
+    assert_equal "fo%20o\n", @stdout_io.string
+  end
+
+  def test_basic_string_arg_space_with_p
+    assert_equal 0, run_with_args(["-p", "fo o"])
+    assert_equal "fo+o\n", @stdout_io.string
+  end
+
+  def test_basic_string_arg_space_with_plus
+    assert_equal 0, run_with_args(["--plus", "fo o"])
+    assert_equal "fo+o\n", @stdout_io.string
+  end
+
+  def test_basic_string_arg_space_with_noplus
+    assert_equal 0, run_with_args(["--no-plus", "fo o"])
+    assert_equal "fo%20o\n", @stdout_io.string
+  end
+
+  def test_allchars
+    allchars_dec = (0..255).map {|c| c.chr}.join
+    allchars_enc =  "%00%01%02%03%04%05%06%07%08%09%0a%0b%0c%0d%0e%0f%10%11%12%13%14%15%16%17%18%19%1a%1b%1c%1d%1e%1f%20%21%22%23%24%25%26%27%28%29%2a%2b%2c-.%2f0123456789%3a%3b%3c%3d%3e%3f%40ABCDEFGHIJKLMNOPQRSTUVWXYZ%5b%5c%5d%5e_%60abcdefghijklmnopqrstuvwxyz%7b%7c%7d~%7f%80%81%82%83%84%85%86%87%88%89%8a%8b%8c%8d%8e%8f%90%91%92%93%94%95%96%97%98%99%9a%9b%9c%9d%9e%9f%a0%a1%a2%a3%a4%a5%a6%a7%a8%a9%aa%ab%ac%ad%ae%af%b0%b1%b2%b3%b4%b5%b6%b7%b8%b9%ba%bb%bc%bd%be%bf%c0%c1%c2%c3%c4%c5%c6%c7%c8%c9%ca%cb%cc%cd%ce%cf%d0%d1%d2%d3%d4%d5%d6%d7%d8%d9%da%db%dc%dd%de%df%e0%e1%e2%e3%e4%e5%e6%e7%e8%e9%ea%eb%ec%ed%ee%ef%f0%f1%f2%f3%f4%f5%f6%f7%f8%f9%fa%fb%fc%fd%fe%ff"
+    assert_equal 0, run_with_args([allchars_dec])
+    assert_equal allchars_enc, @stdout_io.string.chomp
+  end
+
+end

data/test/test_cli_xor.rb

@@ -0,0 +1,71 @@
+require File.join(File.dirname(__FILE__), "test_cli_helper.rb")
+require 'rbkb/cli/xor'
+
+class TestCliXor < Test::Unit::TestCase
+  include CliTest
+
+  def setup
+    @cli_class = Rbkb::Cli::Xor
+    super()
+
+    @tst_dat = "my secrets are very secret"
+    @tst_key_str = "sneaky"
+    @tst_key_hex = "736e65616b79"
+    @tst_xored = "\036\027E\022\016\032\001\v\021\022K\030\001\vE\027\016\v\nN\026\004\b\v\026\032"
+  end
+
+  def test_string_arg_str_key
+    assert_equal 0, go_with_args(["-s", @tst_key_str, @tst_dat])
+    assert_equal @tst_xored, @stdout_io.string
+  end
+
+  def test_string_arg_str_key
+    assert_equal 0, go_with_args(["-x", @tst_key_hex, @tst_dat])
+    assert_equal @tst_xored, @stdout_io.string
+  end
+
+  def test_string_arg_xor_zeroes_out
+    assert_equal 0, go_with_args(%w(-s foo foo))
+    assert_equal "\x00"*3, @stdout_io.string
+  end
+
+  def test_string_arg_xor_zeroes_out_repeating
+    assert_equal 0, go_with_args(%w(-s A AAAAAAAAAA))
+    assert_equal "\x00"*10, @stdout_io.string
+  end
+
+  def test_string_arg_str_key_reverse
+    assert_equal 0, go_with_args(["-s", @tst_key_str, @tst_xored])
+    assert_equal @tst_dat, @stdout_io.string
+  end
+
+  def test_string_arg_str_key_reverse
+    assert_equal 0, go_with_args(["-x", @tst_key_hex, @tst_xored])
+    assert_equal @tst_dat, @stdout_io.string
+  end
+
+  def test_stdin
+    @stdin_io.write(@tst_dat) ; @stdin_io.rewind
+    assert_equal 0, go_with_args(["-s", @tst_key_str])
+    assert_equal @tst_xored, @stdout_io.string
+  end
+
+  def test_file_input
+    with_testfile do |fname, f|
+      f.write(@tst_dat); f.close
+      assert_equal 0, go_with_args(["-s", @tst_key_str, "-f", fname])
+      assert_equal @tst_xored, @stdout_io.string
+    end
+  end
+
+  def test_string_arg_no_key_error
+    assert_equal 1, go_with_args(%w(foo))
+    assert_match(/you must specify a key/i, @stderr_io.string)
+  end
+
+  def test_one_key_opt_error
+    assert_equal 1, go_with_args(%w(-x foo -s foo foo))
+    assert_match(/only one key option/i, @stderr_io.string)
+  end
+
+end

data/test/test_helper.rb

@@ -0,0 +1,5 @@
+require 'pp'
+require 'stringio'
+require 'test/unit'
+$:.unshift  File.dirname(__FILE__) + '/../lib'
+

data/test/test_http.rb

@@ -0,0 +1,27 @@
+require File.dirname(__FILE__) + '/test_http_helper.rb'
+
+class TestHttp < Test::Unit::TestCase
+  include Rbkb::Http
+  
+  def test_names
+    assert_equal CommonInterface, Rbkb::Http::CommonInterface
+    assert_equal Base, Rbkb::Http::Base
+
+    assert_equal Response, Rbkb::Http::Response
+    assert_equal Request, Rbkb::Http::Request
+    assert_equal Parameters, Rbkb::Http::Parameters
+    assert_equal FormUrlencodedParams, Rbkb::Http::FormUrlencodedParams
+
+    assert_equal Headers, Rbkb::Http::Headers
+    assert_equal RequestHeaders, Rbkb::Http::RequestHeaders
+    assert_equal RequestAction, Rbkb::Http::RequestAction
+    assert_equal ResponseHeaders, Rbkb::Http::ResponseHeaders
+    assert_equal ResponseStatus, Rbkb::Http::ResponseStatus
+
+    assert_equal Body, Rbkb::Http::Body
+    assert_equal BoundBody, Rbkb::Http::BoundBody
+    assert_equal ChunkedBody, Rbkb::Http::ChunkedBody
+  end
+
+end
+

data/test/test_http_helper.rb

@@ -0,0 +1,60 @@
+require File.join(File.dirname(__FILE__), 'test_helper.rb')
+require 'rbkb/http'
+
+module HttpTestHelper
+  # contains various test cases to test common interface features
+  module CommonInterfaceTests
+
+    def setup()
+      raise "Helper stub called. Override setup() in TestCase"
+    end
+
+    def do_type_tests(x)
+      raise "Helper stub called. Override do_type_tests() in TestCase"
+    end
+
+    def do_type_tests(x)
+      raise "Helper stub called. Override do_capture_value_tests() in TestCase"
+    end
+
+    def test_init_parse
+      req = @obj_klass.new(@rawdat, @obj_opts)
+      do_capture_value_tests(req)
+      do_type_tests(req)
+    end
+
+    def test_parse
+      req = @obj_klass.parse(@rawdat, @obj_opts)
+      do_capture_value_tests(req)
+      do_type_tests(req)
+    end
+
+    def test_capture
+      req = @obj.capture(@rawdat)
+      do_capture_value_tests(req)
+      do_type_tests(req)
+    end
+
+    def test_back_to_raw
+      req = @obj.capture(@rawdat)
+      assert_equal @rawdat_crlf, req.to_raw
+    end
+
+    def test_capture_and_reuse_nondestructive
+      @obj.capture(@rawdat_crlf)
+      @obj.reset_capture
+      @obj.capture(@rawdat_crlf)
+      do_capture_value_tests(@obj)
+      do_type_tests(@obj)
+    end
+
+    def test_capture_and_reuse_destructive
+      @obj.capture(@rawdat_crlf)
+      @obj.reset_capture!
+      @obj.capture(@rawdat_crlf)
+      do_capture_value_tests(@obj)
+      do_type_tests(@obj)
+    end
+
+  end
+end

data/test/test_http_request.rb

@@ -0,0 +1,136 @@
+require File.dirname(__FILE__) + '/test_http_helper.rb'
+
+class TestHttpRequest < Test::Unit::TestCase
+  include HttpTestHelper::CommonInterfaceTests
+
+  include Rbkb::Http
+
+  def setup
+    @obj_klass = Request
+    @obj = @obj_klass.new
+
+    @rawdat =<<_EOF_
+GET /csi?v=3&s=webhp&action=&tran=undefined HTTP/1.1
+Host: www.google.com
+User-Agent: Mozilla/5.0
+Accept: image/png,image/*;q=0.8,*/*;q=0.5
+Accept-Language: en-us,en;q=0.5
+Accept-Encoding: gzip,deflate
+Accept-Charset: ISO-8859-1,utf-8;q=0.7,*;q=0.7
+Keep-Alive: 300
+Proxy-Connection: keep-alive
+Referer: http://www.google.com/
+Cookie: PREFID=NNNN:LM=1237996892:S=YYYYY; NID=111
+
+_EOF_
+
+    @hstr, @body = @rawdat.split(/^\n/, 2)
+    @rawdat_crlf = @hstr.gsub("\n", "\r\n") + "\r\n" + @body
+
+    @verb = "GET"
+    @uri = "/csi?v=3&s=webhp&action=&tran=undefined"
+    @path, @query = @uri.split('?',2)
+    @req_parameters = [
+      ["v", "3"], 
+      ["s", "webhp"], 
+      ["action", ""], 
+      ["tran", "undefined"]
+    ]
+    @version = "HTTP/1.1"
+
+    @headers = [
+      ["Host", "www.google.com"], 
+      ["User-Agent", "Mozilla/5.0"],
+      ["Accept", "image/png,image/*;q=0.8,*/*;q=0.5"], 
+      ["Accept-Language", "en-us,en;q=0.5"], 
+      ["Accept-Encoding", "gzip,deflate"], 
+      ["Accept-Charset", "ISO-8859-1,utf-8;q=0.7,*;q=0.7"], 
+      ["Keep-Alive", "300"], 
+      ["Proxy-Connection", "keep-alive"], 
+      ["Referer", "http://www.google.com/"], 
+      ["Cookie", "PREFID=NNNN:LM=1237996892:S=YYYYY; NID=111"],
+    ]
+  end
+
+  def do_type_tests(req)
+    assert_kind_of Request, req
+    assert_kind_of Headers, req.headers
+    assert_kind_of Body, req.body
+    assert_kind_of RequestAction, req.action
+    assert_kind_of RequestHeaders, req.headers
+  end
+
+  def do_capture_value_tests(req)
+    assert_equal @headers, req.headers
+    assert_equal @body, req.body.to_s
+    assert_equal @uri, req.action.uri.to_s
+    assert_equal @path, req.action.uri.path
+    assert_equal @query, req.action.uri.query
+    assert_equal @verb, req.action.verb
+    assert_equal @version, req.action.version
+    assert_equal @req_parameters, req.request_parameters
+  end
+
+  def test_capture_crlf_headers
+    req = @obj.capture(@rawdat_crlf)
+    do_capture_value_tests(req)
+    do_type_tests(req)
+    assert_equal @rawdat_crlf, req.to_raw
+  end
+
+  def test_captured_body_type
+    @obj.capture(@rawdat)
+    assert_kind_of Body, @obj.body
+  end
+
+end
+
+
+class TestHttpPostRequest < TestHttpRequest
+
+  def setup
+    @obj_klass = Request
+    @obj = @obj_klass.new
+
+    @rawdat =<<_EOF_
+POST /path/script.cgi?qtype=foo HTTP/1.0
+From: frog@jmarshall.com
+HasColon: this_has_a_colon:yikes
+User-Agent: HTTPTool/1.0
+Content-Type: application/x-www-form-urlencoded
+Content-Length: 32
+
+home=Cosby&favorite+flavor=flies
+_EOF_
+    @rawdat.chomp!
+
+    @hstr, @body = @rawdat.split(/^\n/, 2)
+    @rawdat_crlf = @hstr.gsub("\n", "\r\n") + "\r\n" + @body
+
+    @verb = "POST"
+    @uri = "/path/script.cgi?qtype=foo"
+    @path, @query = @uri.split('?',2)
+    @req_parameters = [
+      ["qtype", "foo"], 
+    ]
+    @version = "HTTP/1.0"
+
+    @headers = [
+      ["From", "frog@jmarshall.com"],
+      ["HasColon", "this_has_a_colon:yikes"],
+      ["User-Agent", "HTTPTool/1.0"],
+      ["Content-Type", "application/x-www-form-urlencoded"],
+      ["Content-Length", "32"],
+    ]
+
+     @post_parameters = [["home", "Cosby"], ["favorite+flavor", "flies"]]
+  end
+
+  def test_captured_body_type
+    @obj.capture(@rawdat)
+    assert_kind_of BoundBody, @obj.body
+  end
+
+
+end
+

data/test/test_http_response.rb

@@ -0,0 +1,222 @@
+require File.dirname(__FILE__) + '/test_http_helper.rb'
+
+class TestHttpResponse < Test::Unit::TestCase
+  include HttpTestHelper::CommonInterfaceTests
+
+  include Rbkb::Http
+
+  def setup
+    @obj_klass = Response
+    @obj_opts = nil
+    @obj = @obj_klass.new(nil, @obj_opts)
+
+    @rawdat =<<_EOF_
+HTTP/1.0 404 Not Found
+Date: Thu, 26 Mar 2009 01:18:52 GMT
+Server: Apache
+Content-Type: text/html; charset=iso-8859-1
+Content-Length: 190
+Connection: Keep-Alive
+
+<html><head>
+<title>Error report</title></head>
+<body><h1>HTTP Status 404</h1><HR size="1" noshade><p><u>The requested resource is not available.</u></p><HR size="1" noshade>
+</body></html>
+_EOF_
+
+    @hstr, @body = @rawdat.split(/^\n/, 2)
+    @rawdat_crlf = @hstr.gsub("\n", "\r\n") + "\r\n" + @body
+
+    @code = 404
+    @text = "Not Found"
+    @version = "HTTP/1.0"
+
+    @headers = [
+      ["Date", "Thu, 26 Mar 2009 01:18:52 GMT"],
+      ["Server", "Apache"],
+      ["Content-Type", "text/html; charset=iso-8859-1"], 
+      ["Content-Length", "190"], 
+      ["Connection", "Keep-Alive"]
+    ]
+
+    @content_length = 190
+  end
+
+  def do_type_tests(rsp)
+    assert_kind_of Response, rsp
+    assert_kind_of Headers, rsp.headers
+    assert_kind_of Body, rsp.body
+    assert_kind_of ResponseStatus, rsp.status
+    assert_kind_of ResponseHeaders, rsp.headers
+  end
+
+  def do_capture_value_tests(rsp)
+    assert_equal @headers, rsp.headers
+    assert_equal @body, rsp.body
+    assert_equal @code, rsp.status.code
+    assert_equal @text, rsp.status.text
+    assert_equal @version, rsp.status.version
+    assert_equal @content_length, rsp.content_length
+  end
+
+  def test_capture_crlf_headers
+    @obj.capture(@rawdat_crlf)
+    do_capture_value_tests(@obj)
+    do_type_tests(@obj)
+    assert_equal @rawdat_crlf, @obj.to_raw
+  end
+
+  def test_captured_body_type
+    @obj.capture(@rawdat)
+    assert_kind_of BoundBody, @obj.body
+  end
+end
+
+
+class TestHttpResponseChunked < TestHttpResponse
+  include Rbkb::Http
+
+  def setup
+    @obj_klass = Response
+    @obj_opts = {}
+    @obj = @obj_klass.new
+
+    @rawdat =<<_EOF_
+HTTP/1.1 200 OK
+Date: Thu, 26 Mar 2009 01:18:52 GMT
+Server: Apache
+Content-Type: text/html; charset=iso-8859-1
+Transfer-Encoding: chunked
+Connection: Keep-Alive
+
+20
+This is a test of a chunked-enco
+
+10
+ded HTTP request
+
+0
+_EOF_
+
+    @hstr, @rawbody = @rawdat.split(/^\n/, 2)
+    @rawdat_crlf = @rawdat.gsub("\n", "\r\n")
+    @hdrs_crlf = @hstr.gsub("\n", "\r\n")
+
+    @body = "This is a test of a chunked-encoded HTTP request"
+
+    @code = 200
+    @text = "OK"
+    @version = "HTTP/1.1"
+
+    @headers = [
+      ["Date", "Thu, 26 Mar 2009 01:18:52 GMT"],
+      ["Server", "Apache"],
+      ["Content-Type", "text/html; charset=iso-8859-1"], 
+      ["Transfer-Encoding", "chunked"], 
+      ["Connection", "Keep-Alive"]
+    ]
+
+    @content_length = nil
+    @tc_chunk_size = 0x20
+  end
+
+  def test_captured_body_type
+    @obj.capture(@rawdat)
+    assert_kind_of ChunkedBody, @obj.body
+  end
+
+  def test_back_to_raw
+    @obj.capture(@rawdat)
+    do_capture_value_tests(@obj)
+    do_type_tests(@obj)
+    @obj.body.opts[:output_chunk_size] = @tc_chunk_size
+    assert_equal @rawdat_crlf, @obj.to_raw
+  end
+
+  def test_capture_crlf_headers
+    @obj.capture(@rawdat_crlf)
+    do_capture_value_tests(@obj)
+    do_type_tests(@obj)
+    @obj.body.opts[:output_chunk_size] = @tc_chunk_size
+    assert_equal @rawdat_crlf, @obj.to_raw
+  end
+
+  def test_default_chunk_size
+    if @body.size > ChunkedBody::DEFAULT_CHUNK_SIZE
+      assert "TESTCASE ERROR!!!", "make the setup() @body < DEFAULT_CHUNK_SIZE"
+    end
+    raw_tc = "#{@hdrs_crlf}\r\n#{@body.size.to_s(16)}\r\n#{@body}\r\n\r\n0\r\n"
+    @obj.capture(@rawdat_crlf)
+    do_capture_value_tests(@obj)
+    do_type_tests(@obj)
+    assert_equal raw_tc, @obj.to_raw
+  end
+end
+
+# This test-case simulates a HTTP response to a HEAD request. This type of
+# response is special since it returns Content-Length: NNN or 
+# Transfer-Encoding: chunked headers without any actual body data. 
+# To handle this special situation, we use the 'ignore_content_length' and 
+# 'ignore_chunked_encoding' options.
+class TestHttpResponseToHead < TestHttpResponse
+  def setup
+    @obj_klass = Response
+
+    # Technically, a server should only respond to HEAD with one of 
+    # content length *or* chunked encoding. However, we ignore them both.
+    @obj_opts = {
+      :ignore_content_length => true, 
+      :ignore_chunked_encoding => true
+    }
+    @obj = @obj_klass.new(nil, @obj_opts)
+
+    # Note, our test-case includes both content length and chunked encoding.
+    # A real server probably wouldn't do this, but we want to make sure
+    # we handle both.
+    @rawdat =<<_EOF_
+HTTP/1.1 200 OK
+Cache-Control: private, max-age=0
+Date: Fri, 27 Mar 2009 04:27:27 GMT
+Expires: -1
+Content-Length: 9140
+Content-Type: text/html; charset=ISO-8859-1
+Server: Booble
+Transfer-Encoding: chunked
+
+_EOF_
+
+    @hstr, @body = @rawdat.split(/^\n/, 2)
+    @rawdat_crlf = @hstr.gsub("\n", "\r\n") + "\r\n" + @body
+
+    @code = 200
+    @text = "OK"
+    @version = "HTTP/1.1"
+
+    @headers = [
+      ["Cache-Control", "private, max-age=0"], 
+      ["Date", "Fri, 27 Mar 2009 04:27:27 GMT"], 
+      ["Expires", "-1"], 
+      ["Content-Length", "9140"], 
+      ["Content-Type", "text/html; charset=ISO-8859-1"], 
+      ["Server", "Booble"], 
+      ["Transfer-Encoding", "chunked"]
+    ]
+
+    # Content length should get ignored
+    @content_length = nil
+  end
+
+  def test_capture_crlf_headers
+    @obj.capture(@rawdat_crlf)
+    do_capture_value_tests(@obj)
+    do_type_tests(@obj)
+    assert_equal @rawdat_crlf, @obj.to_raw
+  end
+
+  def test_captured_body_type
+    @obj.capture(@rawdat)
+    assert_kind_of Body, @obj.body
+  end
+
+end
+