tp2 0.17 → 0.18

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7f064819fba5d42c4376f3895ee0cce07745149e6b9ed06e864acb1ac3515d9e
-  data.tar.gz: 84767b1d2cd87c6006ba7202f48c2bf1cdc9345ddf258c8fcd716bcf2023c264
+  metadata.gz: 6b2eaad3a3794a09c6a10e69cc3813ad27789049bd71055f315edcc5ac61b1a6
+  data.tar.gz: 7a8529db038480936bbb44a2a9e059ea5625b6d3f1cbb9eb7d04c185ab4268b9
 SHA512:
-  metadata.gz: 72b97e276e4abb593566d761f44ac7bfa0a9af5d9ff5dcbece407412dd07eac84a9dfbe6d983c832010fb9a365be27053df471d2b9e76a2955b7332a33b06e69
-  data.tar.gz: 6a13102f1f2008b32aa34fec7da9b2cda496e19f5445525f398b6c4f0a863b15bb68bf94b457e56f0b3c6ace41d3caca4722158ea69fc638d6a828f39e047bbd
+  metadata.gz: 5a8c23907e0e4f4d9024bf2457efdfa1b003d3d4870a65ec6a44e8ab3cc67d5d7bf096a5adba39a0e552c0eefe7fe9306a1612c19382ff64a3689805c98ff903
+  data.tar.gz: c4183ad2dade4aaf4511fd4d9f856c2532bf8c374a99a632010c629112c1f1fc8020f7bc7580cf3295a25bfd2826ce79b99561348f35c727176b20ea4f559cb6

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+# 0.18 2025-09-14
+
+- Add `Request#set_cookie` method
+
 # 0.17 2025-09-14
 
 - Add support for server extensions: `Date`, `Server` headers

data/lib/tp2/connection.rb

@@ -142,6 +142,15 @@ module TP2
       @response_headers ? @response_headers.merge!(headers) : @response_headers = headers
     end
 
+    def set_cookie(*cookies)
+      existing_cookies = @response_headers && @response_headers['Set-Cookie']
+      if existing_cookies
+        @response_headers['Set-Cookie'] = existing_cookies + cookies
+      else
+        set_response_headers('Set-Cookie' => cookies)
+      end
+    end
+
     SEND_FLAGS = UM::MSG_NOSIGNAL | UM::MSG_WAITALL
 
     # Sends response including headers and body. Waits for the request to complete

data/lib/tp2/request_extensions.rb

@@ -15,4 +15,8 @@ class Qeweney::Request
   def set_response_headers(headers)
     adapter.set_response_headers(headers)
   end
+
+  def set_cookie(*)
+    adapter.set_cookie(*)
+  end
 end

data/lib/tp2/version.rb

@@ -1,3 +1,3 @@
 module TP2
-  VERSION = '0.17'
+  VERSION = '0.18'
 end

data/test/test_connection.rb

@@ -582,4 +582,48 @@ class ConnectionTest < Minitest::Test
     expected = "HTTP/1.1 200\r\nTransfer-Encoding: chunked\r\nSet-Cookie: foo=bar\r\nFoo: bar\r\nContent-Type: text/plain\r\n\r\n3\r\nfoo\r\n0\r\n\r\n"
     assert_equal expected, response
   end
+
+  def test_set_cookie_single
+    @hook = ->(req) {
+      req.set_cookie('foo=bar; HttpOnly')
+      req.respond('foo')
+    }
+
+    write_client_side("GET / HTTP/1.1\r\n\r\n")
+    @adapter.serve_request
+    response = read_client_side(65536)
+    expected = "HTTP/1.1 200\r\nTransfer-Encoding: chunked\r\nSet-Cookie: foo=bar; HttpOnly\r\n\r\n3\r\nfoo\r\n0\r\n\r\n"
+    assert_equal expected, response
+    
+  end
+
+  def test_set_cookie_multi1
+    @hook = ->(req) {
+      req.set_cookie('foo=bar; HttpOnly', 'bar=baz')
+      req.respond('foo')
+    }
+
+    write_client_side("GET / HTTP/1.1\r\n\r\n")
+    @adapter.serve_request
+    response = read_client_side(65536)
+    expected = "HTTP/1.1 200\r\nTransfer-Encoding: chunked\r\nSet-Cookie: foo=bar; HttpOnly\r\nSet-Cookie: bar=baz\r\n\r\n3\r\nfoo\r\n0\r\n\r\n"
+    assert_equal expected, response
+    
+  end
+
+  def test_set_cookie_multi2
+    @hook = ->(req) {
+      req.set_cookie('a=1', 'b=2')
+      req.set_cookie('c=3')
+      req.set_cookie('d=4', 'e=5')
+      req.respond('foo')
+    }
+
+    write_client_side("GET / HTTP/1.1\r\n\r\n")
+    @adapter.serve_request
+    response = read_client_side(65536)
+    expected = "HTTP/1.1 200\r\nTransfer-Encoding: chunked\r\nSet-Cookie: a=1\r\nSet-Cookie: b=2\r\nSet-Cookie: c=3\r\nSet-Cookie: d=4\r\nSet-Cookie: e=5\r\n\r\n3\r\nfoo\r\n0\r\n\r\n"
+    assert_equal expected, response
+    
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tp2
 version: !ruby/object:Gem::Version
-  version: '0.17'
+  version: '0.18'
 platform: ruby
 authors:
 - Sharon Rosner