core/tests/components/dyson/test_init.py

"""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