75 lines
2.4 KiB
C++
75 lines
2.4 KiB
C++
|
#include <drogon/Cookie.h>
|
||
|
|
#include <drogon/drogon_test.h>
|
||
|
|
#include <drogon/HttpAppFramework.h>
|
||
|
|
#include <drogon/HttpClient.h>
|
||
|
|
#include <drogon/HttpRequest.h>
|
||
|
|
#include <drogon/HttpResponse.h>
|
||
|
|
#include <drogon/HttpTypes.h>
|
||
|
|
|
||
|
|
using namespace drogon;
|
||
|
|
|
||
|
|
struct CookieSameSiteSequence
|
||
|
|
{
|
||
|
|
CookieSameSiteSequence()
|
||
|
|
{
|
||
|
|
i = valid_sameSite_values.begin();
|
||
|
|
sessionCookie.setKey("JSESSIONID");
|
||
|
|
}
|
||
|
|
|
||
|
|
std::vector<std::string> valid_sameSite_values{"Null",
|
||
|
|
"Lax",
|
||
|
|
"None",
|
||
|
|
"Strict"};
|
||
|
|
|
||
|
|
std::vector<std::string>::const_iterator i;
|
||
|
|
Cookie sessionCookie;
|
||
|
|
};
|
||
|
|
|
||
|
|
DROGON_TEST(CookieSameSite)
|
||
|
|
{
|
||
|
|
auto client =
|
||
|
|
HttpClient::newHttpClient("https://127.0.0.1:8849",
|
||
|
|
HttpAppFramework::instance().getLoop(),
|
||
|
|
false,
|
||
|
|
false);
|
||
|
|
auto req = HttpRequest::newHttpRequest();
|
||
|
|
|
||
|
|
CookieSameSiteSequence seq;
|
||
|
|
|
||
|
|
while (seq.i != seq.valid_sameSite_values.end())
|
||
|
|
{
|
||
|
|
std::promise<void> p1;
|
||
|
|
std::future<void> f1 = p1.get_future();
|
||
|
|
req->setPath(std::string("/CookieSameSiteController/set/") + *seq.i);
|
||
|
|
if (seq.sessionCookie.getValue() != "")
|
||
|
|
{
|
||
|
|
// add session cookie
|
||
|
|
req->addCookie(seq.sessionCookie.getKey(),
|
||
|
|
seq.sessionCookie.getValue());
|
||
|
|
} // endif
|
||
|
|
|
||
|
|
client->sendRequest(
|
||
|
|
req,
|
||
|
|
[TEST_CTX, &seq, &p1](ReqResult res, const HttpResponsePtr &resp) {
|
||
|
|
LOG_INFO << "sameCookie value == " << *seq.i;
|
||
|
|
REQUIRE(res == ReqResult::Ok);
|
||
|
|
REQUIRE(resp != nullptr);
|
||
|
|
|
||
|
|
CHECK(resp->getStatusCode() == HttpStatusCode::k200OK);
|
||
|
|
CHECK(resp->contentType() == CT_APPLICATION_JSON);
|
||
|
|
|
||
|
|
auto json = resp->getJsonObject();
|
||
|
|
LOG_INFO << "BODY\n\t" << resp->getBody() << "\n\t"
|
||
|
|
<< (*json)["result"].asString() << "\n\t"
|
||
|
|
<< (*json)["new value"].asString() << "\n\t"
|
||
|
|
<< (*json)["old value"].asString();
|
||
|
|
|
||
|
|
seq.sessionCookie = resp->getCookie("JSESSIONID");
|
||
|
|
|
||
|
|
p1.set_value();
|
||
|
|
});
|
||
|
|
f1.get();
|
||
|
|
seq.i++;
|
||
|
|
}
|
||
|
|
};
|