spec/cookie_spec: Add test for :store_from_request()

Author: daurnimator

spec/cookie_spec.lua

@@ -381,6 +381,27 @@ describe("cookie module", function()
 			assert.same("foo=path; bar=path; bar=domain; bar=time; foo=time", s:lookup("sub.example.com", "/path/longerpath", true, true))
 		end)
 	end)
+	it("can store cookies from a request+response", function()
+		local s = http_cookie.new_store()
+		local req_headers = http_headers.new()
+		req_headers:append(":scheme", "http")
+		req_headers:append(":method", "GET")
+		req_headers:append(":path", "/")
+		local resp_headers = http_headers.new()
+		resp_headers:append(":status", "200")
+		resp_headers:append("set-cookie", http_cookie.bake("foo", "FOO"))
+		resp_headers:append("set-cookie", http_cookie.bake("bar", "BAR", 0))
+		assert.truthy(s:store_from_request(req_headers, resp_headers, "my.host", "my.host"))
+		assert.same("FOO", s:get("my.host", "/", "foo"))
+		assert.same(nil, s:get("my.host", "/", "bar"))
+		-- Now with an :authority header
+		req_headers:append(":authority", "my.host")
+		resp_headers:append("set-cookie", http_cookie.bake("baz", "BAZ"))
+		assert.truthy(s:store_from_request(req_headers, resp_headers, "my.host", "my.host"))
+		assert.same("FOO", s:get("my.host", "/", "foo"))
+		assert.same(nil, s:get("my.host", "/", "bar"))
+		assert.same("BAZ", s:get("my.host", "/", "baz"))
+	end)
 	it("enforces store.max_cookie_length", function()
 		local s = http_cookie.new_store()
 		s.max_cookie_length = 3