fix: handle websocket disconnects better

custom_components/openevse/__init__.py

@@ -182,9 +182,6 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
     )
 
     await coordinator.async_refresh()
-    # Start the websocket listener
-    manager.ws_start()
-
     await hass.config_entries.async_forward_entry_setups(config_entry, PLATFORMS)
 
     services = OpenEVSEServices(hass, config_entry)
@@ -323,6 +320,26 @@ async def update_sensors(self) -> dict:
             )
             raise UpdateFailed(error) from error
 
+        try:
+            assert self._manager.websocket
+        except AssertionError as error:
+            _LOGGER.debug("Websocket not setup.")
+            raise UpdateFailed(error) from error
+
+        if self._manager.websocket.state != "connected":
+            _LOGGER.info("Connecting to websocket...")
+            try:
+                self._manager.ws_start()
+            except RuntimeError:
+                pass
+            except Exception as error:
+                _LOGGER.debug(
+                    "Error connecting to websocket [%s]: %s",
+                    type(error).__name__,
+                    error,
+                )
+                raise UpdateFailed(error) from error
+
         self.parse_sensors()
         await self.async_parse_sensors()
         _LOGGER.debug("Coordinator data: %s", self._data)

custom_components/openevse/manifest.json

@@ -9,7 +9,7 @@
   "iot_class": "local_push",
   "issue_tracker": "https://github.com/firstof9/openevse/issues",
   "loggers": ["openevsehttp"],
-  "requirements": ["python-openevse-http==0.1.65"],
+  "requirements": ["python_openevse_http==0.1.70"],
   "version": "0.0.0-dev",
   "zeroconf": ["_openevse._tcp.local."]
 }

tests/conftest.py

@@ -98,7 +98,7 @@ def test_charger(mock_aioclient):
     )
     mock_aioclient.get(
         TEST_URL_WS,
-        status=200,
+        status=101,
         body=load_fixture("status.json"),
         repeat=True,
     )
@@ -140,7 +140,7 @@ def test_charger_bad_serial(mock_aioclient):
     )
     mock_aioclient.get(
         TEST_URL_WS,
-        status=200,
+        status=101,
         body=load_fixture("status.json"),
         repeat=True,
     )
@@ -181,7 +181,7 @@ def test_charger_bad_post(mock_aioclient):
     )
     mock_aioclient.get(
         TEST_URL_WS,
-        status=200,
+        status=101,
         body=load_fixture("status.json"),
         repeat=True,
     )

tests/test_services.py

@@ -108,7 +108,7 @@ async def test_make_claim(
         data=CONFIG_DATA,
     )
     mock_aioclient.post(
-        f"{TEST_URL_CLAIMS}/4",
+        f"{TEST_URL_CLAIMS}/20",
         status=200,
         body='[{"msg":"done"}]',
         repeat=True,
@@ -150,7 +150,7 @@ async def test_release_claim(
         data=CONFIG_DATA,
     )
     mock_aioclient.delete(
-        f"{TEST_URL_CLAIMS}/4",
+        f"{TEST_URL_CLAIMS}/20",
         status=200,
         body='[{"msg":"done"}]',
         repeat=True,