50 lines
1.6 KiB
Python
50 lines
1.6 KiB
Python
"""The tests for the rss_feed_api component."""
|
|
import asyncio
|
|
from xml.etree import ElementTree
|
|
|
|
import pytest
|
|
|
|
from homeassistant.setup import async_setup_component
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_http_client(loop, hass, hass_client):
|
|
"""Set up test fixture."""
|
|
config = {
|
|
'rss_feed_template': {
|
|
'testfeed': {
|
|
'title': 'feed title is {{states.test.test1.state}}',
|
|
'items': [{
|
|
'title': 'item title is {{states.test.test2.state}}',
|
|
'description': 'desc {{states.test.test3.state}}'}]}}}
|
|
|
|
loop.run_until_complete(async_setup_component(hass,
|
|
'rss_feed_template',
|
|
config))
|
|
return loop.run_until_complete(hass_client())
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_get_nonexistant_feed(mock_http_client):
|
|
"""Test if we can retrieve the correct rss feed."""
|
|
resp = yield from mock_http_client.get('/api/rss_template/otherfeed')
|
|
assert resp.status == 404
|
|
|
|
|
|
@asyncio.coroutine
|
|
def test_get_rss_feed(mock_http_client, hass):
|
|
"""Test if we can retrieve the correct rss feed."""
|
|
hass.states.async_set('test.test1', 'a_state_1')
|
|
hass.states.async_set('test.test2', 'a_state_2')
|
|
hass.states.async_set('test.test3', 'a_state_3')
|
|
|
|
resp = yield from mock_http_client.get('/api/rss_template/testfeed')
|
|
assert resp.status == 200
|
|
|
|
text = yield from resp.text()
|
|
|
|
xml = ElementTree.fromstring(text)
|
|
assert xml[0].text == 'feed title is a_state_1'
|
|
assert xml[1][0].text == 'item title is a_state_2'
|
|
assert xml[1][1].text == 'desc a_state_3'
|