HTTPMessage getChunkedPostRequest() {
HTTPMessage req;
req.setMethod("POST");
req.setURL<string>("/");
req.setHTTPVersion(1, 1);
req.setIsChunked(true);
req.getHeaders().set(HTTP_HEADER_HOST, "www.foo.com");
req.getHeaders().set(HTTP_HEADER_TRANSFER_ENCODING, "chunked");
return req;
}
TEST(HTTP1xCodecTest, TestGetRequestChunkedResponse) {
HTTP1xCodec codec(TransportDirection::DOWNSTREAM);
HTTP1xCodecCallback callbacks;
codec.setCallback(&callbacks);
auto txnID = codec.createStream();
// Generate a GET request
auto reqBuf = folly::IOBuf::copyBuffer(
"GET /www.facebook.com HTTP/1.1\nHost: www.facebook.com\n\n");
codec.onIngress(*reqBuf);
EXPECT_EQ(callbacks.headersComplete, 1);
// Generate chunked response with body
HTTPMessage resp;
resp.setHTTPVersion(1, 1);
resp.setStatusCode(200);
resp.setIsChunked(true);
resp.getHeaders().set(HTTP_HEADER_TRANSFER_ENCODING, "chunked");
folly::IOBufQueue respBuf(folly::IOBufQueue::cacheChainLength());
codec.generateHeader(respBuf, txnID, resp, 0, false);
auto headerFromBuf = respBuf.split(respBuf.chainLength());
string resp1("Hello");
auto body1 = folly::IOBuf::copyBuffer(resp1);
string resp2("");
auto body2 = folly::IOBuf::copyBuffer(resp2);
codec.generateBody(respBuf, txnID, std::move(body1), HTTPCodec::NoPadding,
false);
auto bodyFromBuf = respBuf.split(respBuf.chainLength());
ASSERT_EQ("5\r\nHello\r\n", bodyFromBuf->moveToFbString());
codec.generateBody(respBuf, txnID, std::move(body2), HTTPCodec::NoPadding,
true);
bodyFromBuf = respBuf.split(respBuf.chainLength());
ASSERT_EQ("0\r\n\r\n", bodyFromBuf->moveToFbString());
}
TEST(HTTP1xCodecTest, TestHeadRequestChunkedResponse) {
HTTP1xCodec codec(TransportDirection::DOWNSTREAM);
HTTP1xCodecCallback callbacks;
codec.setCallback(&callbacks);
auto txnID = codec.createStream();
// Generate a HEAD request
auto reqBuf = folly::IOBuf::copyBuffer(
"HEAD /www.facebook.com HTTP/1.1\nHost: www.facebook.com\n\n");
codec.onIngress(*reqBuf);
EXPECT_EQ(callbacks.headersComplete, 1);
// Generate chunked response with no body
HTTPMessage resp;
resp.setHTTPVersion(1, 1);
resp.setStatusCode(200);
resp.setIsChunked(true);
resp.getHeaders().set(HTTP_HEADER_TRANSFER_ENCODING, "chunked");
folly::IOBufQueue respBuf(folly::IOBufQueue::cacheChainLength());
codec.generateHeader(respBuf, txnID, resp, 0, true);
auto respStr = respBuf.move()->moveToFbString();
EXPECT_TRUE(respStr.find("0\r\n") == string::npos);
}
TEST(HTTP1xCodecTest, TestChunkedUpstream) {
HTTP1xCodec codec(TransportDirection::UPSTREAM);
auto txnID = codec.createStream();
HTTPMessage msg;
msg.setHTTPVersion(1, 1);
msg.setURL("https://www.facebook.com/");
msg.getHeaders().set(HTTP_HEADER_HOST, "www.facebook.com");
msg.getHeaders().set(HTTP_HEADER_TRANSFER_ENCODING, "chunked");
msg.setIsChunked(true);
HTTPHeaderSize size;
folly::IOBufQueue buf(folly::IOBufQueue::cacheChainLength());
codec.generateHeader(buf, txnID, msg, 0, false, &size);
auto headerFromBuf = buf.split(buf.chainLength());
string req1("Hello");
auto body1 = folly::IOBuf::copyBuffer(req1);
string req2("World");
auto body2 = folly::IOBuf::copyBuffer(req2);
codec.generateBody(buf, txnID, std::move(body1), HTTPCodec::NoPadding,
false);
auto bodyFromBuf = buf.split(buf.chainLength());
ASSERT_EQ("5\r\nHello\r\n", bodyFromBuf->moveToFbString());
codec.generateBody(buf, txnID, std::move(body2), HTTPCodec::NoPadding,
true);
LOG(WARNING) << "len chain" << buf.chainLength();
auto eomFromBuf = buf.split(buf.chainLength());
ASSERT_EQ("5\r\nWorld\r\n0\r\n\r\n", eomFromBuf->moveToFbString());
}
