def test_custom_threshold_works(self, mock_req):
"""Test that the config can take a custom forecast threshold."""
uri = re.compile(
r"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt")
mock_req.get(uri, text=load_fixture('aurora.txt'))
entities = []
def mock_add_entities(new_entities, update_before_add=False):
"""Mock add entities."""
if update_before_add:
for entity in new_entities:
entity.update()
for entity in new_entities:
entities.append(entity)
config = {"name": "Test", "forecast_threshold": 1}
self.hass.config.longitude = 5
self.hass.config.latitude = 5
aurora.setup_platform(self.hass, config, mock_add_entities)
aurora_component = entities[0]
self.assertEqual(aurora_component.aurora_data.visibility_level, '5')
self.assertTrue(aurora_component.is_on)
def test_custom_threshold_works(self, mock_req):
"""Test that the the config can take a custom forecast threshold."""
uri = re.compile(
"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt"
)
mock_req.get(uri, text=load_fixture('aurora.txt'))
entities = []
def mock_add_entities(new_entities, update_before_add=False):
"""Mock add entities."""
if update_before_add:
for entity in new_entities:
entity.update()
for entity in new_entities:
entities.append(entity)
config = {
"name": "Test",
"forecast_threshold": 1
}
self.hass.config.longitude = 5
self.hass.config.latitude = 5
aurora.setup_platform(self.hass, config, mock_add_entities)
aurora_component = entities[0]
self.assertEquals(aurora_component.aurora_data.visibility_level, '5')
self.assertTrue(aurora_component.is_on)
def test_setup_and_initial_state(self, mock_req):
"""Test that the component is created and initialized as expected."""
uri = re.compile(
r"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt")
mock_req.get(uri, text=load_fixture('aurora.txt'))
entities = []
def mock_add_entities(new_entities, update_before_add=False):
"""Mock add entities."""
if update_before_add:
for entity in new_entities:
entity.update()
for entity in new_entities:
entities.append(entity)
config = {"name": "Test", "forecast_threshold": 75}
aurora.setup_platform(self.hass, config, mock_add_entities)
aurora_component = entities[0]
self.assertEqual(len(entities), 1)
self.assertEqual(aurora_component.name, "Test")
self.assertEqual(
aurora_component.device_state_attributes["visibility_level"], '0')
self.assertEqual(aurora_component.device_state_attributes["message"],
"nothing's out")
self.assertFalse(aurora_component.is_on)
def test_setup_and_initial_state(self, mock_req):
"""Test that the component is created and initialized as expected."""
uri = re.compile(
r"http://services\.swpc\.noaa\.gov/text/aurora-nowcast-map\.txt"
)
mock_req.get(uri, text=load_fixture('aurora.txt'))
entities = []
def mock_add_entities(new_entities, update_before_add=False):
"""Mock add entities."""
if update_before_add:
for entity in new_entities:
entity.update()
for entity in new_entities:
entities.append(entity)
config = {
"name": "Test",
"forecast_threshold": 75
}
aurora.setup_platform(self.hass, config, mock_add_entities)
aurora_component = entities[0]
assert len(entities) == 1
assert aurora_component.name == "Test"
assert \
aurora_component.device_state_attributes["visibility_level"] == '0'
assert aurora_component.device_state_attributes["message"] == \
"nothing's out"
assert not aurora_component.is_on