Fix switch toggles

custom_components/bhyve/__init__.py

@@ -133,7 +133,7 @@ def __init__(
         self, bhyve, device, name, icon, device_class=None,
     ):
         """Initialize the sensor."""
-        self._bhyve: BHyve = bhyve
+        self._bhyve: Client = bhyve
         self._device_class = device_class
         self._async_unsub_dispatcher_connect = None
 

custom_components/bhyve/pybhyve/client.py

@@ -60,7 +60,7 @@ async def _request(self, method: str, endpoint: str, params: dict = None) -> lis
             try:
                 resp.raise_for_status()
                 return await resp.json(content_type=None)
-            except ClientError as err:
+            except Exception as err:
                 raise RequestError(f"Error requesting data from {url}: {err}")
 
     async def _refresh_devices(self):
@@ -96,7 +96,7 @@ async def login(self) -> bool:
                 _LOGGER.debug("Logged in")
                 self._token = response["orbit_session_token"]
 
-            except ClientError as err:
+            except Exception as err:
                 raise RequestError(f"Error requesting data from {url}: {err}")
 
         if self._token is None:
@@ -130,5 +130,5 @@ async def get_device(self, device_id):
                 return device
         return None
 
-    def send_message(self, payload):
-        self._websocket.send(payload)
+    async def send_message(self, payload):
+        await self._websocket.send(payload)

custom_components/bhyve/switch.py

@@ -139,12 +139,12 @@ async def _send_station_message(self, station_payload):
             await self._bhyve.send_message(payload)
 
         except BHyveError as err:
-            _LOGGER.warning("Failed to connect to BHyve servers. %s", err)
+            _LOGGER.warning("Failed to send to BHyve websocket message %s", err)
             raise (err)
 
     async def async_turn_on(self, **kwargs):
         """Turn the switch on."""
-        station_payload = [{"station": self._zone_id, "run_time": 1.0}]
+        station_payload = [{"station": self._zone_id, "run_time": 10.0}]
         await self._send_station_message(station_payload)
 
     async def async_turn_off(self, **kwargs):