fix: moved device_key to start_update

made more sense than init
Lash-L · Jan 18, 2023 · 1dddd7f · 1dddd7f
1 parent df8d522
commit 1dddd7f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/src/anova_wifi/parser.py b/src/anova_wifi/parser.py
@@ -1,5 +1,4 @@
 import logging
-from typing import Any
 
 import requests
 from sensor_state_data import SensorData, SensorDeviceClass, Units
@@ -32,14 +31,13 @@ class AnovaPrecisionCookerBinarySensor(StrEnum):
 
 
 class AnovaPrecisionCooker(SensorData):
-    def __init__(self, device_key: str):
+    def __init__(self) -> None:
         super().__init__()
-        self.device_key = device_key
 
-    def _start_update(self, data: Any) -> None:
+    def _start_update(self, device_key: str) -> None:
         anova_status = (
             requests.get(
-                f"https://anovaculinary.io/devices/{self.device_key}/states/?limit=1"
+                f"https://anovaculinary.io/devices/{device_key}/states/?limit=1"
             )
             .json()[0]
             .get("body")

diff --git a/tests/test_parser.py b/tests/test_parser.py
@@ -66,12 +66,12 @@
 
 
 def test_can_create():
-    AnovaPrecisionCooker("")
+    AnovaPrecisionCooker()
 
 
 @mock.patch("anova_wifi.parser.requests.get")
 def test_data_1(requests_mocked):
-    apc = AnovaPrecisionCooker("")
+    apc = AnovaPrecisionCooker()
     requests_mocked.return_value = MockResponse(json_data=dataset_one, status_code=200)
     result = apc.update("")
     print(result)