From 2358d959c6a5ef2c39982e4085bc02c8d1d1e607 Mon Sep 17 00:00:00 2001
From: Matt Mastracci <matthew@mastracci.com>
Date: Mon, 24 Feb 2025 13:58:12 -0700
Subject: [PATCH] Wait for further messages

---
 gel-tokio/src/raw/connection.rs | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/gel-tokio/src/raw/connection.rs b/gel-tokio/src/raw/connection.rs
index 48997e99..3cde1434 100644
--- a/gel-tokio/src/raw/connection.rs
+++ b/gel-tokio/src/raw/connection.rs
@@ -418,9 +418,10 @@ async fn connect4(cfg: &Config, mut stream: gel_stream::RawStream) -> Result<Con
                     "Unexpected authentication response",
                 )));
             }
-            ClientAuthResponse::Waiting | ClientAuthResponse::Complete => {
-                continue;
+            ClientAuthResponse::Complete => {
+                break;
             }
+            ClientAuthResponse::Waiting => {}
             ClientAuthResponse::Continue(message) => {
                 send_messages(
                     &mut stream,
@@ -438,6 +439,7 @@ async fn connect4(cfg: &Config, mut stream: gel_stream::RawStream) -> Result<Con
                 return Err(AuthenticationError::with_source(e));
             }
         }
+        msg = wait_message(&mut stream, &mut in_buf, &proto).await?;
     }
 
     let mut server_params = ServerParams::new();