Black
This commit is contained in:
Paulus Schoutsen
parent
da05dfe708
commit
4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
|
|
@ -10,9 +10,15 @@ from unittest import mock
|
|||
from unittest.mock import patch
|
||||
|
||||
from homeassistant.components import feedreader
|
||||
from homeassistant.components.feedreader import CONF_URLS, FeedManager, \
|
||||
StoredData, EVENT_FEEDREADER, DEFAULT_SCAN_INTERVAL, CONF_MAX_ENTRIES, \
|
||||
DEFAULT_MAX_ENTRIES
|
||||
from homeassistant.components.feedreader import (
|
||||
CONF_URLS,
|
||||
FeedManager,
|
||||
StoredData,
|
||||
EVENT_FEEDREADER,
|
||||
DEFAULT_SCAN_INTERVAL,
|
||||
CONF_MAX_ENTRIES,
|
||||
DEFAULT_MAX_ENTRIES,
|
||||
)
|
||||
from homeassistant.const import EVENT_HOMEASSISTANT_START, CONF_SCAN_INTERVAL
|
||||
from homeassistant.core import callback
|
||||
from homeassistant.setup import setup_component
|
||||
|
|
@ -20,24 +26,10 @@ from tests.common import get_test_home_assistant, load_fixture
|
|||
|
||||
_LOGGER = getLogger(__name__)
|
||||
|
||||
URL = 'http://some.rss.local/rss_feed.xml'
|
||||
VALID_CONFIG_1 = {
|
||||
feedreader.DOMAIN: {
|
||||
CONF_URLS: [URL]
|
||||
}
|
||||
}
|
||||
VALID_CONFIG_2 = {
|
||||
feedreader.DOMAIN: {
|
||||
CONF_URLS: [URL],
|
||||
CONF_SCAN_INTERVAL: 60
|
||||
}
|
||||
}
|
||||
VALID_CONFIG_3 = {
|
||||
feedreader.DOMAIN: {
|
||||
CONF_URLS: [URL],
|
||||
CONF_MAX_ENTRIES: 100
|
||||
}
|
||||
}
|
||||
URL = "http://some.rss.local/rss_feed.xml"
|
||||
VALID_CONFIG_1 = {feedreader.DOMAIN: {CONF_URLS: [URL]}}
|
||||
VALID_CONFIG_2 = {feedreader.DOMAIN: {CONF_URLS: [URL], CONF_SCAN_INTERVAL: 60}}
|
||||
VALID_CONFIG_3 = {feedreader.DOMAIN: {CONF_URLS: [URL], CONF_MAX_ENTRIES: 100}}
|
||||
|
||||
|
||||
class TestFeedreaderComponent(unittest.TestCase):
|
||||
|
|
@ -47,7 +39,7 @@ class TestFeedreaderComponent(unittest.TestCase):
|
|||
"""Initialize values for this testcase class."""
|
||||
self.hass = get_test_home_assistant()
|
||||
# Delete any previously stored data
|
||||
data_file = self.hass.config.path("{}.pickle".format('feedreader'))
|
||||
data_file = self.hass.config.path("{}.pickle".format("feedreader"))
|
||||
if exists(data_file):
|
||||
remove(data_file)
|
||||
|
||||
|
|
@ -57,21 +49,23 @@ class TestFeedreaderComponent(unittest.TestCase):
|
|||
|
||||
def test_setup_one_feed(self):
|
||||
"""Test the general setup of this component."""
|
||||
with patch("homeassistant.components.feedreader."
|
||||
"track_time_interval") as track_method:
|
||||
assert setup_component(
|
||||
self.hass, feedreader.DOMAIN, VALID_CONFIG_1)
|
||||
track_method.assert_called_once_with(self.hass, mock.ANY,
|
||||
DEFAULT_SCAN_INTERVAL)
|
||||
with patch(
|
||||
"homeassistant.components.feedreader." "track_time_interval"
|
||||
) as track_method:
|
||||
assert setup_component(self.hass, feedreader.DOMAIN, VALID_CONFIG_1)
|
||||
track_method.assert_called_once_with(
|
||||
self.hass, mock.ANY, DEFAULT_SCAN_INTERVAL
|
||||
)
|
||||
|
||||
def test_setup_scan_interval(self):
|
||||
"""Test the setup of this component with scan interval."""
|
||||
with patch("homeassistant.components.feedreader."
|
||||
"track_time_interval") as track_method:
|
||||
assert setup_component(
|
||||
self.hass, feedreader.DOMAIN, VALID_CONFIG_2)
|
||||
track_method.assert_called_once_with(self.hass, mock.ANY,
|
||||
timedelta(seconds=60))
|
||||
with patch(
|
||||
"homeassistant.components.feedreader." "track_time_interval"
|
||||
) as track_method:
|
||||
assert setup_component(self.hass, feedreader.DOMAIN, VALID_CONFIG_2)
|
||||
track_method.assert_called_once_with(
|
||||
self.hass, mock.ANY, timedelta(seconds=60)
|
||||
)
|
||||
|
||||
def test_setup_max_entries(self):
|
||||
"""Test the setup of this component with max entries."""
|
||||
|
|
@ -91,17 +85,19 @@ class TestFeedreaderComponent(unittest.TestCase):
|
|||
# Loading raw data from fixture and plug in to data object as URL
|
||||
# works since the third-party feedparser library accepts a URL
|
||||
# as well as the actual data.
|
||||
data_file = self.hass.config.path("{}.pickle".format(
|
||||
feedreader.DOMAIN))
|
||||
data_file = self.hass.config.path("{}.pickle".format(feedreader.DOMAIN))
|
||||
storage = StoredData(data_file)
|
||||
with patch("homeassistant.components.feedreader."
|
||||
"track_time_interval") as track_method:
|
||||
manager = FeedManager(feed_data, DEFAULT_SCAN_INTERVAL,
|
||||
max_entries, self.hass, storage)
|
||||
with patch(
|
||||
"homeassistant.components.feedreader." "track_time_interval"
|
||||
) as track_method:
|
||||
manager = FeedManager(
|
||||
feed_data, DEFAULT_SCAN_INTERVAL, max_entries, self.hass, storage
|
||||
)
|
||||
# Can't use 'assert_called_once' here because it's not available
|
||||
# in Python 3.5 yet.
|
||||
track_method.assert_called_once_with(self.hass, mock.ANY,
|
||||
DEFAULT_SCAN_INTERVAL)
|
||||
track_method.assert_called_once_with(
|
||||
self.hass, mock.ANY, DEFAULT_SCAN_INTERVAL
|
||||
)
|
||||
# Artificially trigger update.
|
||||
self.hass.bus.fire(EVENT_HOMEASSISTANT_START)
|
||||
# Collect events.
|
||||
|
|
@ -110,7 +106,7 @@ class TestFeedreaderComponent(unittest.TestCase):
|
|||
|
||||
def test_feed(self):
|
||||
"""Test simple feed with valid data."""
|
||||
feed_data = load_fixture('feedreader.xml')
|
||||
feed_data = load_fixture("feedreader.xml")
|
||||
manager, events = self.setup_manager(feed_data)
|
||||
assert len(events) == 1
|
||||
assert events[0].data.title == "Title 1"
|
||||
|
|
@ -127,41 +123,43 @@ class TestFeedreaderComponent(unittest.TestCase):
|
|||
def test_feed_updates(self):
|
||||
"""Test feed updates."""
|
||||
# 1. Run
|
||||
feed_data = load_fixture('feedreader.xml')
|
||||
feed_data = load_fixture("feedreader.xml")
|
||||
manager, events = self.setup_manager(feed_data)
|
||||
assert len(events) == 1
|
||||
# 2. Run
|
||||
feed_data2 = load_fixture('feedreader1.xml')
|
||||
feed_data2 = load_fixture("feedreader1.xml")
|
||||
# Must patch 'get_timestamp' method because the timestamp is stored
|
||||
# with the URL which in these tests is the raw XML data.
|
||||
with patch("homeassistant.components.feedreader.StoredData."
|
||||
"get_timestamp", return_value=time.struct_time(
|
||||
(2018, 4, 30, 5, 10, 0, 0, 120, 0))):
|
||||
with patch(
|
||||
"homeassistant.components.feedreader.StoredData." "get_timestamp",
|
||||
return_value=time.struct_time((2018, 4, 30, 5, 10, 0, 0, 120, 0)),
|
||||
):
|
||||
manager2, events2 = self.setup_manager(feed_data2)
|
||||
assert len(events2) == 1
|
||||
# 3. Run
|
||||
feed_data3 = load_fixture('feedreader1.xml')
|
||||
with patch("homeassistant.components.feedreader.StoredData."
|
||||
"get_timestamp", return_value=time.struct_time(
|
||||
(2018, 4, 30, 5, 11, 0, 0, 120, 0))):
|
||||
feed_data3 = load_fixture("feedreader1.xml")
|
||||
with patch(
|
||||
"homeassistant.components.feedreader.StoredData." "get_timestamp",
|
||||
return_value=time.struct_time((2018, 4, 30, 5, 11, 0, 0, 120, 0)),
|
||||
):
|
||||
manager3, events3 = self.setup_manager(feed_data3)
|
||||
assert len(events3) == 0
|
||||
|
||||
def test_feed_default_max_length(self):
|
||||
"""Test long feed beyond the default 20 entry limit."""
|
||||
feed_data = load_fixture('feedreader2.xml')
|
||||
feed_data = load_fixture("feedreader2.xml")
|
||||
manager, events = self.setup_manager(feed_data)
|
||||
assert len(events) == 20
|
||||
|
||||
def test_feed_max_length(self):
|
||||
"""Test long feed beyond a configured 5 entry limit."""
|
||||
feed_data = load_fixture('feedreader2.xml')
|
||||
feed_data = load_fixture("feedreader2.xml")
|
||||
manager, events = self.setup_manager(feed_data, max_entries=5)
|
||||
assert len(events) == 5
|
||||
|
||||
def test_feed_without_publication_date_and_title(self):
|
||||
"""Test simple feed with entry without publication date and title."""
|
||||
feed_data = load_fixture('feedreader3.xml')
|
||||
feed_data = load_fixture("feedreader3.xml")
|
||||
manager, events = self.setup_manager(feed_data)
|
||||
assert len(events) == 3
|
||||
|
||||
|
|
@ -172,14 +170,14 @@ class TestFeedreaderComponent(unittest.TestCase):
|
|||
assert len(events) == 0
|
||||
assert manager.last_update_successful is True
|
||||
|
||||
@mock.patch('feedparser.parse', return_value=None)
|
||||
@mock.patch("feedparser.parse", return_value=None)
|
||||
def test_feed_parsing_failed(self, mock_parse):
|
||||
"""Test feed where parsing fails."""
|
||||
data_file = self.hass.config.path("{}.pickle".format(
|
||||
feedreader.DOMAIN))
|
||||
data_file = self.hass.config.path("{}.pickle".format(feedreader.DOMAIN))
|
||||
storage = StoredData(data_file)
|
||||
manager = FeedManager("FEED DATA", DEFAULT_SCAN_INTERVAL,
|
||||
DEFAULT_MAX_ENTRIES, self.hass, storage)
|
||||
manager = FeedManager(
|
||||
"FEED DATA", DEFAULT_SCAN_INTERVAL, DEFAULT_MAX_ENTRIES, self.hass, storage
|
||||
)
|
||||
# Artificially trigger update.
|
||||
self.hass.bus.fire(EVENT_HOMEASSISTANT_START)
|
||||
# Collect events.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue