added tag v1_9_0_4

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/tags/v1_9_0_4@18845 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 101 insertions, 0 deletions

diff --git a/trunk/test/webrick/test_cgi.rb b/trunk/test/webrick/test_cgi.rb
new file mode 100644
index 0000000000..b50038d84c
--- /dev/null
+++ b/trunk/test/webrick/test_cgi.rb
@@ -0,0 +1,101 @@
+require "webrick"
+require File.join(File.dirname(__FILE__), "utils.rb")
+require "test/unit"
+
+class TestWEBrickCGI < Test::Unit::TestCase
+  CRLF = "\r\n"
+
+  def start_cgi_server(&block)
+    config = {
+      :CGIInterpreter => TestWEBrick::RubyBin,
+      :DocumentRoot => File.dirname(__FILE__),
+      :DirectoryIndex => ["webrick.cgi"],
+      :RequestHandler => Proc.new{|req, res|
+        def req.meta_vars
+          meta = super
+          meta["RUBYLIB"] = $:.join(File::PATH_SEPARATOR)
+          return meta
+        end
+      },
+    }
+    if RUBY_PLATFORM =~ /mswin32|mingw|cygwin|bccwin32/
+      config[:CGIPathEnv] = ENV['PATH'] # runtime dll may not be in system dir.
+    end
+    TestWEBrick.start_httpserver(config){|server, addr, port|
+      block.call(server, addr, port)
+    }
+  end
+
+  def test_cgi
+    start_cgi_server{|server, addr, port|
+      http = Net::HTTP.new(addr, port)
+      req = Net::HTTP::Get.new("/webrick.cgi")
+      http.request(req){|res| assert_equal("/webrick.cgi", res.body)}
+      req = Net::HTTP::Get.new("/webrick.cgi/path/info")
+      http.request(req){|res| assert_equal("/path/info", res.body)}
+      req = Net::HTTP::Get.new("/webrick.cgi/%3F%3F%3F?foo=bar")
+      http.request(req){|res| assert_equal("/???", res.body)}
+      req = Net::HTTP::Get.new("/webrick.cgi/%A4%DB%A4%B2/%A4%DB%A4%B2")
+      http.request(req){|res|
+        assert_equal("/\xA4\xDB\xA4\xB2/\xA4\xDB\xA4\xB2", res.body)}
+      req = Net::HTTP::Get.new("/webrick.cgi?a=1;a=2;b=x")
+      http.request(req){|res| assert_equal("a=1, a=2, b=x", res.body)}
+      req = Net::HTTP::Get.new("/webrick.cgi?a=1&a=2&b=x")
+      http.request(req){|res| assert_equal("a=1, a=2, b=x", res.body)}
+
+      req = Net::HTTP::Post.new("/webrick.cgi?a=x;a=y;b=1")
+      req["Content-Type"] = "application/x-www-form-urlencoded"
+      http.request(req, "a=1;a=2;b=x"){|res|
+        assert_equal("a=1, a=2, b=x", res.body)}
+      req = Net::HTTP::Post.new("/webrick.cgi?a=x&a=y&b=1")
+      req["Content-Type"] = "application/x-www-form-urlencoded"
+      http.request(req, "a=1&a=2&b=x"){|res|
+        assert_equal("a=1, a=2, b=x", res.body)}
+      req = Net::HTTP::Get.new("/")
+      http.request(req){|res|
+        ary = res.body.lines.to_a
+        assert_match(%r{/$}, ary[0])
+        assert_match(%r{/webrick.cgi$}, ary[1])
+      }
+
+      req = Net::HTTP::Get.new("/webrick.cgi")
+      req["Cookie"] = "CUSTOMER=WILE_E_COYOTE; PART_NUMBER=ROCKET_LAUNCHER_0001"
+      http.request(req){|res|
+        assert_equal(
+          "CUSTOMER=WILE_E_COYOTE\nPART_NUMBER=ROCKET_LAUNCHER_0001\n",
+          res.body)
+      }
+
+      req = Net::HTTP::Get.new("/webrick.cgi")
+      cookie =  %{$Version="1"; }
+      cookie << %{Customer="WILE_E_COYOTE"; $Path="/acme"; }
+      cookie << %{Part_Number="Rocket_Launcher_0001"; $Path="/acme"; }
+      cookie << %{Shipping="FedEx"; $Path="/acme"}
+      req["Cookie"] = cookie
+      http.request(req){|res|
+        assert_equal("Customer=WILE_E_COYOTE, Shipping=FedEx",
+                     res["Set-Cookie"])
+        assert_equal("Customer=WILE_E_COYOTE\n" +
+                     "Part_Number=Rocket_Launcher_0001\n" +
+                     "Shipping=FedEx\n", res.body)
+      }
+    }
+  end
+
+  def test_bad_request
+    start_cgi_server{|server, addr, port|
+      sock = TCPSocket.new(addr, port)
+      begin
+        sock << "POST /webrick.cgi HTTP/1.0" << CRLF
+        sock << "Content-Type: application/x-www-form-urlencoded" << CRLF
+        sock << "Content-Length: 1024" << CRLF
+        sock << CRLF
+        sock << "a=1&a=2&b=x"
+        sock.close_write
+        assert_match(%r{\AHTTP/\d.\d 400 Bad Request}, sock.read)
+      ensure
+        sock.close
+      end
+    }
+  end
+end