test: changing waitForEndStream to fail if it takes 30s

test/extensions/clusters/aggregate/cluster_integration_test.cc

@@ -288,7 +288,7 @@ TEST_P(AggregateIntegrationTest, PreviousPrioritiesRetryPredicate) {
   waitForNextUpstreamRequest(SecondUpstreamIndex);
   upstream_request_->encodeHeaders(default_response_headers_, true);
 
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   EXPECT_TRUE(upstream_request_->complete());
 
   EXPECT_TRUE(response->complete());

test/extensions/filters/http/adaptive_concurrency/adaptive_concurrency_filter_integration_test.cc

@@ -83,7 +83,7 @@ void AdaptiveConcurrencyIntegrationTest::respondToRequest(bool expect_forwarded)
     upstream_requests_.front()->encodeData(1, true);
   }
 
-  responses_.front()->waitForEndStream();
+  ASSERT_TRUE(responses_.front()->waitForEndStream());
 
   if (expect_forwarded) {
     EXPECT_TRUE(upstream_requests_.front()->complete());

test/extensions/filters/http/admission_control/admission_control_integration_test.cc

@@ -70,7 +70,7 @@ class AdmissionControlIntegrationTest : public Event::TestUsingSimulatedTime,
     au->setResponseTrailers(std::move(trailers));
 
     auto response = codec_client_->makeHeaderOnlyRequest(default_request_headers_);
-    response->waitForEndStream();
+    RELEASE_ASSERT(response->waitForEndStream(), "unexpected timeout");
     codec_client_->close();
     return response;
   }
@@ -84,7 +84,7 @@ class AdmissionControlIntegrationTest : public Event::TestUsingSimulatedTime,
         Http::TestResponseHeaderMapImpl({{":status", code}})));
 
     auto response = codec_client_->makeHeaderOnlyRequest(default_request_headers_);
-    response->waitForEndStream();
+    RELEASE_ASSERT(response->waitForEndStream(), "unexpected timeout");
     codec_client_->close();
     return response;
   }

test/extensions/filters/http/aws_lambda/aws_lambda_filter_integration_test.cc

@@ -118,7 +118,7 @@ class AwsLambdaFilterIntegrationTest : public testing::TestWithParam<Network::Ad
       upstream_request_->encodeData(buffer, true);
     }
 
-    response->waitForEndStream();
+    ASSERT_TRUE(response->waitForEndStream());
     EXPECT_TRUE(response->complete());
 
     // verify headers

test/extensions/filters/http/buffer/buffer_filter_integration_test.cc

@@ -61,7 +61,7 @@ TEST_P(BufferIntegrationTest, RouterRequestPopulateContentLength) {
   ASSERT_NE(content_length, nullptr);
   EXPECT_EQ(content_length->value().getStringView(), "9");
 
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("200", response->headers().getStatusValue());
 }
@@ -90,7 +90,7 @@ TEST_P(BufferIntegrationTest, RouterRequestPopulateContentLengthOnTrailers) {
   ASSERT_NE(content_length, nullptr);
   EXPECT_EQ(content_length->value().getStringView(), "10");
 
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("200", response->headers().getStatusValue());
 }
@@ -117,7 +117,7 @@ TEST_P(BufferIntegrationTest, RouterRequestBufferLimitExceeded) {
                                      {"x-envoy-retry-on", "5xx"}},
       1024 * 65);
 
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("413", response->headers().getStatusValue());
 }
@@ -159,7 +159,7 @@ TEST_P(BufferIntegrationTest, RouteDisabled) {
   waitForNextUpstreamRequest();
   upstream_request_->encodeHeaders(Http::TestResponseHeaderMapImpl{{":status", "200"}}, true);
 
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("200", response->headers().getStatusValue());
 }
@@ -185,7 +185,7 @@ TEST_P(BufferIntegrationTest, RouteOverride) {
   waitForNextUpstreamRequest();
   upstream_request_->encodeHeaders(Http::TestResponseHeaderMapImpl{{":status", "200"}}, true);
 
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("200", response->headers().getStatusValue());
 }

test/extensions/filters/http/cache/cache_filter_integration_test.cc

@@ -69,7 +69,7 @@ TEST_P(CacheIntegrationTest, MissInsertHit) {
     // send 42 'a's
     upstream_request_->encodeData(response_body, /*end_stream=*/true);
     // Wait for the response to be read by the codec client.
-    response_decoder->waitForEndStream();
+    ASSERT_TRUE(response_decoder->waitForEndStream());
     EXPECT_TRUE(response_decoder->complete());
     EXPECT_THAT(response_decoder->headers(), IsSupersetOfHeaders(response_headers));
     EXPECT_TRUE(response_decoder->headers().get(Http::CustomHeaders::get().Age).empty());
@@ -84,7 +84,7 @@ TEST_P(CacheIntegrationTest, MissInsertHit) {
   {
     IntegrationStreamDecoderPtr response_decoder =
         codec_client_->makeHeaderOnlyRequest(request_headers);
-    response_decoder->waitForEndStream();
+    ASSERT_TRUE(response_decoder->waitForEndStream());
     EXPECT_TRUE(response_decoder->complete());
     EXPECT_THAT(response_decoder->headers(), IsSupersetOfHeaders(response_headers));
     EXPECT_EQ(response_decoder->body(), response_body);
@@ -127,7 +127,7 @@ TEST_P(CacheIntegrationTest, ExpiredValidated) {
     // send 42 'a's
     upstream_request_->encodeData(response_body, true);
     // Wait for the response to be read by the codec client.
-    response_decoder->waitForEndStream();
+    ASSERT_TRUE(response_decoder->waitForEndStream());
     EXPECT_TRUE(response_decoder->complete());
     EXPECT_THAT(response_decoder->headers(), IsSupersetOfHeaders(response_headers));
     EXPECT_TRUE(response_decoder->headers().get(Http::CustomHeaders::get().Age).empty());
@@ -159,7 +159,7 @@ TEST_P(CacheIntegrationTest, ExpiredValidated) {
     response_headers.setDate(not_modified_date);
 
     // Wait for the response to be read by the codec client.
-    response_decoder->waitForEndStream();
+    ASSERT_TRUE(response_decoder->waitForEndStream());
 
     // Check that the served response is the cached response
     EXPECT_TRUE(response_decoder->complete());
@@ -208,7 +208,7 @@ TEST_P(CacheIntegrationTest, ExpiredFetchedNewResponse) {
     // send 10 'a's
     upstream_request_->encodeData(response_body, /*end_stream=*/true);
     // Wait for the response to be read by the codec client.
-    response_decoder->waitForEndStream();
+    ASSERT_TRUE(response_decoder->waitForEndStream());
     EXPECT_TRUE(response_decoder->complete());
     EXPECT_THAT(response_decoder->headers(), IsSupersetOfHeaders(response_headers));
     EXPECT_TRUE(response_decoder->headers().get(Http::CustomHeaders::get().Age).empty());
@@ -244,7 +244,7 @@ TEST_P(CacheIntegrationTest, ExpiredFetchedNewResponse) {
     upstream_request_->encodeData(response_body, /*end_stream=*/true);
 
     // Wait for the response to be read by the codec client.
-    response_decoder->waitForEndStream();
+    ASSERT_TRUE(response_decoder->waitForEndStream());
     // Check that the served response is the updated response
     EXPECT_TRUE(response_decoder->complete());
     EXPECT_THAT(response_decoder->headers(), IsSupersetOfHeaders(response_headers));
@@ -289,7 +289,7 @@ TEST_P(CacheIntegrationTest, GetRequestWithBodyAndTrailers) {
     // send 42 'a's
     upstream_request_->encodeData(42, true);
     // Wait for the response to be read by the codec client.
-    response->waitForEndStream();
+    ASSERT_TRUE(response->waitForEndStream());
     EXPECT_TRUE(response->complete());
     EXPECT_THAT(response->headers(), IsSupersetOfHeaders(response_headers));
     EXPECT_TRUE(response->headers().get(Http::CustomHeaders::get().Age).empty());

test/extensions/filters/http/cdn_loop/filter_integration_test.cc

@@ -97,7 +97,7 @@ TEST_P(CdnLoopFilterIntegrationTest, CdnLoop0Allowed1Seen) {
                                                  {"CDN-Loop", "cdn"}};
 
   auto response = codec_client_->makeHeaderOnlyRequest(request_headers);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("502", response->headers().getStatusValue());
 }
@@ -114,7 +114,7 @@ TEST_P(CdnLoopFilterIntegrationTest, UnparseableHeader) {
                                                  {"CDN-Loop", "[bad-header"}};
 
   auto response = codec_client_->makeHeaderOnlyRequest(request_headers);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("400", response->headers().getStatusValue());
 }
@@ -171,7 +171,7 @@ TEST_P(CdnLoopFilterIntegrationTest, CdnLoop2Allowed3Seen) {
                                                  {"CDN-Loop", "cdn, cdn, cdn"}};
 
   auto response = codec_client_->makeHeaderOnlyRequest(request_headers);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   ASSERT_TRUE(response->complete());
   EXPECT_EQ("502", response->headers().getStatusValue());
 }

test/extensions/filters/http/cors/cors_filter_integration_test.cc

@@ -98,7 +98,7 @@ class CorsFilterIntegrationTest : public testing::TestWithParam<Network::Address
     initialize();
     codec_client_ = makeHttpConnection(lookupPort("http"));
     auto response = codec_client_->makeHeaderOnlyRequest(request_headers);
-    response->waitForEndStream();
+    ASSERT_TRUE(response->waitForEndStream());
     EXPECT_TRUE(response->complete());
     compareHeaders(response->headers(), expected_response_headers);
   }

test/extensions/filters/http/csrf/csrf_filter_integration_test.cc

@@ -59,7 +59,7 @@ class CsrfFilterIntegrationTest : public HttpProtocolIntegrationTest {
     auto response = codec_client_->makeRequestWithBody(request_headers, 1024);
     waitForNextUpstreamRequest();
     upstream_request_->encodeHeaders(Http::TestResponseHeaderMapImpl{{":status", "200"}}, true);
-    response->waitForEndStream();
+    RELEASE_ASSERT(response->waitForEndStream(), "unexpected timeout");
 
     return response;
   }
@@ -68,7 +68,7 @@ class CsrfFilterIntegrationTest : public HttpProtocolIntegrationTest {
     initialize();
     codec_client_ = makeHttpConnection(lookupPort("http"));
     auto response = codec_client_->makeRequestWithBody(request_headers, 1024);
-    response->waitForEndStream();
+    RELEASE_ASSERT(response->waitForEndStream(), "unexpected timeout");
 
     return response;
   }

test/extensions/filters/http/decompressor/decompressor_filter_integration_test.cc

@@ -144,7 +144,7 @@ TEST_P(DecompressorIntegrationTest, BidirectionalDecompression) {
   upstream_request_->encodeData(response_data2, true);
 
   // Wait for frames to arrive downstream.
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
 
   // Assert that the total bytes received downstream equal the sum of the uncompressed byte buffers
   // sent.
@@ -264,7 +264,7 @@ TEST_P(DecompressorIntegrationTest, BidirectionalDecompressionError) {
   upstream_request_->encodeData(response_data2, true);
 
   // Wait for frames to arrive downstream.
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
 
   EXPECT_TRUE(response->complete());
   EXPECT_EQ("200", response->headers().Status()->value().getStringView());

test/extensions/filters/http/dynamic_forward_proxy/proxy_filter_integration_test.cc

@@ -222,7 +222,7 @@ TEST_P(ProxyFilterIntegrationTest, DNSCacheHostOverflow) {
       {":scheme", "http"},
       {":authority", fmt::format("localhost2", fake_upstreams_[0]->localAddress()->ip()->port())}};
   response = codec_client_->makeHeaderOnlyRequest(request_headers2);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   EXPECT_EQ("503", response->headers().getStatusValue());
   EXPECT_EQ(1, test_server_->counter("dns_cache.foo.host_overflow")->value());
 }
@@ -248,7 +248,7 @@ TEST_P(ProxyFilterIntegrationTest, UpstreamTls) {
   EXPECT_STREQ("localhost", SSL_get_servername(ssl_socket->ssl(), TLSEXT_NAMETYPE_host_name));
 
   upstream_request_->encodeHeaders(default_response_headers_, true);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   checkSimpleRequestSuccess(0, 0, response.get());
 }
 
@@ -272,7 +272,7 @@ TEST_P(ProxyFilterIntegrationTest, UpstreamTlsWithIpHost) {
   EXPECT_STREQ(nullptr, SSL_get_servername(ssl_socket->ssl(), TLSEXT_NAMETYPE_host_name));
 
   upstream_request_->encodeHeaders(default_response_headers_, true);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   checkSimpleRequestSuccess(0, 0, response.get());
 }
 
@@ -292,7 +292,7 @@ TEST_P(ProxyFilterIntegrationTest, UpstreamTlsInvalidSAN) {
        fmt::format("localhost:{}", fake_upstreams_[0]->localAddress()->ip()->port())}};
 
   auto response = codec_client_->makeHeaderOnlyRequest(request_headers);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   EXPECT_EQ("503", response->headers().getStatusValue());
 
   EXPECT_EQ(1, test_server_->counter("cluster.cluster_0.ssl.fail_verify_san")->value());
@@ -310,7 +310,7 @@ TEST_P(ProxyFilterIntegrationTest, DnsCacheCircuitBreakersInvoked) {
        fmt::format("localhost:{}", fake_upstreams_[0]->localAddress()->ip()->port())}};
 
   auto response = codec_client_->makeRequestWithBody(request_headers, 1024);
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   EXPECT_EQ(1, test_server_->counter("dns_cache.foo.dns_rq_pending_overflow")->value());
 
   EXPECT_TRUE(response->complete());

test/extensions/filters/http/ext_authz/ext_authz_integration_test.cc

@@ -222,7 +222,7 @@ class ExtAuthzGrpcIntegrationTest : public Grpc::VersionedGrpcClientIntegrationP
           upstream_request_->headers().get(Http::LowerCaseString{header_to_remove.first}).empty());
     }
 
-    response_->waitForEndStream();
+    ASSERT_TRUE(response_->waitForEndStream());
 
     EXPECT_TRUE(upstream_request_->complete());
     EXPECT_EQ(request_body_.length(), upstream_request_->bodyLength());
@@ -531,7 +531,7 @@ class ExtAuthzHttpIntegrationTest : public HttpIntegrationTest,
                     .empty());
 
     upstream_request_->encodeHeaders(Http::TestResponseHeaderMapImpl{{":status", "200"}}, true);
-    response_->waitForEndStream();
+    ASSERT_TRUE(response_->waitForEndStream());
     EXPECT_TRUE(response_->complete());
     EXPECT_EQ("200", response_->headers().getStatusValue());
 
@@ -777,7 +777,7 @@ TEST_P(ExtAuthzLocalReplyIntegrationTest, DeniedHeaderTest) {
   };
   ext_authz_request->encodeHeaders(ext_authz_response_headers, true);
 
-  response->waitForEndStream();
+  ASSERT_TRUE(response->waitForEndStream());
   EXPECT_TRUE(response->complete());
 
   EXPECT_EQ("401", response->headers().Status()->value().getStringView());
@@ -853,7 +853,7 @@ TEST_P(ExtAuthzGrpcIntegrationTest, GoogleAsyncClientCreation) {
   upstream_request_->encodeHeaders(Http::TestResponseHeaderMapImpl{{":status", "200"}}, false);
   upstream_request_->encodeData(response_size_, true);
 
-  response_->waitForEndStream();
+  ASSERT_TRUE(response_->waitForEndStream());
 
   EXPECT_TRUE(upstream_request_->complete());
   EXPECT_EQ(request_body_.length(), upstream_request_->bodyLength());

test/extensions/filters/http/ext_proc/ext_proc_integration_test.cc

@@ -104,7 +104,7 @@ class ExtProcIntegrationTest : public HttpIntegrationTest,
   }
 
   void verifyDownstreamResponse(IntegrationStreamDecoder& response, int status_code) {
-    response.waitForEndStream();
+    ASSERT_TRUE(response.waitForEndStream());
     EXPECT_TRUE(response.complete());
     EXPECT_EQ(std::to_string(status_code), response.headers().getStatusValue());
   }