socket_comms: wait for connection properly

Don't return early if `self.receive` returns NULL.

Signed-off-by: Jordan Yates <jordan@embeint.com>

Author: JordanYates

src/infuse_iot/socket_comms.py

@@ -242,12 +242,12 @@ def connection_create(self, infuse_id: int, data_types: GatewayRequestConnection
         req = GatewayRequestConnectionRequest(infuse_id, data_types)
         self.send(req)
         # Wait for response from the server
-        while rsp := self.receive():
-            if isinstance(rsp, ClientNotificationConnectionCreated):
-                return rsp.max_payload
-            elif isinstance(rsp, ClientNotificationConnectionFailed):
-                raise ConnectionRefusedError
-        raise ConnectionRefusedError
+        while True:
+            if rsp := self.receive():
+                if isinstance(rsp, ClientNotificationConnectionCreated):
+                    return rsp.max_payload
+                elif isinstance(rsp, ClientNotificationConnectionFailed):
+                    raise ConnectionRefusedError
 
     def connection_release(self):
         assert self._connection_id is not None