Merge #166

166: Fix: Read from connection until no more bytes are returned r=TheNeikos a=matthiasbeyer

Extracted from #133 

Co-authored-by: Matthias Beyer <mail@beyermatthias.de>

1 files changed, 12 insertions, 4 deletions

diff --git a/mqtt-tester/src/command.rs b/mqtt-tester/src/command.rs
index 18c143d..e0b531a 100644
--- a/mqtt-tester/src/command.rs
+++ b/mqtt-tester/src/command.rs
@@ -117,10 +117,18 @@ impl Output {
                 tracing::trace!("Rest-Len: {}", rest_len);
 
                 let mut rest_buf = buffer.limit(rest_len as usize);
-                self.stdout
-                    .read_buf(&mut rest_buf)
-                    .await
-                    .into_diagnostic()?;
+                loop {
+                    let read_len = self
+                        .stdout
+                        .read_buf(&mut rest_buf)
+                        .await
+                        .into_diagnostic()?;
+
+                    tracing::trace!("Read {} bytes", read_len);
+                    if read_len == 0 {
+                        break;
+                    }
+                }
                 Ok::<_, miette::Error>(rest_buf.into_inner())
             }
             .instrument(tracing::trace_span!("Reading bytes from connection")),