Fix tests

sdb9696 · sdb9696 · commit e94cbb5572b1 · 2024-05-10T12:58:55.000+01:00
diff --git a/kasa/smart/modules/lightpresetmodule.py b/kasa/smart/modules/lightpresetmodule.py
@@ -46,7 +46,7 @@ def _handle_update(self):
         index = 0
         self._presets = {self.PRESET_OFF: None}
 
-        if preset_states := self.data["states"]:
+        if preset_states := self.data.get("states"):
             for preset_state in preset_states:
                 self._presets[f"Light preset {index + 1}"] = BulbPreset(
                     index=index,
@@ -56,11 +56,11 @@ def _handle_update(self):
                     saturation=preset_state.get("saturation"),
                 )
                 index = index + 1
-        elif preset_brightnesses := self.data["brightness"]:
+        elif preset_brightnesses := self.data.get("brightness"):
             for preset_brightness in preset_brightnesses:
                 self._presets[f"Brightness preset {index + 1}"] = BulbPreset(
                     index=index,
-                    brightness=preset_brightness["brightness"],
+                    brightness=preset_brightness,
                 )
                 index = index + 1
 
