1d70005b01
fix check for already created entities remove hepa and carbon filter add AQI attribute initial commit fix check for already created entities remove hepa and carbon filter add AQI attribute add air quality component tests fix method call tests fix line lengths fix pylint issues fix docstrings revert fan related changes remove whitespace change fix fan update state test add for loop for platform initialization add requested changes to aiq platform change string concatenation to new style string formatting update air quality tests update air quality tests refactor sensor component changes fix pylint issues fix debug string in the air quality component replace failing tests for older devices fix line length fan tests remove dependencies const and move imports move back imports to methods remove whitespace from blank line
192 lines
7.9 KiB
Python
192 lines
7.9 KiB
Python
"""Test the parent Dyson component."""
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
from homeassistant.components import dyson
|
|
from tests.common import get_test_home_assistant
|
|
|
|
|
|
def _get_dyson_account_device_available():
|
|
"""Return a valid device provide by Dyson web services."""
|
|
device = mock.Mock()
|
|
device.serial = "XX-XXXXX-XX"
|
|
device.connect = mock.Mock(return_value=True)
|
|
device.auto_connect = mock.Mock(return_value=True)
|
|
return device
|
|
|
|
|
|
def _get_dyson_account_device_not_available():
|
|
"""Return an invalid device provide by Dyson web services."""
|
|
device = mock.Mock()
|
|
device.serial = "XX-XXXXX-XX"
|
|
device.connect = mock.Mock(return_value=False)
|
|
device.auto_connect = mock.Mock(return_value=False)
|
|
return device
|
|
|
|
|
|
def _get_dyson_account_device_error():
|
|
"""Return an invalid device raising OSError while connecting."""
|
|
device = mock.Mock()
|
|
device.serial = "XX-XXXXX-XX"
|
|
device.connect = mock.Mock(side_effect=OSError("Network error"))
|
|
return device
|
|
|
|
|
|
class DysonTest(unittest.TestCase):
|
|
"""Dyson parent component test class."""
|
|
|
|
def setUp(self): # pylint: disable=invalid-name
|
|
"""Set up things to be run when tests are started."""
|
|
self.hass = get_test_home_assistant()
|
|
|
|
def tearDown(self): # pylint: disable=invalid-name
|
|
"""Stop everything that was started."""
|
|
self.hass.stop()
|
|
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=False)
|
|
def test_dyson_login_failed(self, mocked_login):
|
|
"""Test if Dyson connection failed."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR"
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
|
|
@mock.patch('libpurecool.dyson.DysonAccount.devices', return_value=[])
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=True)
|
|
def test_dyson_login(self, mocked_login, mocked_devices):
|
|
"""Test valid connection to dyson web service."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR"
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
assert mocked_devices.call_count == 1
|
|
assert len(self.hass.data[dyson.DYSON_DEVICES]) == 0
|
|
|
|
@mock.patch('homeassistant.helpers.discovery.load_platform')
|
|
@mock.patch('libpurecool.dyson.DysonAccount.devices',
|
|
return_value=[_get_dyson_account_device_available()])
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=True)
|
|
def test_dyson_custom_conf(self, mocked_login, mocked_devices,
|
|
mocked_discovery):
|
|
"""Test device connection using custom configuration."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR",
|
|
dyson.CONF_DEVICES: [
|
|
{
|
|
"device_id": "XX-XXXXX-XX",
|
|
"device_ip": "192.168.0.1"
|
|
}
|
|
]
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
assert mocked_devices.call_count == 1
|
|
assert len(self.hass.data[dyson.DYSON_DEVICES]) == 1
|
|
assert mocked_discovery.call_count == 5
|
|
|
|
@mock.patch('libpurecool.dyson.DysonAccount.devices',
|
|
return_value=[_get_dyson_account_device_not_available()])
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=True)
|
|
def test_dyson_custom_conf_device_not_available(self, mocked_login,
|
|
mocked_devices):
|
|
"""Test device connection with an invalid device."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR",
|
|
dyson.CONF_DEVICES: [
|
|
{
|
|
"device_id": "XX-XXXXX-XX",
|
|
"device_ip": "192.168.0.1"
|
|
}
|
|
]
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
assert mocked_devices.call_count == 1
|
|
assert len(self.hass.data[dyson.DYSON_DEVICES]) == 0
|
|
|
|
@mock.patch('libpurecool.dyson.DysonAccount.devices',
|
|
return_value=[_get_dyson_account_device_error()])
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=True)
|
|
def test_dyson_custom_conf_device_error(self, mocked_login,
|
|
mocked_devices):
|
|
"""Test device connection with device raising an exception."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR",
|
|
dyson.CONF_DEVICES: [
|
|
{
|
|
"device_id": "XX-XXXXX-XX",
|
|
"device_ip": "192.168.0.1"
|
|
}
|
|
]
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
assert mocked_devices.call_count == 1
|
|
assert len(self.hass.data[dyson.DYSON_DEVICES]) == 0
|
|
|
|
@mock.patch('homeassistant.helpers.discovery.load_platform')
|
|
@mock.patch('libpurecool.dyson.DysonAccount.devices',
|
|
return_value=[_get_dyson_account_device_available()])
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=True)
|
|
def test_dyson_custom_conf_with_unknown_device(self, mocked_login,
|
|
mocked_devices,
|
|
mocked_discovery):
|
|
"""Test device connection with custom conf and unknown device."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR",
|
|
dyson.CONF_DEVICES: [
|
|
{
|
|
"device_id": "XX-XXXXX-XY",
|
|
"device_ip": "192.168.0.1"
|
|
}
|
|
]
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
assert mocked_devices.call_count == 1
|
|
assert len(self.hass.data[dyson.DYSON_DEVICES]) == 0
|
|
assert mocked_discovery.call_count == 0
|
|
|
|
@mock.patch('homeassistant.helpers.discovery.load_platform')
|
|
@mock.patch('libpurecool.dyson.DysonAccount.devices',
|
|
return_value=[_get_dyson_account_device_available()])
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=True)
|
|
def test_dyson_discovery(self, mocked_login, mocked_devices,
|
|
mocked_discovery):
|
|
"""Test device connection using discovery."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR",
|
|
dyson.CONF_TIMEOUT: 5,
|
|
dyson.CONF_RETRY: 2
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
assert mocked_devices.call_count == 1
|
|
assert len(self.hass.data[dyson.DYSON_DEVICES]) == 1
|
|
assert mocked_discovery.call_count == 5
|
|
|
|
@mock.patch('libpurecool.dyson.DysonAccount.devices',
|
|
return_value=[_get_dyson_account_device_not_available()])
|
|
@mock.patch('libpurecool.dyson.DysonAccount.login', return_value=True)
|
|
def test_dyson_discovery_device_not_available(self, mocked_login,
|
|
mocked_devices):
|
|
"""Test device connection with discovery and invalid device."""
|
|
dyson.setup(self.hass, {dyson.DOMAIN: {
|
|
dyson.CONF_USERNAME: "email",
|
|
dyson.CONF_PASSWORD: "password",
|
|
dyson.CONF_LANGUAGE: "FR",
|
|
dyson.CONF_TIMEOUT: 5,
|
|
dyson.CONF_RETRY: 2
|
|
}})
|
|
assert mocked_login.call_count == 1
|
|
assert mocked_devices.call_count == 1
|
|
assert len(self.hass.data[dyson.DYSON_DEVICES]) == 0
|