Adjust has_v2_api so test does not trigger 'Was never awaited' error

homewizard_energy/__init__.py

@@ -27,10 +27,10 @@ async def has_v2_api(host: str, websession: ClientSession | None = None) -> bool
         # v2 api is https only and returns a 401 Unauthorized when no key provided,
         # no connection can be made if the device is not v2
         url = f"https://{host}/api"
-        async with websession.get(
-            url, ssl=False, raise_for_status=False, timeout=15
-        ) as res:
-            return res.status == 401
+        res = await websession.get(url, ssl=False, raise_for_status=False, timeout=5)
+        res.close()
+
+        return res.status == 401
     except Exception:  # pylint: disable=broad-except
         # all other status/exceptions means the device is not v2 or not reachable at this time
         return False

tests/test_helpers.py

@@ -47,7 +47,7 @@ async def test_has_v2_api_false(aresponses):
 async def test_has_v2_api_exception():
     """Test if has_v2_api returns False when an exception occurs."""
     session = AsyncMock()
-    session.request = AsyncMock(side_effect=asyncio.TimeoutError())
+    session.get = AsyncMock(side_effect=asyncio.TimeoutError)
 
     result = await has_v2_api("example.com", session)
     assert result is False