Don't make a long stream

hyperium · Jul 1, 2021 · e1e13a1 · e1e13a1
1 parent 9eaffb8
commit e1e13a1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/compression/src/lib.rs b/tests/compression/src/lib.rs
@@ -76,7 +76,9 @@ impl test_server::Test for Svc {
  _req: Request<()>,
  ) -> Result<Response<Self::CompressOutputServerStreamStream>, Status> {
  let data = [0_u8; UNCOMPRESSED_MIN_BODY_SIZE].to_vec();
- let stream = futures::stream::repeat(SomeData { data }).map(Ok::<_, Status>);
+ let stream = futures::stream::repeat(SomeData { data })
+ .take(2)
+ .map(Ok::<_, Status>);
  Ok(self.prepare_response(Response::new(Box::pin(stream))))
  }
 
@@ -120,7 +122,9 @@ impl test_server::Test for Svc {
  }
 
  let data = [0_u8; UNCOMPRESSED_MIN_BODY_SIZE].to_vec();
- let stream = futures::stream::repeat(SomeData { data }).map(Ok::<_, Status>);
+ let stream = futures::stream::repeat(SomeData { data })
+ .take(2)
+ .map(Ok::<_, Status>);
  Ok(self.prepare_response(Response::new(Box::pin(stream))))
  }
 }