c

DorosMichal · May 2, 2023 · 96bde54 · 96bde54
1 parent 8e03d04
commit 96bde54
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/e2e_tests/komfy_database.py b/e2e_tests/komfy_database.py
@@ -90,10 +90,9 @@ def execute(
         QUERY_DONE_FLAG = False
         self.write(query)
         responses = []
-        while not QUERY_DONE_FLAG:
-            while (response := self.readline()):
-                if response:
-                    responses.append(response)
+        while not QUERY_DONE_FLAG or (response := self.readline()):
+            if response:
+                responses.append(response)
 
         if ignore_response:
             return