122 lines
4.8 KiB
Python
122 lines
4.8 KiB
Python
"""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.mhz19.sensor 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
|