require 'test/unit' require 'cgi' require 'stringio' require_relative 'update_env' class CGICoreTest < Test::Unit::TestCase include UpdateEnv def setup @environ = {} #@environ = { # 'SERVER_PROTOCOL' => 'HTTP/1.1', # 'REQUEST_METHOD' => 'GET', # 'SERVER_SOFTWARE' => 'Apache 2.2.0', #} #ENV.update(@environ) end def teardown ENV.update(@environ) $stdout = STDOUT end def test_cgi_parse_illegal_query update_env( 'REQUEST_METHOD' => 'GET', 'QUERY_STRING' => 'a=111&&b=222&c&d=', 'HTTP_COOKIE' => '_session_id=12345; name1=val1&val2;', 'SERVER_SOFTWARE' => 'Apache 2.2.0', 'SERVER_PROTOCOL' => 'HTTP/1.1', ) cgi = CGI.new assert_equal(["a","b","c","d"],cgi.keys.sort) assert_equal("",cgi["d"]) end def test_cgi_core_params_GET update_env( 'REQUEST_METHOD' => 'GET', 'QUERY_STRING' => 'id=123&id=456&id=&id&str=%40h+%3D%7E+%2F%5E%24%2F', 'HTTP_COOKIE' => '_session_id=12345; name1=val1&val2;', 'SERVER_SOFTWARE' => 'Apache 2.2.0', 'SERVER_PROTOCOL' => 'HTTP/1.1', ) cgi = CGI.new ## cgi[] assert_equal('123', cgi['id']) assert_equal('@h =~ /^$/', cgi['str']) ## cgi.params assert_equal(['123', '456', ''], cgi.params['id']) assert_equal(['@h =~ /^$/'], cgi.params['str']) ## cgi.keys assert_equal(['id', 'str'], cgi.keys.sort) ## cgi.key?, cgi.has_key?, cgi.include? assert_equal(true, cgi.key?('id')) assert_equal(true, cgi.has_key?('id')) assert_equal(true, cgi.include?('id')) assert_equal(false, cgi.key?('foo')) assert_equal(false, cgi.has_key?('foo')) assert_equal(false, cgi.include?('foo')) ## invalid parameter name assert_equal('', cgi['*notfound*']) # [ruby-dev:30740] assert_equal([], cgi.params['*notfound*']) end def test_cgi_core_params_POST query_str = 'id=123&id=456&id=&str=%40h+%3D%7E+%2F%5E%24%2F' update_env( 'REQUEST_METHOD' => 'POST', 'CONTENT_LENGTH' => query_str.length.to_s, 'HTTP_COOKIE' => '_session_id=12345; name1=val1&val2;', 'SERVER_SOFTWARE' => 'Apache 2.2.0', 'SERVER_PROTOCOL' => 'HTTP/1.1', ) $stdin = StringIO.new $stdin << query_str $stdin.rewind cgi = CGI.new ## cgi[] assert_equal('123', cgi['id']) assert_equal('@h =~ /^$/', cgi['str']) ## cgi.params assert_equal(['123', '456', ''], cgi.params['id']) assert_equal(['@h =~ /^$/'], cgi.params['str']) ## invalid parameter name assert_equal('', cgi['*notfound*']) assert_equal([], cgi.params['*notfound*']) end def test_cgi_core_params_encoding_check query_str = 'str=%BE%BE%B9%BE' update_env( 'REQUEST_METHOD' => 'POST', 'CONTENT_LENGTH' => query_str.length.to_s, 'SERVER_SOFTWARE' => 'Apache 2.2.0', 'SERVER_PROTOCOL' => 'HTTP/1.1', ) $stdin = StringIO.new $stdin << query_str $stdin.rewind if defined?(::Encoding) hash={} cgi = CGI.new(:accept_charset=>"UTF-8"){|key,val|hash[key]=val} ## cgi[] assert_equal("\xBE\xBE\xB9\xBE".force_encoding("UTF-8"), cgi['str']) ## cgi.params assert_equal(["\xBE\xBE\xB9\xBE".force_encoding("UTF-8")], cgi.params['str']) ## accept-charset error assert_equal({"str"=>"\xBE\xBE\xB9\xBE".force_encoding("UTF-8")},hash) $stdin.rewind assert_raise(CGI::InvalidEncoding) do cgi = CGI.new(:accept_charset=>"UTF-8") end $stdin.rewind cgi = CGI.new(:accept_charset=>"EUC-JP") ## cgi[] assert_equal("\xBE\xBE\xB9\xBE".force_encoding("EUC-JP"), cgi['str']) ## cgi.params assert_equal(["\xBE\xBE\xB9\xBE".force_encoding("EUC-JP")], cgi.params['str']) else assert(true) end end def test_cgi_core_cookie update_env( 'REQUEST_METHOD' => 'GET', 'QUERY_STRING' => 'id=123&id=456&id=&str=%40h+%3D%7E+%2F%5E%24%2F', 'HTTP_COOKIE' => '_session_id=12345; name1=val1&val2;', 'SERVER_SOFTWARE' => 'Apache 2.2.0', 'SERVER_PROTOCOL' => 'HTTP/1.1', ) cgi = CGI.new assert_not_equal(nil,cgi.cookies) [ ['_session_id', ['12345'], ], ['name1', ['val1', 'val2'], ], ].each do |key, expected| cookie = cgi.cookies[key] assert_kind_of(CGI::Cookie, cookie) assert_equal(expected, cookie.value) assert_equal(false, cookie.secure) assert_nil(cookie.expires) assert_nil(cookie.domain) assert_equal('', cookie.path) end end def test_cgi_core_maxcontentlength update_env( 'REQUEST_METHOD' => 'POST', 'CONTENT_LENGTH' => (64 * 1024 * 1024).to_s ) ex = assert_raise(StandardError) do CGI.new end assert_equal("too large post data.", ex.message) end if CGI.const_defined?(:MAX_CONTENT_LENGTH) def test_cgi_core_out update_env( 'REQUEST_METHOD' => 'GET', 'QUERY_STRING' => 'id=123&id=456&id=&str=%40h+%3D%7E+%2F%5E%24%2F', 'HTTP_COOKIE' => '_session_id=12345; name1=val1&val2;', 'SERVER_SOFTWARE' => 'Apache 2.2.0', 'SERVER_PROTOCOL' => 'HTTP/1.1', ) cgi = CGI.new ## euc string euc_str = "\270\253\244\355\241\242\277\315\244\254\245\264\245\337\244\316\244\350\244\246\244\300" ## utf8 (not converted) options = { 'charset'=>'utf8' } $stdout = StringIO.new cgi.out(options) { euc_str } assert_nil(options['language']) actual = $stdout.string expected = "Content-Type: text/html; charset=utf8\r\n" + "Content-Length: 22\r\n" + "\r\n" + euc_str if defined?(::Encoding) actual.force_encoding("ASCII-8BIT") expected.force_encoding("ASCII-8BIT") end assert_equal(expected, actual) ## language is keeped options = { 'charset'=>'Shift_JIS', 'language'=>'en' } $stdout = StringIO.new cgi.out(options) { euc_str } assert_equal('en', options['language']) ## HEAD method update_env('REQUEST_METHOD' => 'HEAD') options = { 'charset'=>'utf8' } $stdout = StringIO.new cgi.out(options) { euc_str } actual = $stdout.string expected = "Content-Type: text/html; charset=utf8\r\n" + "Content-Length: 22\r\n" + "\r\n" assert_equal(expected, actual) end def test_cgi_core_print update_env( 'REQUEST_METHOD' => 'GET', ) cgi = CGI.new $stdout = StringIO.new str = "foobar" cgi.print(str) expected = str actual = $stdout.string assert_equal(expected, actual) end def test_cgi_core_environs update_env( 'REQUEST_METHOD' => 'GET', ) cgi = CGI.new ## list1 = %w[ AUTH_TYPE CONTENT_TYPE GATEWAY_INTERFACE PATH_INFO PATH_TRANSLATED QUERY_STRING REMOTE_ADDR REMOTE_HOST REMOTE_IDENT REMOTE_USER REQUEST_METHOD SCRIPT_NAME SERVER_NAME SERVER_PROTOCOL SERVER_SOFTWARE HTTP_ACCEPT HTTP_ACCEPT_CHARSET HTTP_ACCEPT_ENCODING HTTP_ACCEPT_LANGUAGE HTTP_CACHE_CONTROL HTTP_FROM HTTP_HOST HTTP_NEGOTIATE HTTP_PRAGMA HTTP_REFERER HTTP_USER_AGENT ] # list2 = %w[ CONTENT_LENGTH SERVER_PORT ] ## string expected list1.each do |name| update_env(name => "**#{name}**") end list1.each do |name| method = name.sub(/\AHTTP_/, '').downcase actual = cgi.__send__ method expected = "**#{name}**" assert_equal(expected, actual) end ## integer expected update_env('CONTENT_LENGTH' => '123') update_env('SERVER_PORT' => '8080') assert_equal(123, cgi.content_length) assert_equal(8080, cgi.server_port) ## raw cookie update_env('HTTP_COOKIE' => 'name1=val1') update_env('HTTP_COOKIE2' => 'name2=val2') assert_equal('name1=val1', cgi.raw_cookie) assert_equal('name2=val2', cgi.raw_cookie2) end def test_cgi_core_htmltype_header update_env( 'REQUEST_METHOD' => 'GET', ) ## no htmltype cgi = CGI.new assert_raise(NoMethodError) do cgi.doctype end assert_equal("Content-Type: text/html\r\n\r\n",cgi.header) ## html3 cgi = CGI.new('html3') expected = '' assert_equal(expected, cgi.doctype) assert_equal("Content-Type: text/html\r\n\r\n",cgi.header) ## html4 cgi = CGI.new('html4') expected = '' assert_equal(expected, cgi.doctype) assert_equal("Content-Type: text/html\r\n\r\n",cgi.header) ## html4 transitional cgi = CGI.new('html4Tr') expected = '' assert_equal(expected, cgi.doctype) assert_equal("Content-Type: text/html\r\n\r\n",cgi.header) ## html4 frameset cgi = CGI.new('html4Fr') expected = '' assert_equal(expected, cgi.doctype) assert_equal("Content-Type: text/html\r\n\r\n",cgi.header) ## html5 cgi = CGI.new('html5') expected = '' assert_equal(expected, cgi.doctype) assert_match(/^
<\/HEADER>$/i,cgi.header) end instance_methods.each do |method| private method if method =~ /^test_(.*)/ && $1 != ENV['TEST'] end if ENV['TEST'] end