Get MyStrom device state before checking support #96004
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1 +1,172 @@ | ||
| """Tests for the myStrom integration.""" | ||
| from typing import Any, Optional | ||
|
|
||
|
|
||
| def get_default_device_response(device_type: int) -> dict[str, Any]: | ||
| """Return default device response.""" | ||
| return { | ||
| "version": "2.59.32", | ||
| "mac": "6001940376EB", | ||
| "type": device_type, | ||
| "ssid": "personal", | ||
| "ip": "192.168.0.23", | ||
| "mask": "255.255.255.0", | ||
| "gw": "192.168.0.1", | ||
| "dns": "192.168.0.1", | ||
| "static": False, | ||
| "connected": True, | ||
| "signal": 94, | ||
| } | ||
|
|
||
|
|
||
| def get_default_bulb_state() -> dict[str, Any]: | ||
| """Get default bulb state.""" | ||
| return { | ||
| "type": "rgblamp", | ||
| "battery": False, | ||
| "reachable": True, | ||
| "meshroot": True, | ||
| "on": False, | ||
| "color": "46;18;100", | ||
| "mode": "hsv", | ||
| "ramp": 10, | ||
| "power": 0.45, | ||
| "fw_version": "2.58.0", | ||
| } | ||
|
|
||
|
|
||
| def get_default_switch_state() -> dict[str, Any]: | ||
| """Get default switch state.""" | ||
| return { | ||
| "power": 1.69, | ||
| "Ws": 0.81, | ||
| "relay": True, | ||
| "temperature": 24.87, | ||
| "version": "2.59.32", | ||
| "mac": "6001940376EB", | ||
| "ssid": "personal", | ||
| "ip": "192.168.0.23", | ||
| "mask": "255.255.255.0", | ||
| "gw": "192.168.0.1", | ||
| "dns": "192.168.0.1", | ||
| "static": False, | ||
| "connected": True, | ||
| "signal": 94, | ||
| } | ||
|
|
||
|
|
||
| class MyStromDeviceMock: | ||
| """Base device mock.""" | ||
|
|
||
| def __init__(self, state: dict[str, Any]) -> None: | ||
| """Initialize device mock.""" | ||
| self._requested_state = False | ||
| self._state = state | ||
|
|
||
| async def get_state(self) -> None: | ||
| """Set if state is requested.""" | ||
| self._requested_state = True | ||
|
|
||
|
|
||
| class MyStromBulbMock(MyStromDeviceMock): | ||
| """MyStrom Bulb mock.""" | ||
|
|
||
| def __init__(self, mac: str, state: dict[str, Any]) -> None: | ||
| """Initialize bulb mock.""" | ||
| super().__init__(state) | ||
| self.mac = mac | ||
|
|
||
| @property | ||
| def firmware(self) -> Optional[str]: | ||
| """Return current firmware.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["fw_version"] | ||
|
|
||
| @property | ||
| def consumption(self) -> Optional[float]: | ||
| """Return current firmware.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["power"] | ||
|
|
||
| @property | ||
| def color(self) -> Optional[str]: | ||
| """Return current color settings.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["color"] | ||
|
|
||
| @property | ||
| def mode(self) -> Optional[str]: | ||
| """Return current mode.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["mode"] | ||
|
|
||
| @property | ||
| def transition_time(self) -> Optional[int]: | ||
| """Return current transition time (ramp).""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["ramp"] | ||
|
|
||
| @property | ||
| def bulb_type(self) -> Optional[str]: | ||
| """Return the type of the bulb.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["type"] | ||
|
|
||
| @property | ||
| def state(self) -> Optional[bool]: | ||
| """Return the current state of the bulb.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["on"] | ||
|
|
||
|
|
||
| class MyStromSwitchMock(MyStromDeviceMock): | ||
| """MyStrom Switch mock.""" | ||
|
|
||
| @property | ||
| def relay(self) -> Optional[bool]: | ||
| """Return the relay state.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["on"] | ||
|
|
||
| @property | ||
| def consumption(self) -> Optional[float]: | ||
| """Return the current power consumption in mWh.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["power"] | ||
|
|
||
| @property | ||
| def consumedWs(self) -> Optional[float]: | ||
| """The average of energy consumed per second since last report call.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["Ws"] | ||
|
|
||
| @property | ||
| def firmware(self) -> Optional[str]: | ||
| """Return the current firmware.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["version"] | ||
|
|
||
| @property | ||
| def mac(self) -> Optional[str]: | ||
| """Return the MAC address.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["mac"] | ||
|
|
||
| @property | ||
| def temperature(self) -> Optional[float]: | ||
| """Return the current temperature in celsius.""" | ||
| if not self._requested_state: | ||
| return None | ||
| return self._state["temperature"] |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I currently use this method to be able to inject mocks in the tests. Any other method would be preferred!
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
why do you need to make it a function instead of calling constructor right away like it was ?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Because I want to inject a mock in the tests and I could not figure out how to patch a constructor call. (the mock is now used to mock the behaviour of the external library better, since the current tests arent representing this well)