Add tests for homewizard_energy (#12)

homewizard_energy/homewizard_energy.py

@@ -20,11 +20,14 @@ class HomeWizardEnergy:
 
     _session: ClientSession | None
     _close_session: bool = False
+    _request_timeout: int = 10
 
     _detected_product_type: str | None = None
     _detected_api_version: str | None = None
 
-    def __init__(self, host: str, clientsession: ClientSession = None):
+    def __init__(
+        self, host: str, clientsession: ClientSession = None, timeout: int = 10
+    ):
         """Create a HomeWizard Energy object.
 
         Args:
@@ -34,6 +37,7 @@ def __init__(self, host: str, clientsession: ClientSession = None):
 
         self._host = host
         self._session = clientsession
+        self._request_timeout = timeout
 
     @property
     def host(self) -> str:
@@ -65,11 +69,6 @@ async def data(self) -> Data:
         if not self._detected_api_version:
             await self.device()
 
-        if self._detected_api_version != SUPPORTED_API_VERSION:
-            raise UnsupportedError(
-                f"Unsupported API version, detected {self._detected_api_version}"
-            )
-
         response = await self.request("api/v1/data")
         return Data.from_dict(response)
 
@@ -81,12 +80,6 @@ async def state(self) -> State | None:
         if self._detected_product_type not in DEVICES_WITH_STATE:
             return None
 
-        if self._detected_api_version != SUPPORTED_API_VERSION:
-            raise UnsupportedError(
-                f"Unsupported device or API version, "
-                f"detected API:{self._detected_api_version} with {self._detected_product_type}"
-            )
-
         response = await self.request("api/v1/state")
         return State.from_dict(response)
 
@@ -110,9 +103,8 @@ async def state_set(
             _LOGGER.error("At least one state update is required")
             return False
 
-        response = await self.request("api/v1/state", method=METH_PUT, data=state)
-        if response is not None:
-            return True
+        await self.request("api/v1/state", method=METH_PUT, data=state)
+        return True
 
     async def request(
         self, path: str, method: str = METH_GET, data: object = None
@@ -128,7 +120,7 @@ async def request(
         _LOGGER.debug("%s, %s, %s", method, url, data)
 
         try:
-            async with async_timeout.timeout(8):
+            async with async_timeout.timeout(self._request_timeout):
                 resp = await self._session.request(
                     method,
                     url,
@@ -156,10 +148,10 @@ async def request(
             raise RequestError(f"API request error ({resp.status})")
 
         content_type = resp.headers.get("Content-Type", "")
-        if "application/json" not in content_type:
-            raise RequestError("Unexpected content type")
+        if "application/json" in content_type:
+            return await resp.json()
 
-        return await resp.json()
+        return await resp.text()
 
     async def close(self):
         """Close client session."""

tests/fixtures/device_energysocket.json

@@ -0,0 +1,7 @@
+{
+    "product_type": "HWE-SKT",
+    "product_name": "Energy Socket",
+    "serial": "3c39e7aabbcc",
+    "firmware_version": "2.11",
+    "api_version": "v1"
+}