ByteFlow: bugfix and test improvement

GitOrigin-RevId: 53f684cc96c52cbde4cb2109670f57714724fccc
This commit is contained in:
Arseny Smirnov 2020-07-23 14:39:28 +03:00
parent 813e2b2961
commit 93e0a10ffb
2 changed files with 3 additions and 3 deletions

View File

@ -45,7 +45,7 @@ class ByteFlowBaseCommon : public ByteFlowInterface {
}
void wakeup() final {
if (stop_flag_) {
if (stop_flag_ || !input_) {
return;
}
input_->sync_with_writer();

View File

@ -133,14 +133,14 @@ TEST(Http, reader) {
clear_thread_locals();
SET_VERBOSITY_LEVEL(VERBOSITY_NAME(ERROR));
auto start_mem = BufferAllocator::get_buffer_mem();
{
for (int i = 0; i < 20; i++) {
td::ChainBufferWriter input_writer;
auto input = input_writer.extract_reader();
HttpReader reader;
int max_post_size = 10000;
reader.init(&input, max_post_size, 0);
std::vector<string> contents(1000);
std::vector<string> contents(100);
std::generate(contents.begin(), contents.end(), gen_http_content);
auto v = td::transform(contents, rand_http_query);
auto vec_str = rand_split(join(v));