From 4404faf1a9097de11c1395c46bdb9fc6b5ad998e Mon Sep 17 00:00:00 2001
From: Thisaru Guruge <thisaru@wso2.com>
Date: Mon, 27 Jan 2025 15:55:01 +0530
Subject: [PATCH] Add while loop back

---
 ballerina-tests/graphql-test-common/utils.bal | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/ballerina-tests/graphql-test-common/utils.bal b/ballerina-tests/graphql-test-common/utils.bal
index 708bec983..c715be6bc 100644
--- a/ballerina-tests/graphql-test-common/utils.bal
+++ b/ballerina-tests/graphql-test-common/utils.bal
@@ -111,9 +111,7 @@ public isolated function getContentFromByteStream(stream<byte[], io:Error?> byte
 
 public isolated function readMessageExcludingPingMessages(websocket:Client wsClient) returns json|error {
     json message = null;
-    int i = 0;
-    while i < 10 {
-        i = i + 1;
+    while true {
         message = check wsClient->readMessage();
         if message == null {
             continue;