123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778 |
- # switchbot-mqtt - MQTT client controlling SwitchBot button & curtain automators,
- # compatible with home-assistant.io's MQTT Switch & Cover platform
- #
- # Copyright (C) 2021 Fabian Peter Hammerle <fabian@hammerle.me>
- #
- # This program is free software: you can redistribute it and/or modify
- # it under the terms of the GNU General Public License as published by
- # the Free Software Foundation, either version 3 of the License, or
- # any later version.
- #
- # This program is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- # GNU General Public License for more details.
- #
- # You should have received a copy of the GNU General Public License
- # along with this program. If not, see <https://www.gnu.org/licenses/>.
- import os
- import re
- import unittest.mock
- import bluepy.btle
- import pytest
- import switchbot_mqtt
- # pylint: disable=protected-access
- def test__update_position_le_on_permission_denied():
- actor = switchbot_mqtt._CurtainMotor(
- mac_address="dummy", retry_count=21, password=None
- )
- with unittest.mock.patch(
- "switchbot.SwitchbotCurtain.update",
- side_effect=bluepy.btle.BTLEManagementError(
- "Failed to execute management command 'le on'",
- {
- "rsp": ["mgmt"],
- "code": ["mgmterr"],
- "estat": [20],
- "emsg": ["Permission Denied"],
- },
- ),
- ) as update_mock, unittest.mock.patch.object(
- actor, "_report_position"
- ) as report_position_mock, pytest.raises(
- PermissionError
- ) as exc_info:
- actor._update_position(mqtt_client="client")
- update_mock.assert_called_once_with()
- report_position_mock.assert_not_called()
- assert os.path.isfile(
- re.search(
- r"sudo setcap cap_net_admin\+ep (\S+/bluepy-helper)\b",
- exc_info.exconly(),
- ).group(1)
- )
- assert isinstance(exc_info.value.__cause__, bluepy.btle.BTLEManagementError)
- def test__update_position_other_error():
- actor = switchbot_mqtt._CurtainMotor(
- mac_address="dummy", retry_count=21, password=None
- )
- side_effect = bluepy.btle.BTLEManagementError("test")
- with unittest.mock.patch(
- "switchbot.SwitchbotCurtain.update", side_effect=side_effect
- ) as update_mock, unittest.mock.patch.object(
- actor, "_report_position"
- ) as report_position_mock, pytest.raises(
- type(side_effect)
- ) as exc_info:
- actor._update_position(mqtt_client="client")
- update_mock.assert_called_once_with()
- report_position_mock.assert_not_called()
- assert exc_info.value == side_effect
|