core/tests/components/sensor/test_yr.py

"""
tests.components.sensor.test_yr
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Tests Yr sensor.
"""
import unittest

import homeassistant.core as ha
import homeassistant.components.sensor as sensor


class TestSensorYr(unittest.TestCase):
    """ Test the Yr sensor. """

    def setUp(self):  # pylint: disable=invalid-name
        self.hass = ha.HomeAssistant()
        self.hass.config.latitude = 32.87336
        self.hass.config.longitude = 117.22743

    def tearDown(self):  # pylint: disable=invalid-name
        """ Stop down stuff we started. """
        self.hass.stop()

    def test_default_setup(self):
        self.assertTrue(sensor.setup(self.hass, {
            'sensor': {
                'platform': 'yr',
                'elevation': 0,
            }
        }))
        state = self.hass.states.get('sensor.yr_symbol')

        self.assertTrue(state.state.isnumeric())
        self.assertEqual(None,
                         state.attributes.get('unit_of_measurement'))

    def test_custom_setup(self):
        self.assertTrue(sensor.setup(self.hass, {
            'sensor': {
                'platform': 'yr',
                'elevation': 0,
                'monitored_conditions': {
                    'pressure',
                    'windDirection',
                    'humidity',
                    'fog',
                    'windSpeed'
                }
            }
        }))
        state = self.hass.states.get('sensor.yr_symbol')
        self.assertEqual(None, state)

        state = self.hass.states.get('sensor.yr_pressure')
        self.assertEqual('hPa',
                         state.attributes.get('unit_of_measurement'))

        state = self.hass.states.get('sensor.yr_wind_direction')
        self.assertEqual('°',
                         state.attributes.get('unit_of_measurement'))

        state = self.hass.states.get('sensor.yr_humidity')
        self.assertEqual('%',
                         state.attributes.get('unit_of_measurement'))

        state = self.hass.states.get('sensor.yr_fog')
        self.assertEqual('%',
                         state.attributes.get('unit_of_measurement'))

        state = self.hass.states.get('sensor.yr_wind_speed')
        self.assertEqual('m/s',
                         state.attributes.get('unit_of_measurement'))