"""The tests for the Season sensor platform.""" # pylint: disable=protected-access import unittest from datetime import datetime from homeassistant.setup import setup_component import homeassistant.components.sensor.season as season from tests.common import get_test_home_assistant HEMISPHERE_NORTHERN = { 'homeassistant': { 'latitude': '48.864716', 'longitude': '2.349014', }, 'sensor': { 'platform': 'season', 'type': 'astronomical', } } HEMISPHERE_SOUTHERN = { 'homeassistant': { 'latitude': '-33.918861', 'longitude': '18.423300', }, 'sensor': { 'platform': 'season', 'type': 'astronomical', } } HEMISPHERE_EQUATOR = { 'homeassistant': { 'latitude': '0', 'longitude': '-51.065100', }, 'sensor': { 'platform': 'season', 'type': 'astronomical', } } HEMISPHERE_EMPTY = { 'homeassistant': { }, 'sensor': { 'platform': 'season', 'type': 'meteorological', } } # pylint: disable=invalid-name class TestSeason(unittest.TestCase): """Test the season platform.""" DEVICE = None CONFIG_ASTRONOMICAL = {'type': 'astronomical'} CONFIG_METEOROLOGICAL = {'type': 'meteorological'} def add_devices(self, devices): """Mock add devices.""" for device in devices: self.DEVICE = device def setUp(self): """Setup things to be run when tests are started.""" self.hass = get_test_home_assistant() def tearDown(self): """Stop everything that was started.""" self.hass.stop() def test_season_should_be_summer_northern_astronomical(self): """Test that season should be summer.""" # A known day in summer summer_day = datetime(2017, 9, 3, 0, 0) current_season = season.get_season(summer_day, season.NORTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_SUMMER, current_season) def test_season_should_be_summer_northern_meteorological(self): """Test that season should be summer.""" # A known day in summer summer_day = datetime(2017, 8, 13, 0, 0) current_season = season.get_season(summer_day, season.NORTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_SUMMER, current_season) def test_season_should_be_autumn_northern_astronomical(self): """Test that season should be autumn.""" # A known day in autumn autumn_day = datetime(2017, 9, 23, 0, 0) current_season = season.get_season(autumn_day, season.NORTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_AUTUMN, current_season) def test_season_should_be_autumn_northern_meteorological(self): """Test that season should be autumn.""" # A known day in autumn autumn_day = datetime(2017, 9, 3, 0, 0) current_season = season.get_season(autumn_day, season.NORTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_AUTUMN, current_season) def test_season_should_be_winter_northern_astronomical(self): """Test that season should be winter.""" # A known day in winter winter_day = datetime(2017, 12, 25, 0, 0) current_season = season.get_season(winter_day, season.NORTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_WINTER, current_season) def test_season_should_be_winter_northern_meteorological(self): """Test that season should be winter.""" # A known day in winter winter_day = datetime(2017, 12, 3, 0, 0) current_season = season.get_season(winter_day, season.NORTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_WINTER, current_season) def test_season_should_be_spring_northern_astronomical(self): """Test that season should be spring.""" # A known day in spring spring_day = datetime(2017, 4, 1, 0, 0) current_season = season.get_season(spring_day, season.NORTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_SPRING, current_season) def test_season_should_be_spring_northern_meteorological(self): """Test that season should be spring.""" # A known day in spring spring_day = datetime(2017, 3, 3, 0, 0) current_season = season.get_season(spring_day, season.NORTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_SPRING, current_season) def test_season_should_be_winter_southern_astronomical(self): """Test that season should be winter.""" # A known day in winter winter_day = datetime(2017, 9, 3, 0, 0) current_season = season.get_season(winter_day, season.SOUTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_WINTER, current_season) def test_season_should_be_winter_southern_meteorological(self): """Test that season should be winter.""" # A known day in winter winter_day = datetime(2017, 8, 13, 0, 0) current_season = season.get_season(winter_day, season.SOUTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_WINTER, current_season) def test_season_should_be_spring_southern_astronomical(self): """Test that season should be spring.""" # A known day in spring spring_day = datetime(2017, 9, 23, 0, 0) current_season = season.get_season(spring_day, season.SOUTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_SPRING, current_season) def test_season_should_be_spring_southern_meteorological(self): """Test that season should be spring.""" # A known day in spring spring_day = datetime(2017, 9, 3, 0, 0) current_season = season.get_season(spring_day, season.SOUTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_SPRING, current_season) def test_season_should_be_summer_southern_astronomical(self): """Test that season should be summer.""" # A known day in summer summer_day = datetime(2017, 12, 25, 0, 0) current_season = season.get_season(summer_day, season.SOUTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_SUMMER, current_season) def test_season_should_be_summer_southern_meteorological(self): """Test that season should be summer.""" # A known day in summer summer_day = datetime(2017, 12, 3, 0, 0) current_season = season.get_season(summer_day, season.SOUTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_SUMMER, current_season) def test_season_should_be_autumn_southern_astronomical(self): """Test that season should be spring.""" # A known day in spring autumn_day = datetime(2017, 4, 1, 0, 0) current_season = season.get_season(autumn_day, season.SOUTHERN, season.TYPE_ASTRONOMICAL) self.assertEqual(season.STATE_AUTUMN, current_season) def test_season_should_be_autumn_southern_meteorological(self): """Test that season should be autumn.""" # A known day in autumn autumn_day = datetime(2017, 3, 3, 0, 0) current_season = season.get_season(autumn_day, season.SOUTHERN, season.TYPE_METEOROLOGICAL) self.assertEqual(season.STATE_AUTUMN, current_season) def test_on_equator_results_in_none(self): """Test that season should be unknown.""" # A known day in summer if astronomical and northern summer_day = datetime(2017, 9, 3, 0, 0) current_season = season.get_season(summer_day, season.EQUATOR, season.TYPE_ASTRONOMICAL) self.assertEqual(None, current_season) def test_setup_hemisphere_northern(self): """Test platform setup of northern hemisphere.""" self.hass.config.latitude = HEMISPHERE_NORTHERN[ 'homeassistant']['latitude'] assert setup_component(self.hass, 'sensor', HEMISPHERE_NORTHERN) self.assertEqual(self.hass.config.as_dict()['latitude'], HEMISPHERE_NORTHERN['homeassistant']['latitude']) state = self.hass.states.get('sensor.season') self.assertEqual(state.attributes.get('friendly_name'), 'Season') def test_setup_hemisphere_southern(self): """Test platform setup of southern hemisphere.""" self.hass.config.latitude = HEMISPHERE_SOUTHERN[ 'homeassistant']['latitude'] assert setup_component(self.hass, 'sensor', HEMISPHERE_SOUTHERN) self.assertEqual(self.hass.config.as_dict()['latitude'], HEMISPHERE_SOUTHERN['homeassistant']['latitude']) state = self.hass.states.get('sensor.season') self.assertEqual(state.attributes.get('friendly_name'), 'Season') def test_setup_hemisphere_equator(self): """Test platform setup of equator.""" self.hass.config.latitude = HEMISPHERE_EQUATOR[ 'homeassistant']['latitude'] assert setup_component(self.hass, 'sensor', HEMISPHERE_EQUATOR) self.assertEqual(self.hass.config.as_dict()['latitude'], HEMISPHERE_EQUATOR['homeassistant']['latitude']) state = self.hass.states.get('sensor.season') self.assertEqual(state.attributes.get('friendly_name'), 'Season') def test_setup_hemisphere_empty(self): """Test platform setup of missing latlong.""" self.hass.config.latitude = None assert setup_component(self.hass, 'sensor', HEMISPHERE_EMPTY) self.assertEqual(self.hass.config.as_dict()['latitude'], None)