38 lines
1.1 KiB
C++
38 lines
1.1 KiB
C++
#include <drogon/drogon_test.h>
|
|
#include <drogon/HttpRequest.h>
|
|
#include <drogon/HttpResponse.h>
|
|
#include "../../lib/src/HttpResponseImpl.h"
|
|
|
|
using namespace drogon;
|
|
|
|
DROGON_TEST(HttpHeaderRequest)
|
|
{
|
|
auto req = HttpRequest::newHttpRequest();
|
|
req->addHeader("Abc", "abc");
|
|
CHECK(req->getHeader("Abc") == "abc");
|
|
CHECK(req->getHeader("abc") == "abc");
|
|
|
|
req->removeHeader("Abc");
|
|
CHECK(req->getHeader("abc") == "");
|
|
}
|
|
DROGON_TEST(HttpHeaderResponse)
|
|
{
|
|
auto resp = std::dynamic_pointer_cast<HttpResponseImpl>(
|
|
HttpResponse::newHttpResponse());
|
|
REQUIRE(resp != nullptr);
|
|
resp->addHeader("Abc", "abc");
|
|
CHECK(resp->getHeader("Abc") == "abc");
|
|
CHECK(resp->getHeader("abc") == "abc");
|
|
resp->makeHeaderString();
|
|
|
|
auto buffer = resp->renderToBuffer();
|
|
auto str = std::string{buffer->peek(), buffer->readableBytes()};
|
|
CHECK(str.find("abc") != std::string::npos);
|
|
|
|
resp->removeHeader("Abc");
|
|
buffer = resp->renderToBuffer();
|
|
str = std::string{buffer->peek(), buffer->readableBytes()};
|
|
CHECK(str.find("abc") == std::string::npos);
|
|
CHECK(resp->getHeader("abc") == "");
|
|
}
|