This commit is contained in:
yhirose 2023-04-08 14:53:55 -04:00
parent 21f9c51556
commit d587548250
2 changed files with 21 additions and 6 deletions

View file

@ -4253,6 +4253,16 @@ TEST(ClientProblemDetectionTest, ContentProvider) {
[](bool success) { ASSERT_FALSE(success); });
});
svr.Get("/empty", [&](const Request & /*req*/, Response &res) {
res.set_content_provider(
0, "text/plain",
[&](size_t /*offset*/, size_t /*length*/, DataSink & /*sink*/) -> bool {
EXPECT_TRUE(false);
return true;
},
[](bool success) { ASSERT_FALSE(success); });
});
auto listen_thread = std::thread([&svr]() { svr.listen("localhost", PORT); });
auto se = detail::scope_exit([&] {
svr.stop();
@ -4269,11 +4279,17 @@ TEST(ClientProblemDetectionTest, ContentProvider) {
Client cli("localhost", PORT);
auto res = cli.Get("/hi", [&](const char * /*data*/, size_t /*data_length*/) {
return false;
});
{
auto res = cli.Get("/hi", [&](const char * /*data*/,
size_t /*data_length*/) { return false; });
ASSERT_FALSE(res);
}
ASSERT_FALSE(res);
{
auto res = cli.Get("/empty", [&](const char * /*data*/,
size_t /*data_length*/) { return false; });
ASSERT_TRUE(res);
}
}
TEST(ErrorHandlerWithContentProviderTest, ErrorHandler) {