spec/h2_stream_spec: make sure trailers are properly handled on response

Author: Ph4ntomas

spec/h2_stream_spec.lua

@@ -47,6 +47,66 @@ describe("http.h2_stream", function()
 		assert_loop(cq, TEST_TIMEOUT)
 		assert.truthy(cq:empty())
 	end)
+	it("accepts trailers without pseudo_header", function()
+		local s, c = new_pair()
+		local cq = cqueues.new()
+		cq:wrap(function()
+			local client_stream = c:new_stream()
+			local req_headers = new_headers()
+			req_headers:append(":method", "GET")
+			req_headers:append(":scheme", "http")
+			req_headers:append(":path", "/")
+			assert(client_stream:write_headers(req_headers, true))
+			assert(client_stream:get_headers())
+			assert(client_stream:get_headers())
+			assert(c:close())
+		end)
+		cq:wrap(function()
+			local stream = assert(s:get_next_incoming_stream())
+			assert(stream:get_headers())
+			local res_headers = new_headers()
+			res_headers:append(":status", "200")
+			local res_trailers = new_headers()
+			res_trailers:append("trailer-status", "0")
+			assert(stream:write_headers(res_headers, false))
+			assert(stream:write_headers(res_trailers, true))
+			assert(s:close())
+		end)
+		assert_loop(cq, TEST_TIMEOUT)
+		assert.truthy(cq:empty())
+	end)
+	it("accepts trailers after interim responses", function()
+		local s, c = new_pair()
+		local cq = cqueues.new()
+		cq:wrap(function()
+			local client_stream = c:new_stream()
+			local req_headers = new_headers()
+			req_headers:append(":method", "GET")
+			req_headers:append(":scheme", "http")
+			req_headers:append(":path", "/")
+			assert(client_stream:write_headers(req_headers, true))
+			assert(client_stream:get_headers()) -- 100 continue
+			assert(client_stream:get_headers()) -- 200 OK
+			assert(client_stream:get_headers()) -- trailer
+			assert(c:close())
+		end)
+		cq:wrap(function()
+			local stream = assert(s:get_next_incoming_stream())
+			assert(stream:get_headers())
+			local info_headers = new_headers()
+			info_headers:append(":status", "100")
+			local res_headers = new_headers()
+			res_headers:append(":status", "200")
+			local res_trailers = new_headers()
+			res_trailers:append("trailer-status", "0")
+			assert(stream:write_headers(info_headers, false))
+			assert(stream:write_headers(res_headers, false))
+			assert(stream:write_headers(res_trailers, true))
+			assert(s:close())
+		end)
+		assert_loop(cq, TEST_TIMEOUT)
+		assert.truthy(cq:empty())
+	end)
 	it("can send a body", function()
 		local s, c = new_pair()
 		local cq = cqueues.new()