rbkb 0.6.10

data/test/test_cli_telson.rb

@@ -0,0 +1,11 @@
+require File.join(File.dirname(__FILE__), "test_cli_helper.rb")
+
+class TestCliTelson < Test::Unit::TestCase
+  include CliTest
+
+  def setup
+    @cli_class = Rbkb::Cli::Telson
+    super()
+  end
+
+end

data/test/test_cli_unhexify.rb

@@ -0,0 +1,43 @@
+require File.join(File.dirname(__FILE__), "test_cli_helper.rb")
+require 'rbkb/cli/unhexify'
+
+class TestCliUnhexify < Test::Unit::TestCase
+  include CliTest
+
+  def setup
+    @cli_class = Rbkb::Cli::Unhexify
+    super()
+  end
+
+  def test_string_arg
+    assert_equal 0, go_with_args(%w(666f6f))
+    assert_equal("foo", @stdout_io.string)
+  end
+
+  def test_string_arg_with_delim_arg
+    assert_equal 0, go_with_args(%w(-d : 66:6f:6f))
+    assert_equal("foo", @stdout_io.string)
+  end
+
+  def test_stdin
+    @stdin_io.write("666f6f") ; @stdin_io.rewind
+    assert_equal 0, go_with_args
+    assert_equal("foo", @stdout_io.string)
+  end
+
+  def test_stdin_with_delim_default_allchars
+    @stdin_io.write((0..255).map {|x| x.to_s(16)}.join(' ')); @stdin_io.rewind
+    assert_equal 0, go_with_args
+    assert_equal((0..255).map {|x| x.chr}.join, @stdout_io.string)
+  end
+
+
+  def test_file_input
+    with_testfile do |fname, tf| 
+      tf.write "6865785f746573745f666f6f";  tf.close
+      assert_equal 0, go_with_args(["-f", fname])
+      assert_equal("hex_test_foo", @stdout_io.string)
+    end
+  end
+
+end

data/test/test_cli_urldec.rb

@@ -0,0 +1,50 @@
+require File.join(File.dirname(__FILE__), "test_cli_helper.rb")
+require 'rbkb/cli/urldec'
+
+class TestCliUrldec < Test::Unit::TestCase
+  include CliTest
+
+  def setup
+    @cli_class = Rbkb::Cli::Urldec
+    super()
+  end
+
+  def test_basic_string_arg
+    assert_equal 0, run_with_args(["foo"])
+    assert_equal "foo", @stdout_io.string
+  end
+
+  def test_basic_string_arg_with_space
+    assert_equal 0, run_with_args(["f%20oo"])
+    assert_equal "f oo", @stdout_io.string
+  end
+
+  def test_basic_string_arg_with_space_plus
+    assert_equal 0, run_with_args(["f+oo"])
+    assert_equal "f oo", @stdout_io.string
+  end
+
+  def test_basic_string_arg_with_space_plus_p_arg
+    assert_equal 0, run_with_args(["-p", "f+oo"])
+    assert_equal "f oo", @stdout_io.string
+  end
+
+  def test_basic_string_arg_with_space_plus_plus_arg
+    assert_equal 0, run_with_args(["--plus", "f+oo"])
+    assert_equal "f oo", @stdout_io.string
+  end
+
+  def test_basic_string_arg_with_space_plus_noplus_arg
+    assert_equal 0, run_with_args(["--no-plus", "f+oo"])
+    assert_equal "f+oo", @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_enc])
+    assert_equal allchars_dec, @stdout_io.string
+  end
+
+
+end

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
+