client/legacy_client_session: set transfer-encoding: chunked if content-length is not specified

1 files changed, 46 insertions, 3 deletions

diff --git a/test/plum/client/test_legacy_client_session.rb b/test/plum/client/test_legacy_client_session.rb
index 2c8f146..421a261 100644
--- a/test/plum/client/test_legacy_client_session.rb
+++ b/test/plum/client/test_legacy_client_session.rb
@@ -37,11 +37,54 @@ class LegacyClientSessionTest < Minitest::Test
     io = StringIO.new
     session = LegacyClientSession.new(io, Client::DEFAULT_CONFIG.merge(hostname: "aa"))
     res = session.request({ ":method" => "GET", ":path" => "/aa" }, "aa", {})
-    assert("GET /aa HTTP/1.1\r\nhost: aa\r\n\r\naa")
+    assert_equal("GET /aa HTTP/1.1\r\nhost: aa\r\ntransfer-encoding: chunked\r\n\r\n2\r\naa\r\n", io.string)
     io.string << "HTTP/1.1 200 OK\r\nContent-Length: 3\r\n\r\naaa"
     session.succ until res.finished?
     assert(res.finished?)
-    assert("aaa", res.body)
-    assert({ ":status" => "200", "content-length" => "3" }, res.headers)
+    assert_equal("aaa", res.body)
+    assert_equal({ ":status" => "200", "content-length" => "3" }, res.headers)
+  end
+
+  def test_chunked_chunked_string
+    io = StringIO.new
+    session = LegacyClientSession.new(io, Client::DEFAULT_CONFIG.merge(hostname: "hostname"))
+    res = session.request({ ":method" => "GET", ":path" => "/aa" }, "a" * 1025, {})
+    assert_equal(<<-EOR, io.string)
+GET /aa HTTP/1.1\r
+host: hostname\r
+transfer-encoding: chunked\r
+\r
+401\r
+#{"a"*1025}\r
+    EOR
+  end
+
+  def test_chunked_chunked_io
+    io = StringIO.new
+    session = LegacyClientSession.new(io, Client::DEFAULT_CONFIG.merge(hostname: "hostname"))
+    res = session.request({ ":method" => "GET", ":path" => "/aa" }, StringIO.new("a" * 1025), {})
+    assert_equal(<<-EOR, io.string)
+GET /aa HTTP/1.1\r
+host: hostname\r
+transfer-encoding: chunked\r
+\r
+400\r
+#{"a"*1024}\r
+1\r
+a\r
+    EOR
+  end
+
+  def test_chunked_sized
+    io = StringIO.new
+    session = LegacyClientSession.new(io, Client::DEFAULT_CONFIG.merge(hostname: "hostname"))
+    res = session.request({ ":method" => "GET", ":path" => "/aa", "content-length" => 1025 }, StringIO.new("a" * 1025), {})
+    assert_equal((<<-EOR).chomp, io.string)
+GET /aa HTTP/1.1\r
+content-length: 1025\r
+host: hostname\r
+\r
+#{"a"*1025}
+    EOR
   end
 end