Cleanup the rest reload test to use the pytest requests_mock fixture (#39282)

This commit is contained in:
J. Nick Koston 2020-08-26 10:53:11 -05:00 committed by GitHub
parent 715fe4eef8
commit 79f4b6eb6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -681,26 +681,26 @@ class TestRestData(unittest.TestCase):
assert self.rest.data is None assert self.rest.data is None
async def test_reload(hass): async def test_reload(hass, requests_mock):
"""Verify we can reload reset sensors.""" """Verify we can reload reset sensors."""
with requests_mock.Mocker() as mock_req: requests_mock.get("http://localhost", text="test data")
mock_req.get("http://localhost", text="test data")
await async_setup_component( await async_setup_component(
hass, hass,
"sensor", "sensor",
{ {
"sensor": { "sensor": {
"platform": "rest", "platform": "rest",
"method": "GET", "method": "GET",
"name": "mockrest", "name": "mockrest",
"resource": "http://localhost", "resource": "http://localhost",
} }
}, },
) )
await hass.async_block_till_done() await hass.async_block_till_done()
await hass.async_start() await hass.async_start()
await hass.async_block_till_done() await hass.async_block_till_done()
assert len(hass.states.async_all()) == 1 assert len(hass.states.async_all()) == 1
@ -709,10 +709,7 @@ async def test_reload(hass):
yaml_path = path.join( yaml_path = path.join(
_get_fixtures_base_path(), "fixtures", "rest/configuration.yaml", _get_fixtures_base_path(), "fixtures", "rest/configuration.yaml",
) )
with patch.object( with patch.object(hass_config, "YAML_CONFIG_FILE", yaml_path):
hass_config, "YAML_CONFIG_FILE", yaml_path
), requests_mock.Mocker() as mock_req:
mock_req.get("http://localhost", text="test data 2")
await hass.services.async_call( await hass.services.async_call(
"rest", SERVICE_RELOAD, {}, blocking=True, "rest", SERVICE_RELOAD, {}, blocking=True,
) )