test_kvform.rb
165 lines
| 4.2 KiB
| text/x-ruby
|
RubyLexer
Eric Davis
|
r2376 | require 'test/unit' | ||
| require 'openid/kvform' | ||||
| require 'openid/util' | ||||
| require 'util' | ||||
| include OpenID | ||||
| class KVFormTests < Test::Unit::TestCase | ||||
| include OpenID::TestUtil | ||||
| def test_kvdict | ||||
| [ | ||||
| # (kvform, parsed dictionary, expected warnings) | ||||
| ["", {}, 0], | ||||
| ["\n \n \n", {}, 0], | ||||
| ["college:harvey mudd\n", {"college" => "harvey mudd"}, 0], | ||||
| ["city:claremont\nstate:CA\n", | ||||
| {"city" => "claremont", "state" => "CA"}, 0], | ||||
| ["is_valid:true\ninvalidate_handle:{HMAC-SHA1:2398410938412093}\n", | ||||
| {"is_valid" => "true", | ||||
| "invalidate_handle" => "{HMAC-SHA1:2398410938412093}"}, 0], | ||||
| # Warnings from lines with no colon: | ||||
| ["x\n", {}, 1], | ||||
| ["x\nx\n", {}, 2], | ||||
| ["East is least\n", {}, 1], | ||||
| # But not from blank lines (because LJ generates them) | ||||
| ["x\n\n", {}, 1], | ||||
| # Warning from empty key | ||||
| [":\n", {''=>''}, 1], | ||||
| [":missing key\n", {''=>'missing key'}, 1], | ||||
| # Warnings from leading or trailing whitespace in key or value | ||||
| [" street:foothill blvd\n", {"street"=>"foothill blvd"}, 1], | ||||
| ["major: computer science\n", {"major"=>"computer science"}, 1], | ||||
| [" dorm : east \n", {"dorm"=>"east"}, 2], | ||||
| # Warnings from missing trailing newline | ||||
| ["e^(i*pi)+1:0", {"e^(i*pi)+1" => "0"}, 1], | ||||
| ["east:west\nnorth:south", {"east"=>"west", "north"=>"south"}, 1], | ||||
| ].each { |case_| | ||||
| _run_kvdictTest(case_) | ||||
| } | ||||
| end | ||||
| def _run_kvdictTest(case_) | ||||
| kv, dct, warnings = case_ | ||||
| d = nil | ||||
| d2 = nil | ||||
| assert_log_line_count(warnings) { | ||||
| # Convert KVForm to dict | ||||
| d = Util.kv_to_dict(kv) | ||||
| # Strict mode should raise KVFormError instead of logging | ||||
| # messages | ||||
| if warnings > 0 | ||||
| assert_raise(KVFormError) do | ||||
| Util.kv_to_seq(kv, true) | ||||
| end | ||||
| end | ||||
| # make sure it parses to expected dict | ||||
| assert_equal(dct, d) | ||||
| } | ||||
| # Convert back to KVForm and round-trip back to dict to make sure | ||||
| # that *** dict -> kv -> dict is identity. *** | ||||
| kv = Util.dict_to_kv(d) | ||||
| silence_logging { | ||||
| d2 = Util.kv_to_dict(kv) | ||||
| } | ||||
| assert_equal(d, d2) | ||||
| end | ||||
| def test_kvseq | ||||
| [ | ||||
| [[], "", 0], | ||||
| [[["openid", "useful"], ["a", "b"]], "openid:useful\na:b\n", 0], | ||||
| # Warnings about leading whitespace | ||||
| [[[" openid", "useful"], ["a", "b"]], " openid:useful\na:b\n", 2], | ||||
| # Warnings about leading and trailing whitespace | ||||
| [[[" openid ", " useful "], | ||||
| [" a ", " b "]], " openid : useful \n a : b \n", 8], | ||||
| # warnings about leading and trailing whitespace, but not about | ||||
| # internal whitespace. | ||||
| [[[" open id ", " use ful "], | ||||
| [" a ", " b "]], " open id : use ful \n a : b \n", 8], | ||||
| [[["foo", "bar"]], "foo:bar\n", 0], | ||||
| ].each { |case_| | ||||
| _run_kvseqTest(case_) | ||||
| } | ||||
| end | ||||
| def _cleanSeq(seq) | ||||
| # Create a new sequence by stripping whitespace from start and end | ||||
| # of each value of each pair | ||||
| seq.collect { |k, v| [k.strip(), v.strip()] } | ||||
| end | ||||
| def _run_kvseqTest(case_) | ||||
| seq, kvform, warnings = case_ | ||||
| assert_log_line_count(warnings) { | ||||
| # seq serializes to expected kvform | ||||
| actual = Util.seq_to_kv(seq) | ||||
| assert_equal(kvform, actual) | ||||
| assert actual.is_a?(String) | ||||
| # Strict mode should raise KVFormError instead of logging | ||||
| # messages | ||||
| if warnings > 0 | ||||
| assert_raise(KVFormError) do | ||||
| Util.seq_to_kv(seq, true) | ||||
| end | ||||
| end | ||||
| # Parse back to sequence. Expected to be unchanged, except | ||||
| # stripping whitespace from start and end of values | ||||
| # (i. e. ordering, case, and internal whitespace is preserved) | ||||
| seq = Util.kv_to_seq(actual) | ||||
| clean_seq = _cleanSeq(seq) | ||||
| assert_equal(seq, clean_seq) | ||||
| } | ||||
| end | ||||
| def test_kvexc | ||||
| [ | ||||
| [["openid", "use\nful"]], | ||||
| [["open\nid", "useful"]], | ||||
| [["open\nid", "use\nful"]], | ||||
| [["open:id", "useful"]], | ||||
| [["foo", "bar"], ["ba\n d", "seed"]], | ||||
| [["foo", "bar"], ["bad:", "seed"]], | ||||
| ].each { |case_| | ||||
| _run_kvexcTest(case_) | ||||
| } | ||||
| end | ||||
| def _run_kvexcTest(case_) | ||||
| seq = case_ | ||||
| assert_raise(KVFormError) do | ||||
| Util.seq_to_kv(seq) | ||||
| end | ||||
| end | ||||
| def test_convert | ||||
| assert_log_line_count(2) { | ||||
| result = Util.seq_to_kv([[1, 1]]) | ||||
| assert_equal(result, "1:1\n") | ||||
| } | ||||
| end | ||||
| end | ||||