From c7ebdbeda7198b24a0b271fba1c1eaf554f8e8fe Mon Sep 17 00:00:00 2001
From: Khor Shu Heng <32997938+khorshuheng@users.noreply.github.com>
Date: Wed, 10 Apr 2024 09:30:08 +0800
Subject: [PATCH] fix: handle receiver close for connection test (#462)

---
 tests/websocket/conn_test.rs | 20 ++++++++++++++------
 1 file changed, 14 insertions(+), 6 deletions(-)

diff --git a/tests/websocket/conn_test.rs b/tests/websocket/conn_test.rs
index fda05935..ee09826f 100644
--- a/tests/websocket/conn_test.rs
+++ b/tests/websocket/conn_test.rs
@@ -12,9 +12,13 @@ async fn realtime_connect_test() {
   let connect_info = c.ws_connect_info().await.unwrap();
   tokio::spawn(async move { ws_client.connect(&c.ws_url(), connect_info).await });
   let connect_future = async {
-    while let Ok(state) = state.recv().await {
-      if state == ConnectState::Connected {
-        break;
+    loop {
+      match state.recv().await {
+        Ok(ConnectState::Connected) => {
+          break;
+        },
+        Ok(_) => {},
+        Err(err) => panic!("Receiver Error: {:?}", err),
       }
     }
   };
@@ -42,9 +46,13 @@ async fn realtime_connect_after_token_exp_test() {
   tokio::spawn(async move { ws_client.connect(&c.ws_url(), connect_info).await });
 
   let connect_future = async {
-    while let Ok(state) = state.recv().await {
-      if state == ConnectState::Connected {
-        break;
+    loop {
+      match state.recv().await {
+        Ok(ConnectState::Connected) => {
+          break;
+        },
+        Ok(_) => {},
+        Err(err) => panic!("Receiver Error: {:?}", err),
       }
     }
   };