Update test_charge_point.py

Author: drc38

tests/test_charge_point.py

@@ -44,15 +44,18 @@ async def test_cms_responses(hass):
         cp = ChargePoint("CP_1_test", ws)
         loop = True
         while loop:
-            await asyncio.gather(
+            out = await asyncio.gather(
                 cp.start(),
                 cp.send_boot_notification(),
                 cp.send_start_transaction(),
                 cp.send_meter_data(),
                 cp.send_stop_transaction(),
             )
-            await asyncio.sleep(10)
-            loop = False
+            if ws.closed:
+                ws = websockets.connect("ws://localhost:9000/CP_1", subprotocols=["ocpp1.6"], close_timeout = 10)
+            await asyncio.sleep(1)
+            if out.done():
+                loop = False
 
 class ChargePoint(cpclass):
     """Representation of real client Charge Point."""