Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Geo rss test style (#42038)

pull/41845/head
Julian Engelhardt 2020-10-19 11:59:04 +02:00 committed by GitHub
parent 5a533a8c5d
commit 97b1fffad4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 104 additions and 108 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

212
tests/components/geo_rss_events/test_sensor.py
View File

@ -1,6 +1,5 @@
"""The test for the geo rss events sensor platform.""" """The test for the geo rss events sensor platform."""
import unittest import pytest
from unittest import mock
from homeassistant.components import sensor from homeassistant.components import sensor
import homeassistant.components.geo_rss_events.sensor as geo_rss_events import homeassistant.components.geo_rss_events.sensor as geo_rss_events
@ -10,15 +9,11 @@ from homeassistant.const import (
ATTR_UNIT_OF_MEASUREMENT, ATTR_UNIT_OF_MEASUREMENT,
EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_START,
) )
from homeassistant.setup import setup_component from homeassistant.setup import async_setup_component
import homeassistant.util.dt as dt_util import homeassistant.util.dt as dt_util
from tests.async_mock import MagicMock, patch from tests.async_mock import MagicMock, patch
from tests.common import ( from tests.common import assert_setup_component, async_fire_time_changed
assert_setup_component,
fire_time_changed,
get_test_home_assistant,
)
URL = "http://geo.rss.local/geo_rss_events.xml" URL = "http://geo.rss.local/geo_rss_events.xml"
VALID_CONFIG_WITH_CATEGORIES = { VALID_CONFIG_WITH_CATEGORIES = {
@ -34,124 +29,125 @@ VALID_CONFIG = {
sensor.DOMAIN: [{"platform": "geo_rss_events", geo_rss_events.CONF_URL: URL}] sensor.DOMAIN: [{"platform": "geo_rss_events", geo_rss_events.CONF_URL: URL}]
} }
"""Test the GeoRss service updater."""
class TestGeoRssServiceUpdater(unittest.TestCase):
"""Test the GeoRss service updater."""
def setUp(self): @pytest.fixture
"""Initialize values for this testcase class.""" def mock_feed():
self.hass = get_test_home_assistant() """Pytest fixture for homeassistant.components.geo_rss_events.sensor.GenericFeed."""
# self.config = VALID_CONFIG_WITHOUT_CATEGORIES with patch(
self.addCleanup(self.hass.stop) "homeassistant.components.geo_rss_events.sensor.GenericFeed"
) as mock_feed:
yield mock_feed
@staticmethod
def _generate_mock_feed_entry(
external_id, title, distance_to_home, coordinates, category
):
"""Construct a mock feed entry for testing purposes."""
feed_entry = MagicMock()
feed_entry.external_id = external_id
feed_entry.title = title
feed_entry.distance_to_home = distance_to_home
feed_entry.coordinates = coordinates
feed_entry.category = category
return feed_entry
@mock.patch("homeassistant.components.geo_rss_events.sensor.GenericFeed") def _generate_mock_feed_entry(
def test_setup(self, mock_feed): external_id, title, distance_to_home, coordinates, category
"""Test the general setup of the platform.""" ):
# Set up some mock feed entries for this test. """Construct a mock feed entry for testing purposes."""
mock_entry_1 = self._generate_mock_feed_entry( feed_entry = MagicMock()
"1234", "Title 1", 15.5, (-31.0, 150.0), "Category 1" feed_entry.external_id = external_id
) feed_entry.title = title
mock_entry_2 = self._generate_mock_feed_entry( feed_entry.distance_to_home = distance_to_home
"2345", "Title 2", 20.5, (-31.1, 150.1), "Category 1" feed_entry.coordinates = coordinates
) feed_entry.category = category
mock_feed.return_value.update.return_value = "OK", [mock_entry_1, mock_entry_2] return feed_entry
utcnow = dt_util.utcnow()
# Patching 'utcnow' to gain more control over the timed update.
with patch("homeassistant.util.dt.utcnow", return_value=utcnow):
with assert_setup_component(1, sensor.DOMAIN):
assert setup_component(self.hass, sensor.DOMAIN, VALID_CONFIG)
# Artificially trigger update.
self.hass.bus.fire(EVENT_HOMEASSISTANT_START)
# Collect events.
self.hass.block_till_done()
all_states = self.hass.states.all() async def test_setup(hass, mock_feed):
assert len(all_states) == 1 """Test the general setup of the platform."""
# Set up some mock feed entries for this test.
state = self.hass.states.get("sensor.event_service_any") mock_entry_1 = _generate_mock_feed_entry(
assert state is not None "1234", "Title 1", 15.5, (-31.0, 150.0), "Category 1"
assert state.name == "Event Service Any" )
assert int(state.state) == 2 mock_entry_2 = _generate_mock_feed_entry(
assert state.attributes == { "2345", "Title 2", 20.5, (-31.1, 150.1), "Category 1"
ATTR_FRIENDLY_NAME: "Event Service Any", )
ATTR_UNIT_OF_MEASUREMENT: "Events", mock_feed.return_value.update.return_value = "OK", [mock_entry_1, mock_entry_2]
ATTR_ICON: "mdi:alert",
"Title 1": "16km",
"Title 2": "20km",
}
# Simulate an update - empty data, but successful update,
# so no changes to entities.
mock_feed.return_value.update.return_value = "OK_NO_DATA", None
fire_time_changed(self.hass, utcnow + geo_rss_events.SCAN_INTERVAL)
self.hass.block_till_done()
all_states = self.hass.states.all()
assert len(all_states) == 1
state = self.hass.states.get("sensor.event_service_any")
assert int(state.state) == 2
# Simulate an update - empty data, removes all entities
mock_feed.return_value.update.return_value = "ERROR", None
fire_time_changed(self.hass, utcnow + 2 * geo_rss_events.SCAN_INTERVAL)
self.hass.block_till_done()
all_states = self.hass.states.all()
assert len(all_states) == 1
state = self.hass.states.get("sensor.event_service_any")
assert int(state.state) == 0
assert state.attributes == {
ATTR_FRIENDLY_NAME: "Event Service Any",
ATTR_UNIT_OF_MEASUREMENT: "Events",
ATTR_ICON: "mdi:alert",
}
@mock.patch("homeassistant.components.geo_rss_events.sensor.GenericFeed")
def test_setup_with_categories(self, mock_feed):
"""Test the general setup of the platform."""
# Set up some mock feed entries for this test.
mock_entry_1 = self._generate_mock_feed_entry(
"1234", "Title 1", 15.5, (-31.0, 150.0), "Category 1"
)
mock_entry_2 = self._generate_mock_feed_entry(
"2345", "Title 2", 20.5, (-31.1, 150.1), "Category 1"
)
mock_feed.return_value.update.return_value = "OK", [mock_entry_1, mock_entry_2]
utcnow = dt_util.utcnow()
# Patching 'utcnow' to gain more control over the timed update.
with patch("homeassistant.util.dt.utcnow", return_value=utcnow):
with assert_setup_component(1, sensor.DOMAIN): with assert_setup_component(1, sensor.DOMAIN):
assert setup_component( assert await async_setup_component(hass, sensor.DOMAIN, VALID_CONFIG)
self.hass, sensor.DOMAIN, VALID_CONFIG_WITH_CATEGORIES
)
# Artificially trigger update. # Artificially trigger update.
self.hass.bus.fire(EVENT_HOMEASSISTANT_START) hass.bus.fire(EVENT_HOMEASSISTANT_START)
# Collect events. # Collect events.
self.hass.block_till_done() await hass.async_block_till_done()
all_states = self.hass.states.all() all_states = hass.states.async_all()
assert len(all_states) == 1 assert len(all_states) == 1
state = self.hass.states.get("sensor.event_service_category_1") state = hass.states.get("sensor.event_service_any")
assert state is not None assert state is not None
assert state.name == "Event Service Category 1" assert state.name == "Event Service Any"
assert int(state.state) == 2 assert int(state.state) == 2
assert state.attributes == { assert state.attributes == {
ATTR_FRIENDLY_NAME: "Event Service Category 1", ATTR_FRIENDLY_NAME: "Event Service Any",
ATTR_UNIT_OF_MEASUREMENT: "Events", ATTR_UNIT_OF_MEASUREMENT: "Events",
ATTR_ICON: "mdi:alert", ATTR_ICON: "mdi:alert",
"Title 1": "16km", "Title 1": "16km",
"Title 2": "20km", "Title 2": "20km",
} }
# Simulate an update - empty data, but successful update,
# so no changes to entities.
mock_feed.return_value.update.return_value = "OK_NO_DATA", None
async_fire_time_changed(hass, utcnow + geo_rss_events.SCAN_INTERVAL)
await hass.async_block_till_done()
all_states = hass.states.async_all()
assert len(all_states) == 1
state = hass.states.get("sensor.event_service_any")
assert int(state.state) == 2
# Simulate an update - empty data, removes all entities
mock_feed.return_value.update.return_value = "ERROR", None
async_fire_time_changed(hass, utcnow + 2 * geo_rss_events.SCAN_INTERVAL)
await hass.async_block_till_done()
all_states = hass.states.async_all()
assert len(all_states) == 1
state = hass.states.get("sensor.event_service_any")
assert int(state.state) == 0
assert state.attributes == {
ATTR_FRIENDLY_NAME: "Event Service Any",
ATTR_UNIT_OF_MEASUREMENT: "Events",
ATTR_ICON: "mdi:alert",
}
async def test_setup_with_categories(hass, mock_feed):
"""Test the general setup of the platform."""
# Set up some mock feed entries for this test.
mock_entry_1 = _generate_mock_feed_entry(
"1234", "Title 1", 15.5, (-31.0, 150.0), "Category 1"
)
mock_entry_2 = _generate_mock_feed_entry(
"2345", "Title 2", 20.5, (-31.1, 150.1), "Category 1"
)
mock_feed.return_value.update.return_value = "OK", [mock_entry_1, mock_entry_2]
with assert_setup_component(1, sensor.DOMAIN):
assert await async_setup_component(
hass, sensor.DOMAIN, VALID_CONFIG_WITH_CATEGORIES
)
# Artificially trigger update.
hass.bus.fire(EVENT_HOMEASSISTANT_START)
# Collect events.
await hass.async_block_till_done()
all_states = hass.states.async_all()
assert len(all_states) == 1
state = hass.states.get("sensor.event_service_category_1")
assert state is not None
assert state.name == "Event Service Category 1"
assert int(state.state) == 2
assert state.attributes == {
ATTR_FRIENDLY_NAME: "Event Service Category 1",
ATTR_UNIT_OF_MEASUREMENT: "Events",
ATTR_ICON: "mdi:alert",
"Title 1": "16km",
"Title 2": "20km",
}