239 lines
7.9 KiB
Ruby
239 lines
7.9 KiB
Ruby
# frozen_string_literal: true
|
|
require 'test/unit'
|
|
require 'cgi'
|
|
require 'stringio'
|
|
require_relative 'update_env'
|
|
|
|
|
|
class CGIUtilTest < Test::Unit::TestCase
|
|
include CGI::Util
|
|
include UpdateEnv
|
|
|
|
def setup
|
|
@environ = {}
|
|
update_env(
|
|
'REQUEST_METHOD' => 'GET',
|
|
'SCRIPT_NAME' => nil,
|
|
)
|
|
@str1="&<>\" \xE3\x82\x86\xE3\x82\x93\xE3\x82\x86\xE3\x82\x93".dup
|
|
@str1.force_encoding("UTF-8") if defined?(::Encoding)
|
|
end
|
|
|
|
def teardown
|
|
ENV.update(@environ)
|
|
end
|
|
|
|
|
|
def test_cgi_escape
|
|
assert_equal('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93', CGI.escape(@str1))
|
|
assert_equal('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93'.ascii_only?, CGI.escape(@str1).ascii_only?) if defined?(::Encoding)
|
|
end
|
|
|
|
def test_cgi_escape_with_unreserved_characters
|
|
assert_equal("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-._~",
|
|
CGI.escape("ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789-._~"),
|
|
"should not escape any unreserved characters, as per RFC3986 Section 2.3")
|
|
end
|
|
|
|
def test_cgi_escape_with_invalid_byte_sequence
|
|
assert_equal('%C0%3C%3C', CGI.escape("\xC0\<\<".dup.force_encoding("UTF-8")))
|
|
end
|
|
|
|
def test_cgi_escape_preserve_encoding
|
|
assert_equal(Encoding::US_ASCII, CGI.escape("\xC0\<\<".dup.force_encoding("US-ASCII")).encoding)
|
|
assert_equal(Encoding::ASCII_8BIT, CGI.escape("\xC0\<\<".dup.force_encoding("ASCII-8BIT")).encoding)
|
|
assert_equal(Encoding::UTF_8, CGI.escape("\xC0\<\<".dup.force_encoding("UTF-8")).encoding)
|
|
end
|
|
|
|
def test_cgi_unescape
|
|
str = CGI.unescape('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93')
|
|
assert_equal(@str1, str)
|
|
return unless defined?(::Encoding)
|
|
|
|
assert_equal(@str1.encoding, str.encoding)
|
|
assert_equal("\u{30E1 30E2 30EA 691C 7D22}", CGI.unescape("\u{30E1 30E2 30EA}%E6%A4%9C%E7%B4%A2"))
|
|
end
|
|
|
|
def test_cgi_unescape_preserve_encoding
|
|
assert_equal(Encoding::US_ASCII, CGI.unescape("%C0%3C%3C".dup.force_encoding("US-ASCII")).encoding)
|
|
assert_equal(Encoding::ASCII_8BIT, CGI.unescape("%C0%3C%3C".dup.force_encoding("ASCII-8BIT")).encoding)
|
|
assert_equal(Encoding::UTF_8, CGI.unescape("%C0%3C%3C".dup.force_encoding("UTF-8")).encoding)
|
|
end
|
|
|
|
def test_cgi_unescape_accept_charset
|
|
return unless defined?(::Encoding)
|
|
|
|
assert_raise(TypeError) {CGI.unescape('', nil)}
|
|
assert_separately(%w[-rcgi/util], "#{<<-"begin;"}\n#{<<-"end;"}")
|
|
begin;
|
|
assert_equal("", CGI.unescape(''))
|
|
end;
|
|
end
|
|
|
|
def test_cgi_pretty
|
|
assert_equal("<HTML>\n <BODY>\n </BODY>\n</HTML>\n",CGI.pretty("<HTML><BODY></BODY></HTML>"))
|
|
assert_equal("<HTML>\n\t<BODY>\n\t</BODY>\n</HTML>\n",CGI.pretty("<HTML><BODY></BODY></HTML>","\t"))
|
|
end
|
|
|
|
def test_cgi_escapeHTML
|
|
assert_equal("'&"><", CGI.escapeHTML("'&\"><"))
|
|
end
|
|
|
|
def test_cgi_escape_html_duplicated
|
|
orig = "Ruby".dup.force_encoding("US-ASCII")
|
|
str = CGI.escapeHTML(orig)
|
|
assert_equal(orig, str)
|
|
assert_not_same(orig, str)
|
|
end
|
|
|
|
def assert_cgi_escape_html_preserve_encoding(str, encoding)
|
|
assert_equal(encoding, CGI.escapeHTML(str.dup.force_encoding(encoding)).encoding)
|
|
end
|
|
|
|
def test_cgi_escape_html_preserve_encoding
|
|
Encoding.list do |enc|
|
|
assert_cgi_escape_html_preserve_encoding("'&\"><", enc)
|
|
assert_cgi_escape_html_preserve_encoding("Ruby", enc)
|
|
end
|
|
end
|
|
|
|
def test_cgi_escape_html_dont_freeze
|
|
assert_not_predicate CGI.escapeHTML("'&\"><".dup), :frozen?
|
|
assert_not_predicate CGI.escapeHTML("'&\"><".freeze), :frozen?
|
|
assert_not_predicate CGI.escapeHTML("Ruby".dup), :frozen?
|
|
assert_not_predicate CGI.escapeHTML("Ruby".freeze), :frozen?
|
|
end
|
|
|
|
def test_cgi_escape_html_large
|
|
return if RUBY_ENGINE == 'jruby'
|
|
ulong_max, size_max = RbConfig::LIMITS.values_at("ULONG_MAX", "SIZE_MAX")
|
|
return unless ulong_max < size_max # Platforms not concerned
|
|
|
|
size = (ulong_max / 6 + 1)
|
|
begin
|
|
str = '"' * size
|
|
escaped = CGI.escapeHTML(str)
|
|
rescue NoMemoryError
|
|
omit "Not enough memory"
|
|
rescue => e
|
|
end
|
|
assert_raise_with_message(ArgumentError, /overflow/, ->{"length = #{escaped.length}"}) do
|
|
raise e if e
|
|
end
|
|
end
|
|
|
|
def test_cgi_unescapeHTML
|
|
assert_equal("'&\"><", CGI.unescapeHTML("'&"><"))
|
|
end
|
|
|
|
def test_cgi_unescapeHTML_invalid
|
|
assert_equal('&<&>"&abcdefghijklmn', CGI.unescapeHTML('&<&>"&abcdefghijklmn'))
|
|
end
|
|
|
|
Encoding.list.each do |enc|
|
|
begin
|
|
escaped = "'&"><".encode(enc)
|
|
unescaped = "'&\"><".encode(enc)
|
|
rescue Encoding::ConverterNotFoundError
|
|
next
|
|
else
|
|
define_method("test_cgi_escapeHTML:#{enc.name}") do
|
|
assert_equal(escaped, CGI.escapeHTML(unescaped))
|
|
end
|
|
define_method("test_cgi_unescapeHTML:#{enc.name}") do
|
|
assert_equal(unescaped, CGI.unescapeHTML(escaped))
|
|
end
|
|
end
|
|
end
|
|
|
|
Encoding.list.each do |enc|
|
|
next unless enc.ascii_compatible?
|
|
begin
|
|
escaped = "%25+%2B"
|
|
unescaped = "% +".encode(enc)
|
|
rescue Encoding::ConverterNotFoundError
|
|
next
|
|
else
|
|
define_method("test_cgi_escape:#{enc.name}") do
|
|
assert_equal(escaped, CGI.escape(unescaped))
|
|
end
|
|
define_method("test_cgi_unescape:#{enc.name}") do
|
|
assert_equal(unescaped, CGI.unescape(escaped, enc))
|
|
end
|
|
end
|
|
end
|
|
|
|
def test_cgi_unescapeHTML_uppercasecharacter
|
|
assert_equal("\xE3\x81\x82\xE3\x81\x84\xE3\x81\x86", CGI.unescapeHTML("あいう"))
|
|
end
|
|
|
|
def test_cgi_include_escape
|
|
assert_equal('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93', escape(@str1))
|
|
end
|
|
|
|
def test_cgi_include_escapeHTML
|
|
assert_equal("'&"><", escapeHTML("'&\"><"))
|
|
end
|
|
|
|
def test_cgi_include_h
|
|
assert_equal("'&"><", h("'&\"><"))
|
|
end
|
|
|
|
def test_cgi_include_unescape
|
|
str = unescape('%26%3C%3E%22+%E3%82%86%E3%82%93%E3%82%86%E3%82%93')
|
|
assert_equal(@str1, str)
|
|
return unless defined?(::Encoding)
|
|
|
|
assert_equal(@str1.encoding, str.encoding)
|
|
assert_equal("\u{30E1 30E2 30EA 691C 7D22}", unescape("\u{30E1 30E2 30EA}%E6%A4%9C%E7%B4%A2"))
|
|
end
|
|
|
|
def test_cgi_include_unescapeHTML
|
|
assert_equal("'&\"><", unescapeHTML("'&"><"))
|
|
end
|
|
|
|
def test_cgi_escapeElement
|
|
assert_equal("<BR><A HREF="url"></A>", escapeElement('<BR><A HREF="url"></A>', "A", "IMG"))
|
|
assert_equal("<BR><A HREF="url"></A>", escapeElement('<BR><A HREF="url"></A>', ["A", "IMG"]))
|
|
assert_equal("<BR><A HREF="url"></A>", escape_element('<BR><A HREF="url"></A>', "A", "IMG"))
|
|
assert_equal("<BR><A HREF="url"></A>", escape_element('<BR><A HREF="url"></A>', ["A", "IMG"]))
|
|
end
|
|
|
|
|
|
def test_cgi_unescapeElement
|
|
assert_equal('<BR><A HREF="url"></A>', unescapeElement(escapeHTML('<BR><A HREF="url"></A>'), "A", "IMG"))
|
|
assert_equal('<BR><A HREF="url"></A>', unescapeElement(escapeHTML('<BR><A HREF="url"></A>'), ["A", "IMG"]))
|
|
assert_equal('<BR><A HREF="url"></A>', unescape_element(escapeHTML('<BR><A HREF="url"></A>'), "A", "IMG"))
|
|
assert_equal('<BR><A HREF="url"></A>', unescape_element(escapeHTML('<BR><A HREF="url"></A>'), ["A", "IMG"]))
|
|
end
|
|
end
|
|
|
|
class CGIUtilPureRubyTest < Test::Unit::TestCase
|
|
def setup
|
|
CGI::Escape.module_eval do
|
|
alias _escapeHTML escapeHTML
|
|
remove_method :escapeHTML
|
|
alias _unescapeHTML unescapeHTML
|
|
remove_method :unescapeHTML
|
|
end
|
|
end
|
|
|
|
def teardown
|
|
CGI::Escape.module_eval do
|
|
alias escapeHTML _escapeHTML
|
|
remove_method :_escapeHTML
|
|
alias unescapeHTML _unescapeHTML
|
|
remove_method :_unescapeHTML
|
|
end
|
|
end
|
|
|
|
def test_cgi_escapeHTML_with_invalid_byte_sequence
|
|
assert_equal("<\xA4??>", CGI.escapeHTML(%[<\xA4??>]))
|
|
end
|
|
|
|
def test_cgi_unescapeHTML_with_invalid_byte_sequence
|
|
input = "\xFF&"
|
|
assert_equal(input, CGI.unescapeHTML(input))
|
|
end
|
|
end
|