From 22c969acc900d2b5ecc0ecd6b7ec2c16332589ec Mon Sep 17 00:00:00 2001
From: Joannis Orlandos <joannis@orlandos.nl>
Date: Mon, 4 Mar 2024 10:48:42 +0100
Subject: [PATCH] Make sure that data is buffered instead of discarded

---
 Sources/Citadel/TTY/Client/TTY.swift | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Sources/Citadel/TTY/Client/TTY.swift b/Sources/Citadel/TTY/Client/TTY.swift
index 0ec9086..d4cc01b 100644
--- a/Sources/Citadel/TTY/Client/TTY.swift
+++ b/Sources/Citadel/TTY/Client/TTY.swift
@@ -155,7 +155,7 @@ extension SSHClient {
     /// - inShell:  Whether to request the remote server to start a shell before executing the command.
     public func executeCommandStream(_ command: String, inShell: Bool = false) async throws -> AsyncThrowingStream<ExecCommandOutput, Error> {
         var streamContinuation: AsyncThrowingStream<ExecCommandOutput, Error>.Continuation!
-        let stream = AsyncThrowingStream<ExecCommandOutput, Error> { continuation in
+        let stream = AsyncThrowingStream<ExecCommandOutput, Error>(bufferingPolicy: .unbounded) { continuation in
             streamContinuation = continuation
         }
         
@@ -218,11 +218,11 @@ extension SSHClient {
     public func executeCommandPair(_ command: String, inShell: Bool = false) async throws -> ExecCommandStream {
         var stdoutContinuation: AsyncThrowingStream<ByteBuffer, Error>.Continuation!
         var stderrContinuation: AsyncThrowingStream<ByteBuffer, Error>.Continuation!
-        let stdout = AsyncThrowingStream<ByteBuffer, Error> { continuation in
+        let stdout = AsyncThrowingStream<ByteBuffer, Error>(bufferingPolicy: .unbounded) { continuation in
             stdoutContinuation = continuation
         }
         
-        let stderr = AsyncThrowingStream<ByteBuffer, Error> { continuation in
+        let stderr = AsyncThrowingStream<ByteBuffer, Error>(bufferingPolicy: .unbounded) { continuation in
             stderrContinuation = continuation
         }