core/tests/components/sensor/test_mhz19.py

"""Tests for MH-Z19 sensor."""
import unittest
from unittest.mock import patch, DEFAULT, Mock

from homeassistant.setup import setup_component
from homeassistant.components.sensor import DOMAIN
import homeassistant.components.sensor.mhz19 as mhz19
from homeassistant.const import TEMP_FAHRENHEIT
from tests.common import get_test_home_assistant, assert_setup_component


class TestMHZ19Sensor(unittest.TestCase):
    """Test the MH-Z19 sensor."""

    hass = None

    def setup_method(self, method):
        """Set up things to be run when tests are started."""
        self.hass = get_test_home_assistant()

    def teardown_method(self, method):
        """Stop everything that was started."""
        self.hass.stop()

    def test_setup_missing_config(self):
        """Test setup with configuration missing required entries."""
        with assert_setup_component(0):
            assert setup_component(self.hass, DOMAIN, {
                'sensor': {'platform': 'mhz19'}})

    @patch('pmsensor.co2sensor.read_mh_z19', side_effect=OSError('test error'))
    def test_setup_failed_connect(self, mock_co2):
        """Test setup when connection error occurs."""
        assert not mhz19.setup_platform(self.hass, {
            'platform': 'mhz19',
            mhz19.CONF_SERIAL_DEVICE: 'test.serial',
            }, None)

    def test_setup_connected(self):
        """Test setup when connection succeeds."""
        with patch.multiple('pmsensor.co2sensor', read_mh_z19=DEFAULT,
                            read_mh_z19_with_temperature=DEFAULT):
            from pmsensor.co2sensor import read_mh_z19_with_temperature
            read_mh_z19_with_temperature.return_value = None
            mock_add = Mock()
            assert mhz19.setup_platform(self.hass, {
                'platform': 'mhz19',
                'monitored_conditions': ['co2', 'temperature'],
                mhz19.CONF_SERIAL_DEVICE: 'test.serial',
                }, mock_add)
        assert 1 == mock_add.call_count

    @patch('pmsensor.co2sensor.read_mh_z19_with_temperature',
           side_effect=OSError('test error'))
    def aiohttp_client_update_oserror(self, mock_function):
        """Test MHZClient when library throws OSError."""
        from pmsensor import co2sensor
        client = mhz19.MHZClient(co2sensor, 'test.serial')
        client.update()
        assert {} == client.data

    @patch('pmsensor.co2sensor.read_mh_z19_with_temperature',
           return_value=(5001, 24))
    def aiohttp_client_update_ppm_overflow(self, mock_function):
        """Test MHZClient when ppm is too high."""
        from pmsensor import co2sensor
        client = mhz19.MHZClient(co2sensor, 'test.serial')
        client.update()
        assert client.data.get('co2') is None

    @patch('pmsensor.co2sensor.read_mh_z19_with_temperature',
           return_value=(1000, 24))
    def aiohttp_client_update_good_read(self, mock_function):
        """Test MHZClient when ppm is too high."""
        from pmsensor import co2sensor
        client = mhz19.MHZClient(co2sensor, 'test.serial')
        client.update()
        assert {'temperature': 24, 'co2': 1000} == client.data

    @patch('pmsensor.co2sensor.read_mh_z19_with_temperature',
           return_value=(1000, 24))
    def test_co2_sensor(self, mock_function):
        """Test CO2 sensor."""
        from pmsensor import co2sensor
        client = mhz19.MHZClient(co2sensor, 'test.serial')
        sensor = mhz19.MHZ19Sensor(client, mhz19.SENSOR_CO2, None, 'name')
        sensor.update()

        assert 'name: CO2' == sensor.name
        assert 1000 == sensor.state
        assert 'ppm' == sensor.unit_of_measurement
        assert sensor.should_poll
        assert {'temperature': 24} == sensor.device_state_attributes

    @patch('pmsensor.co2sensor.read_mh_z19_with_temperature',
           return_value=(1000, 24))
    def test_temperature_sensor(self, mock_function):
        """Test temperature sensor."""
        from pmsensor import co2sensor
        client = mhz19.MHZClient(co2sensor, 'test.serial')
        sensor = mhz19.MHZ19Sensor(
            client, mhz19.SENSOR_TEMPERATURE, None, 'name')
        sensor.update()

        assert 'name: Temperature' == sensor.name
        assert 24 == sensor.state
        assert '°C' == sensor.unit_of_measurement
        assert sensor.should_poll
        assert {'co2_concentration': 1000} == sensor.device_state_attributes

    @patch('pmsensor.co2sensor.read_mh_z19_with_temperature',
           return_value=(1000, 24))
    def test_temperature_sensor_f(self, mock_function):
        """Test temperature sensor."""
        from pmsensor import co2sensor
        client = mhz19.MHZClient(co2sensor, 'test.serial')
        sensor = mhz19.MHZ19Sensor(
            client, mhz19.SENSOR_TEMPERATURE, TEMP_FAHRENHEIT, 'name')
        sensor.update()

        assert 75.2 == sensor.state