hass-core/tests/components/test_dyson.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)
    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)
    return device


class DysonTest(unittest.TestCase):
    """Dyson parent component test class."""

    def setUp(self):  # pylint: disable=invalid-name
        """Setup 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('libpurecoollink.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"
        }})
        self.assertEqual(mocked_login.call_count, 1)

    @mock.patch('libpurecoollink.dyson.DysonAccount.devices', return_value=[])
    @mock.patch('libpurecoollink.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"
        }})
        self.assertEqual(mocked_login.call_count, 1)
        self.assertEqual(mocked_devices.call_count, 1)
        self.assertEqual(len(self.hass.data[dyson.DYSON_DEVICES]), 0)

    @mock.patch('homeassistant.helpers.discovery.load_platform')
    @mock.patch('libpurecoollink.dyson.DysonAccount.devices',
                return_value=[_get_dyson_account_device_available()])
    @mock.patch('libpurecoollink.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"
                }
            ]
        }})
        self.assertEqual(mocked_login.call_count, 1)
        self.assertEqual(mocked_devices.call_count, 1)
        self.assertEqual(len(self.hass.data[dyson.DYSON_DEVICES]), 1)
        self.assertEqual(mocked_discovery.call_count, 2)

    @mock.patch('libpurecoollink.dyson.DysonAccount.devices',
                return_value=[_get_dyson_account_device_not_available()])
    @mock.patch('libpurecoollink.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"
                }
            ]
        }})
        self.assertEqual(mocked_login.call_count, 1)
        self.assertEqual(mocked_devices.call_count, 1)
        self.assertEqual(len(self.hass.data[dyson.DYSON_DEVICES]), 0)

    @mock.patch('homeassistant.helpers.discovery.load_platform')
    @mock.patch('libpurecoollink.dyson.DysonAccount.devices',
                return_value=[_get_dyson_account_device_available()])
    @mock.patch('libpurecoollink.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"
                }
            ]
        }})
        self.assertEqual(mocked_login.call_count, 1)
        self.assertEqual(mocked_devices.call_count, 1)
        self.assertEqual(len(self.hass.data[dyson.DYSON_DEVICES]), 0)
        self.assertEqual(mocked_discovery.call_count, 0)

    @mock.patch('homeassistant.helpers.discovery.load_platform')
    @mock.patch('libpurecoollink.dyson.DysonAccount.devices',
                return_value=[_get_dyson_account_device_available()])
    @mock.patch('libpurecoollink.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
        }})
        self.assertEqual(mocked_login.call_count, 1)
        self.assertEqual(mocked_devices.call_count, 1)
        self.assertEqual(len(self.hass.data[dyson.DYSON_DEVICES]), 1)
        self.assertEqual(mocked_discovery.call_count, 2)

    @mock.patch('libpurecoollink.dyson.DysonAccount.devices',
                return_value=[_get_dyson_account_device_not_available()])
    @mock.patch('libpurecoollink.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
        }})
        self.assertEqual(mocked_login.call_count, 1)
        self.assertEqual(mocked_devices.call_count, 1)
        self.assertEqual(len(self.hass.data[dyson.DYSON_DEVICES]), 0)