d4c80f160c
* Add bandit to pre-commit and CI, use to catch known vulnerable XML parsing * Use defusedxml instead of direct xml.etree to parse XML * Move config to tests/bandit.yaml
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
"""The tests for the rss_feed_api component."""
|
|
import asyncio
|
|
|
|
from defusedxml 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"
|